This commit is contained in:
withchao 2023-02-27 18:21:41 +08:00
parent 4cf29ad565
commit cae6c53b01
2 changed files with 31 additions and 19 deletions

View File

@ -1,49 +1,43 @@
package a2r package a2r
import ( import (
"OpenIM/internal/apiresp"
"OpenIM/pkg/common/constant"
"context" "context"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"google.golang.org/grpc" "google.golang.org/grpc"
"net/http"
) )
//// Call TEST
//func Call2[A, B, C, D, E any](
// apiReq *A,
// apiResp *B,
// rpc func(client E, ctx context.Context, req C, options ...grpc.CallOption) (D, error),
// client func() (E, error),
// c *gin.Context,
// before func(apiReq *A, rpcReq *C, bind func() error) error,
// after func(rpcResp *D, apiResp *B, bind func() error) error,
//) {
//
//}
func Call[A, B, C any]( func Call[A, B, C any](
rpc func(client C, ctx context.Context, req *A, options ...grpc.CallOption) (*B, error), rpc func(client C, ctx context.Context, req *A, options ...grpc.CallOption) (*B, error),
client func() (C, error), client func() (C, error),
c *gin.Context, c *gin.Context,
) { ) {
var resp *apiresp.ApiResponse
defer func() {
c.JSON(http.StatusOK, resp)
}()
var req A var req A
if err := c.BindJSON(&req); err != nil { if err := c.BindJSON(&req); err != nil {
// todo 参数错误 resp = apiresp.Error(constant.ErrArgs.Wrap(err.Error())) // 参数错误
return return
} }
if check, ok := any(&req).(interface{ Check() error }); ok { if check, ok := any(&req).(interface{ Check() error }); ok {
if err := check.Check(); err != nil { if err := check.Check(); err != nil {
// todo 参数校验失败 resp = apiresp.Error(constant.ErrArgs.Wrap(err.Error())) // 参数校验失败
return return
} }
} }
cli, err := client() cli, err := client()
if err != nil { if err != nil {
// todo 获取rpc连接失败 resp = apiresp.Error(constant.ErrInternalServer.Wrap(err.Error())) // 参数校验失败
return return
} }
resp, err := rpc(cli, c, &req) data, err := rpc(cli, c, &req)
if err != nil { if err != nil {
// todo rpc请求失败 resp = apiresp.Error(err) // 参数校验失败
return return
} }
_ = resp resp = apiresp.Success(data) // 成功
} }

18
internal/apiresp/resp.go Normal file
View File

@ -0,0 +1,18 @@
package apiresp
type ApiResponse struct {
ErrCode int `json:"errCode"`
ErrMsg string `json:"errMsg"`
ErrDlt string `json:"errDlt"`
Data any `json:"data"`
}
func Success(data any) *ApiResponse {
return &ApiResponse{
Data: data,
}
}
func Error(err error) *ApiResponse {
return &ApiResponse{}
}