feat 排队

internal/ws/manager.go

@@ -1,19 +1,19 @@
 package ws
 
 import (
-	"common/utils"
 	"robot/config"
-	"strconv"
 	"time"
 )
 
 type Manager struct {
-	addr string
+	httpAddr      string
+	websocketAddr string
 }
 
-func NewManager(addr string) *Manager {
+func NewManager(httpAddr, websocketAddr string) *Manager {
 	c := &Manager{
-		addr: addr,
+		httpAddr:      httpAddr,
+		websocketAddr: websocketAddr,
 	}
 	return c
 }
@@ -21,8 +21,7 @@ func NewManager(addr string) *Manager {
 func (c *Manager) Start() {
 	cfg := config.Get().Client
 	for i := int32(0); i < cfg.Count; i++ {
-		client := NewClient(c.addr, strconv.Itoa(utils.RandInt(int(cfg.USN[0]), int(cfg.USN[1]))))
-		client.Start()
+		NewClient(c.httpAddr, c.websocketAddr).Start()
 		time.Sleep(time.Millisecond * 10)
 	}
 }