Fixes memory leaks

.gitignore

@@ -5,3 +5,4 @@ node_modules
 static
 a_main-packr.go
 dozzle
+gin-bin

main.go

@@ -122,6 +122,10 @@ func streamLogs(w http.ResponseWriter, r *http.Request) {
         return
     }
     defer reader.Close()
+    go func() {
+        <-r.Context().Done()
+        reader.Close()
+    }()
 
     w.Header().Set("Content-Type", "text/event-stream")
     w.Header().Set("Cache-Control", "no-cache")
@@ -133,17 +137,15 @@ func streamLogs(w http.ResponseWriter, r *http.Request) {
     for {
         _, err := reader.Read(hdr)
         if err != nil {
-            log.Panicln(err)
+            break
         }
         count := binary.BigEndian.Uint32(hdr[4:])
         n, err := reader.Read(content[:count])
         if err != nil {
-            log.Println(err)
             break
         }
         _, err = fmt.Fprintf(w, "data: %s\n\n", content[:n])
         if err != nil {
-            log.Println(err)
             break
         }
         f.Flush()
@@ -166,7 +168,14 @@ func streamEvents(w http.ResponseWriter, r *http.Request) {
     defer cancel()
     messages, _ := dockerClient.Events(ctx)
 
-    for message := range messages {
+    for {
+        exit := false
+        select {
+        case message, closed := <-messages:
+            if closed {
+                log.Println("Breaking from messages")
+                exit = true
+            }
             switch message.Action {
             case "connect":
                 fallthrough
@@ -183,8 +192,14 @@ func streamEvents(w http.ResponseWriter, r *http.Request) {
                     break
                 }
                 f.Flush()
-        default:
-            // Do nothing
+            }
+        case <-r.Context().Done():
+            cancel()
+            exit = true
+        }
+
+        if exit {
+            break
         }
     }
 }