Fix bug where sniffer would not stop gracefully upon context cancella…

…tion (#248)
otterize · Oct 24, 2024 · a646148 · a646148
1 parent d6d71a6
commit a646148
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/sniffer/pkg/sniffer/sniffer.go b/src/sniffer/pkg/sniffer/sniffer.go
@@ -123,6 +123,8 @@ func (s *Sniffer) RunForever(ctx context.Context) error {
 
 	for {
 		select {
+		case <-ctx.Done():
+			return ctx.Err()
 		case packet := <-dnsPacketsChan:
 			s.dnsSniffer.HandlePacket(packet)
 		case packet := <-tcpPacketsChan: