fix: server

Author: BUGLAN

main.go

@@ -15,6 +15,8 @@ import (
 func main() {
 	// set runtime process
 	runtime.GOMAXPROCS(1)
+
+	// set start time
 	startTime := time.Now()
 
 	// set flag vars
@@ -25,6 +27,7 @@ func main() {
 	flag.BoolVar(&model.Debug, "d", false, "debug模式")
 	flag.Parse()
 
+	// hint
 	if model.Help {
 		flag.Usage()
 		return
@@ -39,14 +42,16 @@ func main() {
 		wgReceive sync.WaitGroup
 	)
 
-	ch := make(chan *model.ReqResult, 200)
+	// create server
+	srv := server.NewServer()
+	ch := make(chan *model.ReqResult, 1000)
 	wgReceive.Add(1)
-	go server.ReceiveData(model.CoroutineNum, ch, &wgReceive)
+	go srv.Receiver(model.CoroutineNum, ch, &wgReceive)
 
+	// do worker
 	for i := 0; i < int(model.CoroutineNum); i++ {
 		wg.Add(1)
-		go server.Stress(model.Url, ch, &wg)
-	}
+		go srv.Worker(model.Url, ch, &wg) }
 
 	wg.Wait()
 	close(ch)

server/statistic.go

@@ -8,7 +8,7 @@ import (
 	"github.com/BUGLAN/stress/model"
 )
 
-func ReceiveData(coroutineNum uint64, ch chan *model.ReqResult, wg *sync.WaitGroup) {
+func (srv *Server) Receiver(coroutineNum uint64, ch chan *model.ReqResult, wg *sync.WaitGroup) {
 	defer wg.Done()
 	stopChan := make(chan struct{})
 	ticker := time.NewTicker(time.Second * 1)

server/stress.go

@@ -0,0 +1,18 @@
+package server
+
+import (
+	"sync"
+
+	"github.com/BUGLAN/stress/model"
+)
+
+type Stress interface {
+	Worker(url string, ch chan *model.ReqResult, wg *sync.WaitGroup)
+	Receiver(coroutineNum uint64, ch chan *model.ReqResult, wg *sync.WaitGroup)
+}
+
+type Server struct{}
+
+func NewServer() *Server {
+	return &Server{}
+}

server/verify.go

@@ -0,0 +1 @@
+package server

server/work.go renamed to server/worker.go

@@ -11,7 +11,7 @@ import (
 )
 
 // Stress run the stress testing
-func Stress(url string, ch chan *model.ReqResult, wg *sync.WaitGroup) {
+func (srv *Server) Worker(url string, ch chan *model.ReqResult, wg *sync.WaitGroup) {
 	var err error
 	for i := 0; i < model.Num; i++ {
 		requestTime := time.Now().UnixNano()