Merge pull request #21 from ddosify/add-proxy-done

Add Done method
getanteon · Oct 23, 2021 · 3855d68 · 3855d68
2 parents c2b5670 + 94ae7b8
commit 3855d68
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/core/engine.go b/core/engine.go
@@ -180,6 +180,7 @@ func (e *engine) stop() {
 	close(e.responseChan)
 	<-e.reportService.DoneChan()
 	e.reportService.Report()
+	e.proxyService.Done()
 }
 
 func (e *engine) initReqCountArr() {

diff --git a/core/proxy/base.go b/core/proxy/base.go
@@ -48,6 +48,7 @@ type ProxyService interface {
 	GetProxy() *url.URL
 	ReportProxy(addr *url.URL, reason string) *url.URL
 	GetProxyCountry(*url.URL) string
+	Done() error
 }
 
 // NewProxyService is the factory method of the ProxyService.

diff --git a/core/proxy/single.go b/core/proxy/single.go
@@ -56,3 +56,7 @@ func (sp *singleProxyStrategy) ReportProxy(addr *url.URL, reason string) *url.UR
 func (sp *singleProxyStrategy) GetProxyCountry(addr *url.URL) string {
 	return "unknown"
 }
+
+func (sp *singleProxyStrategy) Done() error {
+	return nil
+}