fix: close loki stream correctly

traPtitech · May 6, 2023 · b4d2634 · b4d2634
1 parent 3f5c65c
commit b4d2634
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/infrastructure/log/loki/streamer.go b/pkg/infrastructure/log/loki/streamer.go
@@ -90,8 +90,13 @@ func (l *lokiStreamer) Stream(ctx context.Context, appID string, after time.Time
 	ch := make(chan *domain.ContainerLog, 100)
 
 	go func() {
-		defer conn.Close()
+		<-ctx.Done()
+		_ = conn.Close()
 		defer close(ch)
+	}()
+	go func() {
+		defer log.Infof("closing loki websocket stream")
+		log.Infof("new loki websocket stream")
 
 		for {
 			typ, b, err := conn.ReadMessage()