a2r call option

pull/476/head
withchao 1 year ago
parent 7bb069d435
commit c2b7d3bc2c

@ -48,7 +48,9 @@ func (o *FriendApi) AddBlack(c *gin.Context) {
} }
func (o *FriendApi) GetPaginationBlacks(c *gin.Context) { func (o *FriendApi) GetPaginationBlacks(c *gin.Context) {
a2r.Call(friend.FriendClient.GetPaginationBlacks, o.Client, c) a2r.Call(friend.FriendClient.GetPaginationBlacks, o.Client, c, func(resp *friend.GetPaginationBlacksResp) {
a2r.List(&resp.Blacks)
})
} }
func (o *FriendApi) RemoveBlack(c *gin.Context) { func (o *FriendApi) RemoveBlack(c *gin.Context) {

@ -11,10 +11,17 @@ import (
"google.golang.org/grpc" "google.golang.org/grpc"
) )
func List[T any](val *[]T) {
if val != nil && *val == nil {
*val = []T{}
}
}
func Call[A, B, C any]( func Call[A, B, C any](
rpc func(client C, ctx context.Context, req *A, options ...grpc.CallOption) (*B, error), rpc func(client C, ctx context.Context, req *A, options ...grpc.CallOption) (*B, error),
client C, client C,
c *gin.Context, c *gin.Context,
after ...func(resp *B),
) { ) {
var req A var req A
if err := c.BindJSON(&req); err != nil { if err := c.BindJSON(&req); err != nil {
@ -31,5 +38,8 @@ func Call[A, B, C any](
apiresp.GinError(c, err) // RPC调用失败 apiresp.GinError(c, err) // RPC调用失败
return return
} }
for _, fn := range after {
fn(data)
}
apiresp.GinSuccess(c, data) // 成功 apiresp.GinSuccess(c, data) // 成功
} }

Loading…
Cancel
Save