Implement Close method for session (#846)

projectdiscovery · May 27, 2023 · 0a25894 · 0a25894
1 parent acde91c
commit 0a25894
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/v2/pkg/passive/passive.go b/v2/pkg/passive/passive.go
@@ -28,6 +28,7 @@ func (a *Agent) EnumerateSubdomainsWithCtx(ctx context.Context, domain string, p
 			}
 			return
 		}
+		defer session.Close()
 
 		ctx, cancel := context.WithTimeout(ctx, maxEnumTime)
 

diff --git a/v2/pkg/subscraping/agent.go b/v2/pkg/subscraping/agent.go
@@ -123,6 +123,12 @@ func (s *Session) DiscardHTTPResponse(response *http.Response) {
 	}
 }
 
+// Close the session
+func (s *Session) Close() {
+	s.RateLimiter.Stop()
+	s.Client.CloseIdleConnections()
+}
+
 func httpRequestWrapper(client *http.Client, request *http.Request) (*http.Response, error) {
 	response, err := client.Do(request)
 	if err != nil {
-Original file line number
+Diff line change
@@ Expand Up @@
     			}
     			return
     		}
+    		defer session.Close()
     		ctx, cancel := context.WithTimeout(ctx, maxEnumTime)
@@ Expand Down @@