feat 优化网络库

Server/robot/app/app.go

@@ -20,6 +20,7 @@ type Module interface {
 func (p *Program) Init(_ svc.Environment) error {
 	p.moduleList = append(p.moduleList, &ModuleBase{})
 	p.moduleList = append(p.moduleList, &ModuleWebsocket{})
+	p.moduleList = append(p.moduleList, &ModulePprof{})
 
 	for _, module := range p.moduleList {
 		if err := module.init(); err != nil {

Server/robot/app/pprof.go

@@ -0,0 +1,44 @@
+package app
+
+import (
+	"common/log"
+	"context"
+	"errors"
+	"net/http"
+	_ "net/http/pprof"
+	"sync"
+)
+
+// ModulePprof pprof模块
+type ModulePprof struct {
+	wg     *sync.WaitGroup
+	server *http.Server
+}
+
+func (m *ModulePprof) init() error {
+	m.wg = &sync.WaitGroup{}
+	return nil
+}
+
+func (m *ModulePprof) start() error {
+	m.wg.Add(1)
+	go func() {
+		defer m.wg.Done()
+		m.server = &http.Server{
+			Addr: "localhost:6060",
+		}
+		if err := m.server.ListenAndServe(); err != nil && !errors.Is(err, http.ErrServerClosed) {
+			log.Errorf("pprof server failed: %v", err.Error())
+		}
+		log.Infof("pprof server stop.")
+	}()
+	return nil
+}
+
+func (m *ModulePprof) stop() error {
+	if err := m.server.Shutdown(context.Background()); err != nil {
+		log.Errorf("stop pprof server failed: %v", err)
+	}
+	m.wg.Wait()
+	return nil
+}

Server/robot/config/config.dev.yaml

@@ -9,9 +9,9 @@ log:
   maxAge: 7
 
 client:
-  count: 100
+  count: 1100
   usn: [ 1,1000000 ]
   websocket:
-    address: "ws://47.108.184.184/ws/"
-#    address: "ws://127.0.0.1"
-    port: 0
+#    address: "ws://47.108.184.184/ws/"
+    address: "ws://127.0.0.1"
+    port: 8501

Server/robot/internal/ws/handler.go

@@ -6,6 +6,7 @@ import (
 	"google.golang.org/protobuf/proto"
 	"math"
 	"math/rand"
+	"sync"
 	"time"
 )
 
@@ -49,14 +50,21 @@ func (_ *EnterInstance) Handle(data []byte, client *Client) {
 	}()
 }
 
+var s2cPositionPool = sync.Pool{
+	New: func() interface{} {
+		return &sc_pb.Message{}
+	},
+}
+
 type Position struct {
 }
 
 func (_ *Position) Handle(data []byte, client *Client) {
-	msg := &sc_pb.S2C_Position{}
-	if err := proto.Unmarshal(data, msg); err != nil {
-		log.Errorf("handle msg error")
-		client.Stop()
-		return
-	}
+	//msg := s2cPositionPool.Get().(*sc_pb.Message)
+	//if err := proto.Unmarshal(data, msg); err != nil {
+	//	log.Errorf("handle msg error")
+	//	client.Stop()
+	//	return
+	//}
+	//s2cPositionPool.Put(msg)
 }