mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-06 04:15:46 +08:00
1680b96b65
# Conflicts: # cmd/api/main.go # internal/api/a2r/api2rpc.go # internal/apiresp/resp.go # internal/msgtransfer/online_history_msg_handler.go # internal/push/consumer_init.go # pkg/common/db/controller/conversation.go # pkg/common/mw/gin.go # pkg/statistics/statistics.go
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
package a2r
|
|
|
|
import (
|
|
"context"
|
|
"github.com/OpenIMSDK/Open-IM-Server/internal/apiresp"
|
|
"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() (C, error),
|
|
c *gin.Context,
|
|
) {
|
|
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.Wrap(err.Error())) // 参数错误
|
|
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
|
|
}
|
|
}
|
|
cli, err := client()
|
|
if err != nil {
|
|
apiresp.GinError(c, errs.ErrInternalServer.Wrap(err.Error())) // 获取RPC连接失败
|
|
return
|
|
}
|
|
data, err := rpc(cli, c, &req)
|
|
if err != nil {
|
|
apiresp.GinError(c, err) // RPC调用失败
|
|
return
|
|
}
|
|
apiresp.GinSuccess(c, data) // 成功
|
|
}
|