read deadline

Wikidepia · Feb 1, 2025 · 9954d98 · 9954d98
1 parent 4718a04
commit 9954d98
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/handlers/scraper/remote.go b/handlers/scraper/remote.go
@@ -85,6 +85,12 @@ func handleConnection(conn net.Conn) {
 				}
 			}
 
+			if err := conn.SetReadDeadline(time.Now().Add(3 * time.Second)); err != nil {
+				slog.Error("failed to set deadline", "err", err)
+				rm.outChan <- err
+				return
+			}
+
 			outBuf := make([]byte, 1024*1024)
 			n, err := conn.Read(outBuf)
 			if err != nil {