发布v2.5.3版本，更新内容请查看：https://github.com/bufanyun/hotgo/blob/v2.0/docs/guide-zh-CN/start-update-log.md

server/internal/library/network/tcp/rpc.go

@@ -0,0 +1,103 @@
+package tcp
+
+import (
+	"context"
+	"fmt"
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/net/gtcp"
+	"hotgo/internal/consts"
+	"hotgo/utility/simple"
+	"sync"
+	"time"
+)
+
+type Rpc struct {
+	ctx       context.Context
+	mutex     sync.Mutex
+	callbacks map[string]RpcRespFunc
+}
+
+type RpcResp struct {
+	res interface{}
+	err error
+}
+
+type RpcRespFunc func(resp interface{}, err error)
+
+func NewRpc(ctx context.Context) *Rpc {
+	return &Rpc{
+		ctx:       ctx,
+		callbacks: make(map[string]RpcRespFunc),
+	}
+}
+
+// GetCallId 获取回调id
+func (r *Rpc) GetCallId(client *gtcp.Conn, traceID string) string {
+	return fmt.Sprintf("%v.%v", client.LocalAddr().String(), traceID)
+}
+
+// HandleMsg 处理rpc消息
+func (r *Rpc) HandleMsg(ctx context.Context, cancel context.CancelFunc, data interface{}) bool {
+	user := GetCtx(ctx)
+	callId := r.GetCallId(user.Conn, user.TraceID)
+
+	if call, ok := r.callbacks[callId]; ok {
+		r.mutex.Lock()
+		delete(r.callbacks, callId)
+		r.mutex.Unlock()
+
+		simple.SafeGo(ctx, func(ctx context.Context) {
+			call(data, nil)
+			cancel()
+		})
+		return true
+	}
+
+	return false
+}
+
+// Request 发起rpc请求
+func (r *Rpc) Request(callId string, send func()) (res interface{}, err error) {
+	var (
+		waitCh  = make(chan struct{})
+		resCh   = make(chan RpcResp, 1)
+		isClose = false
+	)
+
+	defer func() {
+		isClose = true
+		close(resCh)
+
+		// 移除消息
+		if _, ok := r.callbacks[callId]; ok {
+			r.mutex.Lock()
+			delete(r.callbacks, callId)
+			r.mutex.Unlock()
+		}
+	}()
+
+	simple.SafeGo(r.ctx, func(ctx context.Context) {
+		close(waitCh)
+
+		// 加入回调
+		r.mutex.Lock()
+		r.callbacks[callId] = func(res interface{}, err error) {
+			if !isClose {
+				resCh <- RpcResp{res: res, err: err}
+			}
+		}
+		r.mutex.Unlock()
+
+		// 发送消息
+		send()
+	})
+
+	<-waitCh
+	select {
+	case <-time.After(consts.TCPRpcTimeout):
+		err = gerror.New("rpc response timeout")
+		return
+	case got := <-resCh:
+		return got.res, got.err
+	}
+}