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)