package common import ( "Open_IM/pkg/common/trace_log" "Open_IM/pkg/getcdv3" utils2 "Open_IM/pkg/utils" "fmt" utils "github.com/OpenIMSDK/open_utils" "github.com/gin-gonic/gin" "google.golang.org/grpc/metadata" "reflect" ) func ApiToRpc(c *gin.Context, apiReq, apiResp interface{}, rpcName string, rpcClientFunc interface{}, rpcFuncName string, tokenFunc func(token string, operationID string) (string, error)) { logFuncName := fmt.Sprintf("[ApiToRpc: %s]%s", utils2.GetFuncName(1), rpcFuncName) operationID := c.GetHeader("operationID") nCtx := trace_log.NewCtx1(c, rpcFuncName, operationID) defer trace_log.ShowLog(nCtx) if err := c.BindJSON(apiReq); err != nil { trace_log.WriteErrorResponse(nCtx, "BindJSON", err) return } trace_log.SetContextInfo(nCtx, logFuncName, nil, "apiReq", apiReq) etcdConn, err := getcdv3.GetConn(nCtx, rpcName) if err != nil { trace_log.WriteErrorResponse(nCtx, "GetDefaultConn", err) return } rpc := reflect.ValueOf(rpcClientFunc).Call([]reflect.Value{ reflect.ValueOf(etcdConn), })[0].MethodByName(rpcFuncName) // rpc func rpcReqPtr := reflect.New(rpc.Type().In(1).Elem()) // *req参数 var opUserID string if tokenFunc != nil { var err error opUserID, err = tokenFunc(c.GetHeader("token"), operationID) if err != nil { trace_log.WriteErrorResponse(nCtx, "TokenFunc", err) return } } if err := utils.CopyStructFields(rpcReqPtr.Interface(), apiReq); err != nil { trace_log.WriteErrorResponse(nCtx, "CopyStructFields_RpcReq", err) return } trace_log.SetContextInfo(nCtx, logFuncName, nil, "opUserID", opUserID, "callRpcReq", rpcString(rpcReqPtr.Elem().Interface())) md := metadata.Pairs("operationID", operationID, "opUserID", opUserID) respArr := rpc.Call([]reflect.Value{ reflect.ValueOf(metadata.NewOutgoingContext(c, md)), // context.Context rpcReqPtr, // rpc apiReq }) // respArr => (apiResp, error) if !respArr[1].IsNil() { // rpc err != nil err := respArr[1].Interface().(error) trace_log.WriteErrorResponse(nCtx, rpcFuncName, err, "callRpcResp", "error") return } rpcResp := respArr[0].Elem() trace_log.SetContextInfo(nCtx, rpcFuncName, nil, "callRpcResp", rpcString(rpcResp.Interface())) if apiResp != nil { if err := utils.CopyStructFields(apiResp, rpcResp.Interface()); err != nil { trace_log.SetContextInfo(nCtx, "CopyStructFields_RpcResp", err, "apiResp", fmt.Sprintf("%T", apiResp), "rpcResp", fmt.Sprintf("%T", rpcResp.Interface())) } } trace_log.SetSuccess(nCtx, rpcFuncName, apiResp) } func rpcString(v interface{}) string { if s, ok := v.(interface{ String() string }); ok { return s.String() } return fmt.Sprintf("%+v", v) } //func ApiToRpc(c *gin.Context, apiReq, apiResp interface{}, rpcName string, fn interface{}, rpcFuncName string, tokenFunc func(token string, operationID string) (string, error)) { // nCtx := trace_log.NewCtx(c, rpcFuncName) // defer trace_log.ShowLog(nCtx) // if err := c.BindJSON(apiReq); err != nil { // trace_log.WriteErrorResponse(nCtx, "BindJSON", err) // return // } // reqValue := reflect.ValueOf(apiReq).Elem() // operationID := reqValue.FieldByName("OperationID").String() // trace_log.SetOperationID(nCtx, operationID) // trace_log.SetContextInfo(nCtx, "BindJSON", nil, "params", apiReq) // etcdConn, err := utils2.GetConn(c, rpcName) // if err != nil { // trace_log.WriteErrorResponse(nCtx, "GetDefaultConn", err) // return // } // rpc := reflect.ValueOf(fn).Call([]reflect.Value{ // reflect.ValueOf(etcdConn), // })[0].MethodByName(rpcFuncName) // rpc func // rpcReqPtr := reflect.New(rpc.Type().In(1).Elem()) // *req参数 // var opUserID string // if tokenFunc != nil { // var err error // opUserID, err = tokenFunc(c.GetHeader("token"), operationID) // if err != nil { // trace_log.WriteErrorResponse(nCtx, "TokenFunc", err) // return // } // } // if opID := rpcReqPtr.Elem().FieldByName("OperationID"); opID.IsValid() { // opID.SetString(operationID) // if opU := rpcReqPtr.Elem().FieldByName("OpUserID"); opU.IsValid() { // opU.SetString(opUserID) // } // } else { // op := rpcReqPtr.Elem().FieldByName("Operation").Elem() // op.FieldByName("OperationID").SetString(operationID) // op.FieldByName("OpUserID").SetString(opUserID) // } // if err := utils.CopyStructFields(rpcReqPtr.Interface(), apiReq); err != nil { // trace_log.WriteErrorResponse(nCtx, "CopyStructFields_RpcReq", err) // return // } // respArr := rpc.Call([]reflect.Value{ // reflect.ValueOf(context.Context(c)), // context.Context // rpcReqPtr, // rpc apiReq // }) // respArr => (apiResp, error) // if !respArr[1].IsNil() { // rpc err != nil // err := respArr[1].Interface().(error) // trace_log.WriteErrorResponse(nCtx, rpcFuncName, err, "rpc req", rpcReqPtr.Interface()) // return // } // rpcResp := respArr[0].Elem() // trace_log.SetContextInfo(nCtx, rpcFuncName, nil, "rpc req", rpcReqPtr.Interface(), "resp", rpcResp.Interface()) // commonResp := rpcResp.FieldByName("CommonResp").Elem() // errCodeVal := commonResp.FieldByName("ErrCode") // errMsgVal := commonResp.FieldByName("ErrMsg").Interface().(string) // errCode := errCodeVal.Interface().(int32) // if errCode != 0 { // trace_log.WriteErrorResponse(nCtx, "RpcErrCode", &constant.ErrInfo{ // ErrCode: errCode, // ErrMsg: errMsgVal, // }) // return // } // if apiResp != nil { // if err := utils.CopyStructFields(apiResp, rpcResp.Interface()); err != nil { // trace_log.WriteErrorResponse(nCtx, "CopyStructFields_RpcResp", err) // return // } // } // trace_log.SetSuccess(nCtx, rpcFuncName, apiResp) //}