对服务增加基础的数据统计/部分代码分Package

internal/rpc/utils/mock.go

@@ -0,0 +1,13 @@
+package rpcutils
+
+import "context"
+
+type MockNodeContext struct {
+	context.Context
+
+	NodeId int64
+}
+
+func NewMockNodeContext(nodeId int64) context.Context {
+	return &MockNodeContext{NodeId: nodeId}
+}

internal/rpc/utils/utils.go

@@ -56,6 +56,12 @@ func ValidateRequest(ctx context.Context, userTypes ...UserType) (userType UserT
 		return
 	}
 
+	// 是否是模拟测试
+	mockCtx, isMock := ctx.(*MockNodeContext)
+	if isMock {
+		return UserTypeNode, mockCtx.NodeId, nil
+	}
+
 	md, ok := metadata.FromIncomingContext(ctx)
 	if !ok {
 		return UserTypeNone, 0, errors.New("context: need 'nodeId'")