api to rpc use generic paradigm

internal/api/group/group_test.go

@@ -0,0 +1,107 @@
+package group
+
+import (
+	"Open_IM/pkg/api_struct"
+	"Open_IM/pkg/proto/group"
+	"context"
+	"github.com/gin-gonic/gin"
+	"google.golang.org/grpc"
+	"testing"
+)
+
+type Ignore struct{}
+
+func temp(client group.GroupClient, ctx context.Context, in *group.KickGroupMemberReq, opts ...grpc.CallOption) (*group.KickGroupMemberResp, error) {
+
+	return nil, nil
+}
+
+type ApiBind[A, B any] interface {
+	OperationID() string
+	OpUserID() (string, error)
+	Bind(*A) error
+	Error(error)
+	Write(*B)
+}
+
+func NewApiBind[A, B any](c *gin.Context) ApiBind[A, B] {
+	return &ginApiBind[A, B]{
+		c: c,
+	}
+}
+
+type ginApiBind[A, B any] struct {
+	c *gin.Context
+}
+
+func (g *ginApiBind[A, B]) OperationID() string {
+	return g.c.GetHeader("operationID")
+}
+
+func (g *ginApiBind[A, B]) OpUserID() (string, error) {
+	return "", nil
+}
+
+func (g *ginApiBind[A, B]) Bind(a *A) error {
+	return g.c.BindJSON(a)
+}
+
+func (g *ginApiBind[A, B]) Error(err error) {
+	//TODO implement me
+}
+
+func (g *ginApiBind[A, B]) Write(b *B) {
+	//TODO implement me
+}
+
+func TestName(t *testing.T) {
+	//var bind ApiBind[int, int]
+	//NewRpc(bind, "", group.NewGroupClient, temp)
+
+	var c *gin.Context
+	NewRpc(NewApiBind[api_struct.KickGroupMemberReq, api_struct.KickGroupMemberResp](c), "", group.NewGroupClient, group.GroupClient.KickGroupMember)
+
+}
+
+func KickGroupMember(c *gin.Context) {
+	// 默认 全部自动
+	NewRpc(NewApiBind[api_struct.KickGroupMemberReq, api_struct.KickGroupMemberResp](c), "", group.NewGroupClient, group.GroupClient.KickGroupMember).Execute()
+	// 可以自定义编辑请求和响应
+	NewRpc(NewApiBind[api_struct.KickGroupMemberReq, api_struct.KickGroupMemberResp](c), "", group.NewGroupClient, group.GroupClient.KickGroupMember).Before(func(apiReq *interface{}, rpcReq *interface{}, bind func() error) error {
+		return bind()
+	}).After(func(rpcResp api_struct.KickGroupMemberResp, apiResp *interface{}, bind func() error) error {
+		return bind()
+	}).Execute()
+}
+
+func NewRpc[A, B any, C, D any, Z any](bind ApiBind[A, B], name string, client func(conn *grpc.ClientConn) Z, rpc func(client Z, ctx context.Context, req C, options ...grpc.CallOption) (D, error)) *RpcRun[A, B, C, D, Z] {
+	return &RpcRun[A, B, C, D, Z]{
+		bind:   bind,
+		name:   name,
+		client: client,
+		rpc:    rpc,
+	}
+}
+
+type RpcRun[A, B any, C, D any, Z any] struct {
+	bind   ApiBind[A, B]
+	name   string
+	client func(conn *grpc.ClientConn) Z
+	rpc    func(client Z, ctx context.Context, req C, options ...grpc.CallOption) (D, error)
+	before func(apiReq C, rpcReq C, bind func() error) error
+	after  func(rpcResp B, apiResp D, bind func() error) error
+}
+
+func (a *RpcRun[A, B, C, D, Z]) Before(fn func(apiReq C, rpcReq C, bind func() error) error) *RpcRun[A, B, C, D, Z] {
+	a.before = fn
+	return a
+}
+
+func (a *RpcRun[A, B, C, D, Z]) After(fn func(rpcResp B, apiResp D, bind func() error) error) *RpcRun[A, B, C, D, Z] {
+	a.after = fn
+	return a
+}
+
+func (a *RpcRun[A, B, C, D, Z]) Execute() {
+
+}