open-im-server/pkg/a2r/api2rpc.go

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,
) {
	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, "gin bind json success", "req", req)
	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, "get conn success", "req", req)
	data, err := rpc(cli, c, &req)
	if err != nil {
		apiresp.GinError(c, err) // RPC调用失败
		return
	}
	apiresp.GinSuccess(c, data) // 成功
}