feat(proxymanager): add Count method

Signed-off-by: Dwi Siswanto <git@dw1.io>
mubeng · Sep 15, 2024 · 8a90f36 · 8a90f36
1 parent e51a07a
commit 8a90f36
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/internal/proxymanager/proxymanager.go b/internal/proxymanager/proxymanager.go
@@ -50,7 +50,8 @@ func New(filename string) (*ProxyManager, error) {
 		}
 	}
 
-	manager.Length = len(manager.Proxies)
+	manager.Count()
+
 	if manager.Length < 1 {
 		return manager, fmt.Errorf("open %s: has no valid proxy URLs", filename)
 	}

diff --git a/internal/proxymanager/utils.go b/internal/proxymanager/utils.go
@@ -8,10 +8,17 @@ import (
 	"github.com/kitabisa/mubeng/pkg/helper"
 )
 
+// Count counts total proxies
+func (p *ProxyManager) Count() int {
+	p.Length = len(p.Proxies)
+
+	return p.Length
+}
+
 // NextProxy will navigate the next proxy to use
 func (p *ProxyManager) NextProxy() string {
 	p.CurrentIndex++
-	if p.CurrentIndex > len(p.Proxies)-1 {
+	if p.CurrentIndex > p.Count()-1 {
 		p.CurrentIndex = 0
 	}
 
@@ -22,7 +29,7 @@ func (p *ProxyManager) NextProxy() string {
 
 // RandomProxy will choose a proxy randomly from the list
 func (p *ProxyManager) RandomProxy() string {
-	return p.Proxies[rand.Intn(len(p.Proxies))]
+	return p.Proxies[rand.Intn(p.Count())]
 }
 
 // RemoveProxy removes target proxy from proxy pool