blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-05-12 06:21:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

operationID

Browse Source
This commit is contained in:
withchao 2023-01-03 16:47:11 +08:00
parent 84df63e249
commit b2ce96eb67
1 changed files with 2 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
internal/api_to_rpc/api.go
View File

@ -7,11 +7,10 @@ import (
"context"
utils "github.com/OpenIMSDK/open_utils"
"github.com/gin-gonic/gin"
"net/http"
"reflect"
)
func ApiToRpc(c *gin.Context, apiReq, apiResp interface{}, rpcName string, fn interface{}, rpcFuncName string, tokenFunc func(token string) (string, error)) {
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 {
@ -36,7 +35,7 @@ func ApiToRpc(c *gin.Context, apiReq, apiResp interface{}, rpcName string, fn in
var opUserID string
if tokenFunc != nil {
var err error
opUserID, err = tokenFunc(c.GetHeader("token"))
opUserID, err = tokenFunc(c.GetHeader("token"), operationID)
if err != nil {
trace_log.WriteErrorResponse(nCtx, "TokenFunc", err)
return
@ -86,23 +85,3 @@ func ApiToRpc(c *gin.Context, apiReq, apiResp interface{}, rpcName string, fn in
}
trace_log.SetSuccess(nCtx, rpcFuncName, apiResp)
}
func WriteDataJson(c *gin.Context, code int32, msg string, data interface{}) {
c.JSON(http.StatusOK, &BaseResp{
ErrCode: code,
ErrMsg: msg,
Data: data,
})
}
//func WriteErr(c *gin.Context, err error) {
// if cerr, ok := err.(interface{ Code() int32 }); ok {
// WriteDataJson(c, cerr.Code(), err.Error(), nil)
// } else {
// WriteDataJson(c, ErrDefault.ErrCode, err.Error(), nil)
// }
//}
func WriteSuccess(c *gin.Context, msg string, data interface{}) {
WriteDataJson(c, 0, msg, data)
}