You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.3 KiB
48 lines
1.3 KiB
package a2r
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/apiresp"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/log"
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
|
|
"github.com/gin-gonic/gin"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func Call[A, B, C any](
|
|
rpc func(client C, ctx context.Context, req *A, options ...grpc.CallOption) (*B, error),
|
|
client func(ctx context.Context) (C, error),
|
|
c *gin.Context,
|
|
) {
|
|
log.ZDebug(c, "before bind")
|
|
var req A
|
|
if err := c.BindJSON(&req); err != nil {
|
|
log.ZWarn(c, "gin bind json error", err, "req", req)
|
|
apiresp.GinError(c, errs.ErrArgs.WithDetail(err.Error()).Wrap()) // 参数错误
|
|
return
|
|
}
|
|
if check, ok := any(&req).(interface{ Check() error }); ok {
|
|
if err := check.Check(); err != nil {
|
|
log.ZWarn(c, "custom check error", err, "req", req)
|
|
apiresp.GinError(c, errs.ErrArgs.Wrap(err.Error())) // 参数校验失败
|
|
return
|
|
}
|
|
}
|
|
log.ZDebug(c, "before get grpc conn")
|
|
cli, err := client(c)
|
|
if err != nil {
|
|
log.ZError(c, "get conn error", err, "req", req)
|
|
apiresp.GinError(c, errs.ErrInternalServer.Wrap(err.Error())) // 获取RPC连接失败
|
|
return
|
|
}
|
|
log.ZDebug(c, "before call rpc")
|
|
data, err := rpc(cli, c, &req)
|
|
if err != nil {
|
|
log.ZError(c, "rpc call error", err, "req", req)
|
|
apiresp.GinError(c, err) // RPC调用失败
|
|
return
|
|
}
|
|
apiresp.GinSuccess(c, data) // 成功
|
|
}
|