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.
Open-IM-Server/internal/a2r/api2rpc.go

44 lines
1.0 KiB

package a2r
import (
"OpenIM/internal/apiresp"
"OpenIM/pkg/common/constant"
"context"
"github.com/gin-gonic/gin"
"google.golang.org/grpc"
"net/http"
)
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 resp *apiresp.ApiResponse
defer func() {
c.JSON(http.StatusOK, resp)
}()
var req A
if err := c.BindJSON(&req); err != nil {
resp = apiresp.Error(constant.ErrArgs.Wrap(err.Error())) // 参数错误
return
}
if check, ok := any(&req).(interface{ Check() error }); ok {
if err := check.Check(); err != nil {
resp = apiresp.Error(constant.ErrArgs.Wrap(err.Error())) // 参数校验失败
return
}
}
cli, err := client()
if err != nil {
resp = apiresp.Error(constant.ErrInternalServer.Wrap(err.Error())) // 参数校验失败
return
}
data, err := rpc(cli, c, &req)
if err != nil {
resp = apiresp.Error(err) // 参数校验失败
return
}
resp = apiresp.Success(data) // 成功
}