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/api2rpc/gin.go

99 lines
2.3 KiB

2 years ago
package api2rpc
import (
"context"
"github.com/gin-gonic/gin"
)
//func KickGroupMember(c *gin.Context) {
// // 默认 全部自动
// //var api ApiBind[apistruct.KickGroupMemberReq, apistruct.KickGroupMemberResp] = NewGin[apistruct.KickGroupMemberReq, apistruct.KickGroupMemberResp](c)
// var api ApiBind[apistruct.KickGroupMemberReq, apistruct.KickGroupMemberResp] = nil
// var client func(conn *grpc.ClientConn) group.GroupClient = nil
2 years ago
// var rpcFn func(ctx context.Context, in *group.KickGroupMemberReq, opts ...grpc.CallOption) (*group.KickGroupMemberResp, error) = nil
// //NewRpc(api, client, rpcFn).Name("group").Call()
// NewRpc(api, client, rpcFn).Name("group").Call()
2 years ago
//
// // 可以自定义编辑请求和响应
// //a := NewRpc(NewGin[apistruct.KickGroupMemberReq, apistruct.KickGroupMemberResp](c), "", group.NewGroupClient, group.GroupClient.KickGroupMember)
// //a.Before(func(apiReq *apistruct.KickGroupMemberReq, rpcReq *group.KickGroupMemberReq, bind func() error) error {
// // return bind()
// //}).After(func(rpcResp *group.KickGroupMemberResp, apiResp *apistruct.KickGroupMemberResp, bind func() error) error {
// // return bind()
// //}).Execute()
//}
//
func NewGin[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]) Resp(resp *B, err error) {
if err == nil {
g.Write(resp)
} else {
g.Error(err)
}
}
func (g *ginApiBind[A, B]) Error(err error) {
//TODO implement me
}
func (g *ginApiBind[A, B]) Write(b *B) {
//TODO implement me
}
func (g *ginApiBind[A, B]) Context() context.Context {
return g.c
}
2 years ago
func NewGin1(c *gin.Context) Api {
return &ginApi{
c: c,
}
}
type ginApi struct {
c *gin.Context
}
func (g *ginApi) OperationID() string {
return g.c.GetHeader("operationID")
}
func (g *ginApi) OpUserID() string {
return g.c.MustGet("opUserID").(string)
}
func (g *ginApi) Context() context.Context {
return g.c
}
func (g *ginApi) Bind(req any) error {
return g.c.BindJSON(req)
}
func (g *ginApi) Resp(resp any, err error) {
//TODO implement me
panic("implement me")
}