mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-30 14:56:19 +00:00
YuriandGitHub
9d4ebacc35
## What changed? * Add implementation for DescribeWorker API (to both frontend and matching) * Add feature flag * Add all quotas/etc. * Add unit tests * Add functional tests ## Why? Part of worker commands work. Users should be able to directly get worker info for the worker they need. Corresponding API PR: https://github.com/temporalio/api/pull/622 ## How did you test it? - [X] add unit tests - [X] add functional tests ## Potential risks None, behind feature flag
459 lines
15 KiB
Go
459 lines
15 KiB
Go
// Code generated by cmd/tools/genrpcwrappers. DO NOT EDIT.
|
|
|
|
package matching
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.temporal.io/server/api/matchingservice/v1"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func (c *metricClient) ApplyTaskQueueUserDataReplicationEvent(
|
|
ctx context.Context,
|
|
request *matchingservice.ApplyTaskQueueUserDataReplicationEventRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ApplyTaskQueueUserDataReplicationEventResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientApplyTaskQueueUserDataReplicationEvent")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ApplyTaskQueueUserDataReplicationEvent(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) CancelOutstandingPoll(
|
|
ctx context.Context,
|
|
request *matchingservice.CancelOutstandingPollRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.CancelOutstandingPollResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientCancelOutstandingPoll")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.CancelOutstandingPoll(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) CheckTaskQueueUserDataPropagation(
|
|
ctx context.Context,
|
|
request *matchingservice.CheckTaskQueueUserDataPropagationRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.CheckTaskQueueUserDataPropagationResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientCheckTaskQueueUserDataPropagation")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.CheckTaskQueueUserDataPropagation(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) CreateNexusEndpoint(
|
|
ctx context.Context,
|
|
request *matchingservice.CreateNexusEndpointRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.CreateNexusEndpointResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientCreateNexusEndpoint")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.CreateNexusEndpoint(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) DeleteNexusEndpoint(
|
|
ctx context.Context,
|
|
request *matchingservice.DeleteNexusEndpointRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.DeleteNexusEndpointResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientDeleteNexusEndpoint")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.DeleteNexusEndpoint(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) DescribeTaskQueue(
|
|
ctx context.Context,
|
|
request *matchingservice.DescribeTaskQueueRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.DescribeTaskQueueResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientDescribeTaskQueue")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.DescribeTaskQueue(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) DescribeTaskQueuePartition(
|
|
ctx context.Context,
|
|
request *matchingservice.DescribeTaskQueuePartitionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.DescribeTaskQueuePartitionResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientDescribeTaskQueuePartition")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.DescribeTaskQueuePartition(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) DescribeVersionedTaskQueues(
|
|
ctx context.Context,
|
|
request *matchingservice.DescribeVersionedTaskQueuesRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.DescribeVersionedTaskQueuesResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientDescribeVersionedTaskQueues")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.DescribeVersionedTaskQueues(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) DescribeWorker(
|
|
ctx context.Context,
|
|
request *matchingservice.DescribeWorkerRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.DescribeWorkerResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientDescribeWorker")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.DescribeWorker(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) DispatchNexusTask(
|
|
ctx context.Context,
|
|
request *matchingservice.DispatchNexusTaskRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.DispatchNexusTaskResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientDispatchNexusTask")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.DispatchNexusTask(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) ForceLoadTaskQueuePartition(
|
|
ctx context.Context,
|
|
request *matchingservice.ForceLoadTaskQueuePartitionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ForceLoadTaskQueuePartitionResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientForceLoadTaskQueuePartition")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ForceLoadTaskQueuePartition(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) ForceUnloadTaskQueue(
|
|
ctx context.Context,
|
|
request *matchingservice.ForceUnloadTaskQueueRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ForceUnloadTaskQueueResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientForceUnloadTaskQueue")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ForceUnloadTaskQueue(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) ForceUnloadTaskQueuePartition(
|
|
ctx context.Context,
|
|
request *matchingservice.ForceUnloadTaskQueuePartitionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ForceUnloadTaskQueuePartitionResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientForceUnloadTaskQueuePartition")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ForceUnloadTaskQueuePartition(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) GetBuildIdTaskQueueMapping(
|
|
ctx context.Context,
|
|
request *matchingservice.GetBuildIdTaskQueueMappingRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.GetBuildIdTaskQueueMappingResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientGetBuildIdTaskQueueMapping")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.GetBuildIdTaskQueueMapping(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) GetTaskQueueUserData(
|
|
ctx context.Context,
|
|
request *matchingservice.GetTaskQueueUserDataRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.GetTaskQueueUserDataResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientGetTaskQueueUserData")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.GetTaskQueueUserData(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) GetWorkerBuildIdCompatibility(
|
|
ctx context.Context,
|
|
request *matchingservice.GetWorkerBuildIdCompatibilityRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.GetWorkerBuildIdCompatibilityResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientGetWorkerBuildIdCompatibility")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.GetWorkerBuildIdCompatibility(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) GetWorkerVersioningRules(
|
|
ctx context.Context,
|
|
request *matchingservice.GetWorkerVersioningRulesRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.GetWorkerVersioningRulesResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientGetWorkerVersioningRules")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.GetWorkerVersioningRules(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) ListNexusEndpoints(
|
|
ctx context.Context,
|
|
request *matchingservice.ListNexusEndpointsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ListNexusEndpointsResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientListNexusEndpoints")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ListNexusEndpoints(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) ListTaskQueuePartitions(
|
|
ctx context.Context,
|
|
request *matchingservice.ListTaskQueuePartitionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ListTaskQueuePartitionsResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientListTaskQueuePartitions")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ListTaskQueuePartitions(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) ListWorkers(
|
|
ctx context.Context,
|
|
request *matchingservice.ListWorkersRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ListWorkersResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientListWorkers")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ListWorkers(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) PollNexusTaskQueue(
|
|
ctx context.Context,
|
|
request *matchingservice.PollNexusTaskQueueRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.PollNexusTaskQueueResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientPollNexusTaskQueue")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.PollNexusTaskQueue(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) RecordWorkerHeartbeat(
|
|
ctx context.Context,
|
|
request *matchingservice.RecordWorkerHeartbeatRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.RecordWorkerHeartbeatResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientRecordWorkerHeartbeat")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.RecordWorkerHeartbeat(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) ReplicateTaskQueueUserData(
|
|
ctx context.Context,
|
|
request *matchingservice.ReplicateTaskQueueUserDataRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.ReplicateTaskQueueUserDataResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientReplicateTaskQueueUserData")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.ReplicateTaskQueueUserData(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) RespondNexusTaskCompleted(
|
|
ctx context.Context,
|
|
request *matchingservice.RespondNexusTaskCompletedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.RespondNexusTaskCompletedResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientRespondNexusTaskCompleted")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.RespondNexusTaskCompleted(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) RespondNexusTaskFailed(
|
|
ctx context.Context,
|
|
request *matchingservice.RespondNexusTaskFailedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.RespondNexusTaskFailedResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientRespondNexusTaskFailed")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.RespondNexusTaskFailed(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) RespondQueryTaskCompleted(
|
|
ctx context.Context,
|
|
request *matchingservice.RespondQueryTaskCompletedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.RespondQueryTaskCompletedResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientRespondQueryTaskCompleted")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.RespondQueryTaskCompleted(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) SyncDeploymentUserData(
|
|
ctx context.Context,
|
|
request *matchingservice.SyncDeploymentUserDataRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.SyncDeploymentUserDataResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientSyncDeploymentUserData")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.SyncDeploymentUserData(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) UpdateNexusEndpoint(
|
|
ctx context.Context,
|
|
request *matchingservice.UpdateNexusEndpointRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.UpdateNexusEndpointResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientUpdateNexusEndpoint")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.UpdateNexusEndpoint(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) UpdateTaskQueueConfig(
|
|
ctx context.Context,
|
|
request *matchingservice.UpdateTaskQueueConfigRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.UpdateTaskQueueConfigResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientUpdateTaskQueueConfig")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.UpdateTaskQueueConfig(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) UpdateTaskQueueUserData(
|
|
ctx context.Context,
|
|
request *matchingservice.UpdateTaskQueueUserDataRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.UpdateTaskQueueUserDataResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientUpdateTaskQueueUserData")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.UpdateTaskQueueUserData(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) UpdateWorkerBuildIdCompatibility(
|
|
ctx context.Context,
|
|
request *matchingservice.UpdateWorkerBuildIdCompatibilityRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.UpdateWorkerBuildIdCompatibilityResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientUpdateWorkerBuildIdCompatibility")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.UpdateWorkerBuildIdCompatibility(ctx, request, opts...)
|
|
}
|
|
|
|
func (c *metricClient) UpdateWorkerVersioningRules(
|
|
ctx context.Context,
|
|
request *matchingservice.UpdateWorkerVersioningRulesRequest,
|
|
opts ...grpc.CallOption,
|
|
) (_ *matchingservice.UpdateWorkerVersioningRulesResponse, retError error) {
|
|
|
|
metricsHandler, startTime := c.startMetricsRecording(ctx, "MatchingClientUpdateWorkerVersioningRules")
|
|
defer func() {
|
|
c.finishMetricsRecording(metricsHandler, startTime, retError)
|
|
}()
|
|
|
|
return c.client.UpdateWorkerVersioningRules(ctx, request, opts...)
|
|
}
|