58 lines
2.7 KiB
Go
58 lines
2.7 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
|
|
walletv1 "hyapp.local/api/proto/wallet/v1"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// WalletClient 抽象 gateway 对 wallet-service 余额查询能力的依赖。
|
|
type WalletClient interface {
|
|
GetBalances(ctx context.Context, req *walletv1.GetBalancesRequest) (*walletv1.GetBalancesResponse, error)
|
|
TransferCoinFromSeller(ctx context.Context, req *walletv1.TransferCoinFromSellerRequest) (*walletv1.TransferCoinFromSellerResponse, error)
|
|
ListResources(ctx context.Context, req *walletv1.ListResourcesRequest) (*walletv1.ListResourcesResponse, error)
|
|
GetResourceGroup(ctx context.Context, req *walletv1.GetResourceGroupRequest) (*walletv1.GetResourceGroupResponse, error)
|
|
ListGiftConfigs(ctx context.Context, req *walletv1.ListGiftConfigsRequest) (*walletv1.ListGiftConfigsResponse, error)
|
|
ListUserResources(ctx context.Context, req *walletv1.ListUserResourcesRequest) (*walletv1.ListUserResourcesResponse, error)
|
|
EquipUserResource(ctx context.Context, req *walletv1.EquipUserResourceRequest) (*walletv1.EquipUserResourceResponse, error)
|
|
}
|
|
|
|
type grpcWalletClient struct {
|
|
client walletv1.WalletServiceClient
|
|
}
|
|
|
|
// NewGRPCWalletClient 用 gRPC 连接构造 wallet-service client。
|
|
func NewGRPCWalletClient(conn *grpc.ClientConn) WalletClient {
|
|
return &grpcWalletClient{client: walletv1.NewWalletServiceClient(conn)}
|
|
}
|
|
|
|
func (c *grpcWalletClient) GetBalances(ctx context.Context, req *walletv1.GetBalancesRequest) (*walletv1.GetBalancesResponse, error) {
|
|
return c.client.GetBalances(ctx, req)
|
|
}
|
|
|
|
func (c *grpcWalletClient) TransferCoinFromSeller(ctx context.Context, req *walletv1.TransferCoinFromSellerRequest) (*walletv1.TransferCoinFromSellerResponse, error) {
|
|
return c.client.TransferCoinFromSeller(ctx, req)
|
|
}
|
|
|
|
func (c *grpcWalletClient) ListResources(ctx context.Context, req *walletv1.ListResourcesRequest) (*walletv1.ListResourcesResponse, error) {
|
|
return c.client.ListResources(ctx, req)
|
|
}
|
|
|
|
func (c *grpcWalletClient) GetResourceGroup(ctx context.Context, req *walletv1.GetResourceGroupRequest) (*walletv1.GetResourceGroupResponse, error) {
|
|
return c.client.GetResourceGroup(ctx, req)
|
|
}
|
|
|
|
func (c *grpcWalletClient) ListGiftConfigs(ctx context.Context, req *walletv1.ListGiftConfigsRequest) (*walletv1.ListGiftConfigsResponse, error) {
|
|
return c.client.ListGiftConfigs(ctx, req)
|
|
}
|
|
|
|
func (c *grpcWalletClient) ListUserResources(ctx context.Context, req *walletv1.ListUserResourcesRequest) (*walletv1.ListUserResourcesResponse, error) {
|
|
return c.client.ListUserResources(ctx, req)
|
|
}
|
|
|
|
func (c *grpcWalletClient) EquipUserResource(ctx context.Context, req *walletv1.EquipUserResourceRequest) (*walletv1.EquipUserResourceResponse, error) {
|
|
return c.client.EquipUserResource(ctx, req)
|
|
}
|