mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
767 lines
28 KiB
Go
767 lines
28 KiB
Go
// The MIT License
|
|
//
|
|
// Copyright (c) 2020 Temporal Technologies Inc. All rights reserved.
|
|
//
|
|
// Copyright (c) 2020 Uber Technologies, Inc.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
// THE SOFTWARE.
|
|
|
|
package frontend
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.temporal.io/api/workflowservice/v1"
|
|
"google.golang.org/grpc"
|
|
|
|
"go.temporal.io/server/common/metrics"
|
|
)
|
|
|
|
var _ workflowservice.WorkflowServiceClient = (*metricClient)(nil)
|
|
|
|
type metricClient struct {
|
|
client workflowservice.WorkflowServiceClient
|
|
metricsClient metrics.Client
|
|
}
|
|
|
|
// NewMetricClient creates a new instance of workflowservice.WorkflowServiceClient that emits metrics
|
|
func NewMetricClient(client workflowservice.WorkflowServiceClient, metricsClient metrics.Client) workflowservice.WorkflowServiceClient {
|
|
return &metricClient{
|
|
client: client,
|
|
metricsClient: metricsClient,
|
|
}
|
|
}
|
|
|
|
func (c *metricClient) DeprecateNamespace(
|
|
ctx context.Context,
|
|
request *workflowservice.DeprecateNamespaceRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.DeprecateNamespaceResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDeprecateNamespaceScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientDeprecateNamespaceScope, metrics.ClientLatency)
|
|
resp, err := c.client.DeprecateNamespace(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDeprecateNamespaceScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) DescribeNamespace(
|
|
ctx context.Context,
|
|
request *workflowservice.DescribeNamespaceRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.DescribeNamespaceResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDescribeNamespaceScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientDescribeNamespaceScope, metrics.ClientLatency)
|
|
resp, err := c.client.DescribeNamespace(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDescribeNamespaceScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) DescribeTaskQueue(
|
|
ctx context.Context,
|
|
request *workflowservice.DescribeTaskQueueRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.DescribeTaskQueueResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDescribeTaskQueueScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientDescribeTaskQueueScope, metrics.ClientLatency)
|
|
resp, err := c.client.DescribeTaskQueue(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDescribeTaskQueueScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) DescribeWorkflowExecution(
|
|
ctx context.Context,
|
|
request *workflowservice.DescribeWorkflowExecutionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.DescribeWorkflowExecutionResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDescribeWorkflowExecutionScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientDescribeWorkflowExecutionScope, metrics.ClientLatency)
|
|
resp, err := c.client.DescribeWorkflowExecution(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientDescribeWorkflowExecutionScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) GetWorkflowExecutionHistory(
|
|
ctx context.Context,
|
|
request *workflowservice.GetWorkflowExecutionHistoryRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.GetWorkflowExecutionHistoryResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetWorkflowExecutionHistoryScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientGetWorkflowExecutionHistoryScope, metrics.ClientLatency)
|
|
resp, err := c.client.GetWorkflowExecutionHistory(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetWorkflowExecutionHistoryScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) GetWorkflowExecutionHistoryReverse(
|
|
ctx context.Context,
|
|
request *workflowservice.GetWorkflowExecutionHistoryReverseRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.GetWorkflowExecutionHistoryReverseResponse, error) {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetWorkflowExecutionHistoryReverseScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientGetWorkflowExecutionHistoryReverseScope, metrics.ClientLatency)
|
|
resp, err := c.client.GetWorkflowExecutionHistoryReverse(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetWorkflowExecutionHistoryReverseScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ListArchivedWorkflowExecutions(
|
|
ctx context.Context,
|
|
request *workflowservice.ListArchivedWorkflowExecutionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ListArchivedWorkflowExecutionsResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListArchivedWorkflowExecutionsScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientListArchivedWorkflowExecutionsScope, metrics.ClientLatency)
|
|
resp, err := c.client.ListArchivedWorkflowExecutions(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListArchivedWorkflowExecutionsScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ListClosedWorkflowExecutions(
|
|
ctx context.Context,
|
|
request *workflowservice.ListClosedWorkflowExecutionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ListClosedWorkflowExecutionsResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListClosedWorkflowExecutionsScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientListClosedWorkflowExecutionsScope, metrics.ClientLatency)
|
|
resp, err := c.client.ListClosedWorkflowExecutions(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListClosedWorkflowExecutionsScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ListNamespaces(
|
|
ctx context.Context,
|
|
request *workflowservice.ListNamespacesRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ListNamespacesResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListNamespacesScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientListNamespacesScope, metrics.ClientLatency)
|
|
resp, err := c.client.ListNamespaces(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListNamespacesScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ListOpenWorkflowExecutions(
|
|
ctx context.Context,
|
|
request *workflowservice.ListOpenWorkflowExecutionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ListOpenWorkflowExecutionsResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListOpenWorkflowExecutionsScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientListOpenWorkflowExecutionsScope, metrics.ClientLatency)
|
|
resp, err := c.client.ListOpenWorkflowExecutions(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListOpenWorkflowExecutionsScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ListWorkflowExecutions(
|
|
ctx context.Context,
|
|
request *workflowservice.ListWorkflowExecutionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ListWorkflowExecutionsResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListWorkflowExecutionsScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientListWorkflowExecutionsScope, metrics.ClientLatency)
|
|
resp, err := c.client.ListWorkflowExecutions(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListWorkflowExecutionsScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ScanWorkflowExecutions(
|
|
ctx context.Context,
|
|
request *workflowservice.ScanWorkflowExecutionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ScanWorkflowExecutionsResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientScanWorkflowExecutionsScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientScanWorkflowExecutionsScope, metrics.ClientLatency)
|
|
resp, err := c.client.ScanWorkflowExecutions(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientScanWorkflowExecutionsScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) CountWorkflowExecutions(
|
|
ctx context.Context,
|
|
request *workflowservice.CountWorkflowExecutionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.CountWorkflowExecutionsResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientCountWorkflowExecutionsScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientCountWorkflowExecutionsScope, metrics.ClientLatency)
|
|
resp, err := c.client.CountWorkflowExecutions(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientCountWorkflowExecutionsScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) GetSearchAttributes(
|
|
ctx context.Context,
|
|
request *workflowservice.GetSearchAttributesRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.GetSearchAttributesResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetSearchAttributesScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientGetSearchAttributesScope, metrics.ClientLatency)
|
|
resp, err := c.client.GetSearchAttributes(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetSearchAttributesScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) PollActivityTaskQueue(
|
|
ctx context.Context,
|
|
request *workflowservice.PollActivityTaskQueueRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.PollActivityTaskQueueResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientPollActivityTaskQueueScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientPollActivityTaskQueueScope, metrics.ClientLatency)
|
|
resp, err := c.client.PollActivityTaskQueue(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientPollActivityTaskQueueScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) PollWorkflowTaskQueue(
|
|
ctx context.Context,
|
|
request *workflowservice.PollWorkflowTaskQueueRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.PollWorkflowTaskQueueResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientPollWorkflowTaskQueueScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientPollWorkflowTaskQueueScope, metrics.ClientLatency)
|
|
resp, err := c.client.PollWorkflowTaskQueue(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientPollWorkflowTaskQueueScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) QueryWorkflow(
|
|
ctx context.Context,
|
|
request *workflowservice.QueryWorkflowRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.QueryWorkflowResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientQueryWorkflowScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientQueryWorkflowScope, metrics.ClientLatency)
|
|
resp, err := c.client.QueryWorkflow(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientQueryWorkflowScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RecordActivityTaskHeartbeat(
|
|
ctx context.Context,
|
|
request *workflowservice.RecordActivityTaskHeartbeatRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RecordActivityTaskHeartbeatResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRecordActivityTaskHeartbeatScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRecordActivityTaskHeartbeatScope, metrics.ClientLatency)
|
|
resp, err := c.client.RecordActivityTaskHeartbeat(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRecordActivityTaskHeartbeatScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RecordActivityTaskHeartbeatById(
|
|
ctx context.Context,
|
|
request *workflowservice.RecordActivityTaskHeartbeatByIdRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RecordActivityTaskHeartbeatByIdResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRecordActivityTaskHeartbeatByIdScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRecordActivityTaskHeartbeatByIdScope, metrics.ClientLatency)
|
|
resp, err := c.client.RecordActivityTaskHeartbeatById(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRecordActivityTaskHeartbeatByIdScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RegisterNamespace(
|
|
ctx context.Context,
|
|
request *workflowservice.RegisterNamespaceRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RegisterNamespaceResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRegisterNamespaceScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRegisterNamespaceScope, metrics.ClientLatency)
|
|
resp, err := c.client.RegisterNamespace(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRegisterNamespaceScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RequestCancelWorkflowExecution(
|
|
ctx context.Context,
|
|
request *workflowservice.RequestCancelWorkflowExecutionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RequestCancelWorkflowExecutionResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRequestCancelWorkflowExecutionScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRequestCancelWorkflowExecutionScope, metrics.ClientLatency)
|
|
resp, err := c.client.RequestCancelWorkflowExecution(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRequestCancelWorkflowExecutionScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ResetStickyTaskQueue(
|
|
ctx context.Context,
|
|
request *workflowservice.ResetStickyTaskQueueRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ResetStickyTaskQueueResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientResetStickyTaskQueueScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientResetStickyTaskQueueScope, metrics.ClientLatency)
|
|
resp, err := c.client.ResetStickyTaskQueue(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientResetStickyTaskQueueScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ResetWorkflowExecution(
|
|
ctx context.Context,
|
|
request *workflowservice.ResetWorkflowExecutionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ResetWorkflowExecutionResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientResetWorkflowExecutionScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientResetWorkflowExecutionScope, metrics.ClientLatency)
|
|
resp, err := c.client.ResetWorkflowExecution(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientResetWorkflowExecutionScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondActivityTaskCanceled(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondActivityTaskCanceledRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondActivityTaskCanceledResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCanceledScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondActivityTaskCanceledScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondActivityTaskCanceled(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCanceledScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondActivityTaskCanceledById(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondActivityTaskCanceledByIdRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondActivityTaskCanceledByIdResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCanceledByIdScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondActivityTaskCanceledByIdScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondActivityTaskCanceledById(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCanceledByIdScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondActivityTaskCompleted(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondActivityTaskCompletedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondActivityTaskCompletedResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCompletedScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondActivityTaskCompletedScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondActivityTaskCompleted(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCompletedScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondActivityTaskCompletedById(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondActivityTaskCompletedByIdRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondActivityTaskCompletedByIdResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCompletedByIdScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondActivityTaskCompletedByIdScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondActivityTaskCompletedById(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskCompletedByIdScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondActivityTaskFailed(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondActivityTaskFailedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondActivityTaskFailedResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskFailedScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondActivityTaskFailedScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondActivityTaskFailed(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskFailedScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondActivityTaskFailedById(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondActivityTaskFailedByIdRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondActivityTaskFailedByIdResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskFailedByIdScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondActivityTaskFailedByIdScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondActivityTaskFailedById(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondActivityTaskFailedByIdScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondWorkflowTaskCompleted(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondWorkflowTaskCompletedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondWorkflowTaskCompletedResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondWorkflowTaskCompletedScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondWorkflowTaskCompletedScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondWorkflowTaskCompleted(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondWorkflowTaskCompletedScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondWorkflowTaskFailed(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondWorkflowTaskFailedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondWorkflowTaskFailedResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondWorkflowTaskFailedScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondWorkflowTaskFailedScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondWorkflowTaskFailed(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondWorkflowTaskFailedScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) RespondQueryTaskCompleted(
|
|
ctx context.Context,
|
|
request *workflowservice.RespondQueryTaskCompletedRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.RespondQueryTaskCompletedResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondQueryTaskCompletedScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientRespondQueryTaskCompletedScope, metrics.ClientLatency)
|
|
resp, err := c.client.RespondQueryTaskCompleted(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientRespondQueryTaskCompletedScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) SignalWithStartWorkflowExecution(
|
|
ctx context.Context,
|
|
request *workflowservice.SignalWithStartWorkflowExecutionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.SignalWithStartWorkflowExecutionResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientSignalWithStartWorkflowExecutionScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientSignalWithStartWorkflowExecutionScope, metrics.ClientLatency)
|
|
resp, err := c.client.SignalWithStartWorkflowExecution(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientSignalWithStartWorkflowExecutionScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) SignalWorkflowExecution(
|
|
ctx context.Context,
|
|
request *workflowservice.SignalWorkflowExecutionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.SignalWorkflowExecutionResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientSignalWorkflowExecutionScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientSignalWorkflowExecutionScope, metrics.ClientLatency)
|
|
resp, err := c.client.SignalWorkflowExecution(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientSignalWorkflowExecutionScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) StartWorkflowExecution(
|
|
ctx context.Context,
|
|
request *workflowservice.StartWorkflowExecutionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.StartWorkflowExecutionResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientStartWorkflowExecutionScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientStartWorkflowExecutionScope, metrics.ClientLatency)
|
|
resp, err := c.client.StartWorkflowExecution(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientStartWorkflowExecutionScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) TerminateWorkflowExecution(
|
|
ctx context.Context,
|
|
request *workflowservice.TerminateWorkflowExecutionRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.TerminateWorkflowExecutionResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientTerminateWorkflowExecutionScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientTerminateWorkflowExecutionScope, metrics.ClientLatency)
|
|
resp, err := c.client.TerminateWorkflowExecution(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientTerminateWorkflowExecutionScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) UpdateNamespace(
|
|
ctx context.Context,
|
|
request *workflowservice.UpdateNamespaceRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.UpdateNamespaceResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientUpdateNamespaceScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientUpdateNamespaceScope, metrics.ClientLatency)
|
|
resp, err := c.client.UpdateNamespace(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientUpdateNamespaceScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) GetClusterInfo(
|
|
ctx context.Context,
|
|
request *workflowservice.GetClusterInfoRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.GetClusterInfoResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetClusterInfoScope, metrics.ClientRequests)
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientGetClusterInfoScope, metrics.ClientLatency)
|
|
resp, err := c.client.GetClusterInfo(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetClusterInfoScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) GetSystemInfo(
|
|
ctx context.Context,
|
|
request *workflowservice.GetSystemInfoRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.GetSystemInfoResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetSystemInfoScope, metrics.ClientRequests)
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientGetSystemInfoScope, metrics.ClientLatency)
|
|
resp, err := c.client.GetSystemInfo(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientGetSystemInfoScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|
|
|
|
func (c *metricClient) ListTaskQueuePartitions(
|
|
ctx context.Context,
|
|
request *workflowservice.ListTaskQueuePartitionsRequest,
|
|
opts ...grpc.CallOption,
|
|
) (*workflowservice.ListTaskQueuePartitionsResponse, error) {
|
|
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListTaskQueuePartitionsScope, metrics.ClientRequests)
|
|
|
|
sw := c.metricsClient.StartTimer(metrics.FrontendClientListTaskQueuePartitionsScope, metrics.ClientLatency)
|
|
resp, err := c.client.ListTaskQueuePartitions(ctx, request, opts...)
|
|
sw.Stop()
|
|
|
|
if err != nil {
|
|
c.metricsClient.IncCounter(metrics.FrontendClientListTaskQueuePartitionsScope, metrics.ClientFailures)
|
|
}
|
|
return resp, err
|
|
}
|