1 Star 0 Fork 0

csingo / cRpc

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
RpcController.go 1.96 KB
一键复制 编辑 原始数据 按行查看 历史
joe 提交于 2023-12-14 09:55 . update
package cRpc
import (
"errors"
"fmt"
"reflect"
"github.com/gin-gonic/gin"
"gitee.com/csingo/cCommon"
"gitee.com/csingo/cContext"
"gitee.com/csingo/cHelper"
"gitee.com/csingo/cLog"
"gitee.com/csingo/cResponse"
)
type RpcController struct {
}
func (c *RpcController) ControllerName() (app, name string) {
return "csingo", "RpcController"
}
func (c *RpcController) Handler(ctx *gin.Context) (err error) {
rpcApp := ctx.Request.Header.Get(cCommon.Header_Rpc_App)
rpcService := ctx.Request.Header.Get(cCommon.Header_Rpc_Service)
rpcMethod := ctx.Request.Header.Get(cCommon.Header_Rpc_Method)
rpcSign := ctx.Request.Header.Get(cCommon.Header_Rpc_Sign)
if container.DecryptFunc != nil {
if !container.DecryptFunc(ctx) {
return cResponse.Api(ctx, nil, errors.New("签名错误"))
}
} else if clientContainer.Signature != "" {
traceid := cContext.GetTraceId(ctx)
sign := cHelper.Md5([]byte(fmt.Sprintf("%s:%s", clientContainer.Signature, traceid)))
if rpcSign != sign {
return cResponse.Api(ctx, nil, errors.New("签名错误"))
}
}
instance := GetService(rpcApp, rpcService)
if instance == nil {
return cResponse.Api(ctx, nil, errors.New("rpc 服务异常"))
}
caller := reflect.ValueOf(instance).MethodByName(rpcMethod)
if !caller.IsValid() {
return cResponse.Api(ctx, nil, errors.New("rpc 服务方法不存在"))
}
paramType := caller.Type().In(1)
param := reflect.New(paramType)
err = ctx.ShouldBind(param.Interface())
if err != nil {
cLog.WithContext(ctx, map[string]interface{}{
"source": "cRpc.RpcController.Handler:ctx.BindJSON",
"err": err.Error(),
}).Error("rpc 调用参数解析异常")
return cResponse.Api(ctx, nil, errors.New("rpc 服务调用参数解析失败"))
}
responseValues := caller.Call([]reflect.Value{reflect.ValueOf(ctx), param.Elem()})
if !responseValues[1].IsNil() {
return cResponse.Api(ctx, nil, responseValues[1].Interface().(error))
}
return cResponse.Api(ctx, responseValues[0].Interface(), nil)
}
Go
1
https://gitee.com/csingo/cRpc.git
git@gitee.com:csingo/cRpc.git
csingo
cRpc
cRpc
v0.2.52

搜索帮助