first commit

pkg/websocket/service/service_manager.go

@@ -0,0 +1,59 @@
+package service
+
+import (
+	"fmt"
+	"git.bvbej.com/bvbej/base-golang/pkg/websocket/codec"
+	_ "git.bvbej.com/bvbej/base-golang/pkg/websocket/codec/json"
+	"git.bvbej.com/bvbej/base-golang/pkg/websocket/peer"
+	"go.uber.org/zap"
+)
+
+var (
+	serviceLogger *zap.Logger
+
+	RegisteredServiceList = make(map[string]*Service) // all registered service
+	RouterCodec           codec.Codec
+)
+
+const ChildName = "Hbase"
+
+type Hbase any
+
+func RegisterService(logger *zap.Logger, comp ...Component) {
+	serviceLogger = logger
+	for _, v := range comp {
+		s := NewService(v)
+		if _, ok := RegisteredServiceList[s.Name]; ok {
+			serviceLogger.Sugar().Errorf("service: service already defined: %s", s.Name)
+		}
+		if err := s.ExtractHandler(); err != nil {
+			serviceLogger.Sugar().Errorf("service: extract handler function failed: %v", err)
+		}
+		RegisteredServiceList[s.Name] = s
+		for name, handler := range s.Handlers {
+			router := fmt.Sprintf("%s.%s", s.Name, name)
+			//注册消息 用于解码
+			codec.RegisterMessage(router, handler.Type)
+			serviceLogger.Sugar().Debugf("service: router %s param %s registed", router, handler.Type)
+		}
+	}
+}
+
+func SetCodec(name string) error {
+	RouterCodec = codec.GetCodec(name)
+	if RouterCodec == nil {
+		return fmt.Errorf("service: codec %s not registered", name)
+	}
+	return nil
+}
+
+func Send(session *peer.Session, router string, data any) error {
+	if RouterCodec == nil {
+		return fmt.Errorf("service: codec not set")
+	}
+	rb, err := RouterCodec.Marshal(router, data, nil)
+	if err != nil {
+		return fmt.Errorf("service: %v", err)
+	}
+	return session.Conn.Send(rb)
+}