37 lines
1.5 KiB
Go
37 lines
1.5 KiB
Go
package activityclient
|
|
|
|
import (
|
|
"context"
|
|
|
|
activityv1 "hyapp.local/api/proto/activity/v1"
|
|
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// Client 是 admin-server 访问 activity-service 任务配置能力的最小 gRPC 依赖。
|
|
type Client interface {
|
|
ListTaskDefinitions(ctx context.Context, req *activityv1.ListTaskDefinitionsRequest) (*activityv1.ListTaskDefinitionsResponse, error)
|
|
UpsertTaskDefinition(ctx context.Context, req *activityv1.UpsertTaskDefinitionRequest) (*activityv1.UpsertTaskDefinitionResponse, error)
|
|
SetTaskDefinitionStatus(ctx context.Context, req *activityv1.SetTaskDefinitionStatusRequest) (*activityv1.SetTaskDefinitionStatusResponse, error)
|
|
}
|
|
|
|
type GRPCClient struct {
|
|
client activityv1.AdminTaskServiceClient
|
|
}
|
|
|
|
func NewGRPC(conn grpc.ClientConnInterface) *GRPCClient {
|
|
return &GRPCClient{client: activityv1.NewAdminTaskServiceClient(conn)}
|
|
}
|
|
|
|
func (c *GRPCClient) ListTaskDefinitions(ctx context.Context, req *activityv1.ListTaskDefinitionsRequest) (*activityv1.ListTaskDefinitionsResponse, error) {
|
|
return c.client.ListTaskDefinitions(ctx, req)
|
|
}
|
|
|
|
func (c *GRPCClient) UpsertTaskDefinition(ctx context.Context, req *activityv1.UpsertTaskDefinitionRequest) (*activityv1.UpsertTaskDefinitionResponse, error) {
|
|
return c.client.UpsertTaskDefinition(ctx, req)
|
|
}
|
|
|
|
func (c *GRPCClient) SetTaskDefinitionStatus(ctx context.Context, req *activityv1.SetTaskDefinitionStatusRequest) (*activityv1.SetTaskDefinitionStatusResponse, error) {
|
|
return c.client.SetTaskDefinitionStatus(ctx, req)
|
|
}
|