wshandler peer 추가

wshandler/wshandler.go

@@ -86,9 +86,7 @@ type send_msg_queue_elem struct {
 	msg []byte
 }
 
-type WebsocketHandler struct {
-	WebsocketApiBroker
-
+type websocketHandlerBase struct {
 	redisMsgChanName  string
 	redisCmdChanName  string
 	redisSync         *redis.Client
@@ -101,6 +99,16 @@ type WebsocketHandler struct {
 	sessionConsumer session.Consumer
 }
 
+type WebsocketHandler struct {
+	WebsocketApiBroker
+	*websocketHandlerBase
+}
+
+type WebsocketPeerHandler struct {
+	WebsocketPeerApiBroker
+	*websocketHandlerBase
+}
+
 type wsConfig struct {
 	gocommon.StorageAddr `json:"storage"`
 }
@@ -116,7 +124,7 @@ func init() {
 	gob.Register([]any{})
 }
 
-func NewWebsocketHandler(consumer session.Consumer, redisUrl string) (*WebsocketHandler, error) {
+func makeWebsocketHandlerBase(consumer session.Consumer, redisUrl string) (*websocketHandlerBase, error) {
 	var config wsConfig
 	if err := gocommon.LoadConfig(&config); err != nil {
 		return nil, err
@@ -151,7 +159,7 @@ func NewWebsocketHandler(consumer session.Consumer, redisUrl string) (*Websocket
 		}
 	}()
 
-	return &WebsocketHandler{
+	return &websocketHandlerBase{
 		redisMsgChanName:  fmt.Sprintf("_wsh_msg_%d", redisSync.Options().DB),
 		redisCmdChanName:  fmt.Sprintf("_wsh_cmd_%d", redisSync.Options().DB),
 		redisSync:         redisSync,
@@ -163,6 +171,28 @@ func NewWebsocketHandler(consumer session.Consumer, redisUrl string) (*Websocket
 	}, nil
 }
 
+func NewWebsocketPeerHandler(consumer session.Consumer, redisUrl string) (*WebsocketPeerHandler, error) {
+	base, err := makeWebsocketHandlerBase(consumer, redisUrl)
+	if err != nil {
+		return nil, err
+	}
+
+	return &WebsocketPeerHandler{
+		websocketHandlerBase: base,
+	}, nil
+}
+
+func NewWebsocketHandler(consumer session.Consumer, redisUrl string) (*WebsocketHandler, error) {
+	base, err := makeWebsocketHandlerBase(consumer, redisUrl)
+	if err != nil {
+		return nil, err
+	}
+
+	return &WebsocketHandler{
+		websocketHandlerBase: base,
+	}, nil
+}
+
 func (ws *WebsocketHandler) Start(ctx context.Context) {
 	ws.connWaitGroup.Add(1)
 	go ws.mainLoop(ctx)