From 3471ce066e4063927d04038bf42c11983b86ae50 Mon Sep 17 00:00:00 2001 From: Mathieu Lecarme Date: Fri, 17 Jun 2022 15:13:28 +0200 Subject: [PATCH] Close the websocket. --- pkg/acquisition/modules/loki/loki.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pkg/acquisition/modules/loki/loki.go b/pkg/acquisition/modules/loki/loki.go index 4c60c663b..cb8cade81 100644 --- a/pkg/acquisition/modules/loki/loki.go +++ b/pkg/acquisition/modules/loki/loki.go @@ -340,6 +340,10 @@ func (l *LokiSource) StreamingAcquisition(out chan types.Event, t *tomb.Tomb) er for { ctx, cancel := context.WithTimeout(context.TODO(), readyTimeout) defer cancel() + go func() { + <-t.Dying() + cancel() // close the websocket. + }() c, res, err := l.dialer.DialContext(ctx, l.lokiWebsocket, l.header) if err != nil { buf, err2 := ioutil.ReadAll(res.Body)