beenull/crowdsec
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Close the websocket.

Browse source
This commit is contained in:
Mathieu Lecarme 2022-06-17 15:13:28 +02:00 committed by lperdereau
parent 066d2a1dc5
commit 3471ce066e
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pkg/acquisition/modules/loki/loki.go
View file

@ -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)