Prevents blocking when waiting for idle timeout

robgonnella · Jul 15, 2024 · 0fce8c4 · 0fce8c4
1 parent 6a50156
commit 0fce8c4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 16 deletions.
diff --git a/pkg/scanner/arpscan.go b/pkg/scanner/arpscan.go
@@ -127,15 +127,15 @@ func (s *ArpScanner) Scan() error {
 		})
 	}
 
-	time.Sleep(s.idleTimeout)
+	time.AfterFunc(s.idleTimeout, func() {
+		go s.Stop()
 
-	go s.Stop()
-
-	go func() {
-		s.resultChan <- &ScanResult{
-			Type: ARPDone,
-		}
-	}()
+		go func() {
+			s.resultChan <- &ScanResult{
+				Type: ARPDone,
+			}
+		}()
+	})
 
 	return err
 }

diff --git a/pkg/scanner/synscan.go b/pkg/scanner/synscan.go
@@ -156,15 +156,15 @@ func (s *SynScanner) Scan() error {
 		}
 	}
 
-	time.Sleep(s.idleTimeout)
+	time.AfterFunc(s.idleTimeout, func() {
+		go s.Stop()
 
-	go s.Stop()
-
-	go func() {
-		s.resultChan <- &ScanResult{
-			Type: SYNDone,
-		}
-	}()
+		go func() {
+			s.resultChan <- &ScanResult{
+				Type: SYNDone,
+			}
+		}()
+	})
 
 	return nil
 }