Open-IM-Server/internal/api_to_rpc/api.go

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)
//}