base-golang/pkg/websocket/service/session_callback.go

package service

import (
	"fmt"
	"git.bvbej.com/bvbej/base-golang/pkg/websocket/util"
	"reflect"
	"strings"
	"time"

	"git.bvbej.com/bvbej/base-golang/pkg/websocket/peer"
)

type callBackEntity struct{}

func GetSessionManager() *peer.SessionManager {
	return peer.NewSessionMgr(&callBackEntity{})
}

func (cb *callBackEntity) OnClosed(session *peer.Session) {
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(fmt.Sprintf("OnClosed: session:%d err:%v", session.Conn.ID(), err))
		}
	}()
	for _, v := range RegisteredServiceList {
		if ok := v.OnSessionClose(session); ok {
			return
		}
	}
}

func (cb *callBackEntity) OnReceive(session *peer.Session, msg []byte) error {
	_, msgPack, err := RouterCodec.Unmarshal(msg)
	if err != nil {
		return fmt.Errorf("onreceive: %v", err)
	}
	router, ok := msgPack.Router.(string)
	if !ok {
		return fmt.Errorf("onreceive: invalid router:%v", msgPack.Router)
	}
	routerArr := strings.Split(router, ".")
	if len(routerArr) != 2 {
		return fmt.Errorf("onreceive: invalid router:%s", msgPack.Router)
	}
	s, ok := RegisteredServiceList[routerArr[0]]
	if !ok {
		return fmt.Errorf("onreceive: function not registed router:%s err:%v", msgPack.Router, err)
	}
	h, ok := s.Handlers[routerArr[1]]
	if !ok {
		return fmt.Errorf("onreceive: function not registed router:%s err:%v", msgPack.Router, err)
	}
	t1 := time.Now()

	var args = []reflect.Value{h.Receiver, reflect.ValueOf(session), reflect.ValueOf(msgPack.DataPtr)}
	var res any
	var rb []byte
	res, err = util.CallHandlerFunc(h.Method, args)
	if res != nil && !reflect.ValueOf(res).IsNil() {
		rb, err = RouterCodec.Marshal(router, res, nil)
		if err != nil {
			return fmt.Errorf("service: %v", err)
		}
		err = session.Conn.Send(rb)
		if err != nil {
			serviceLogger.Sugar().Warnf("warn! service send msg failed router:%s err:%v", router, err)
		}
	} else {
		rb, err = RouterCodec.Marshal(router, nil, err)
		if err != nil {
			return fmt.Errorf("service: %v", err)
		}
		err = session.Conn.Send(rb)
		if err != nil {
			serviceLogger.Sugar().Warnf("warn! service send msg failed router:%s err:%v", router, err)
		}
	}
	var errs string
	if err != nil {
		errs = err.Error()
	}
	dt := time.Since(t1)
	go s.Component.OnRequestFinished(session, router, RouterCodec.ToString(msgPack.DataPtr), errs, dt)
	return nil
}