diff --git a/common/metrics/metric_defs.go b/common/metrics/metric_defs.go index dbae56d9e554..8b73f306414e 100644 --- a/common/metrics/metric_defs.go +++ b/common/metrics/metric_defs.go @@ -694,6 +694,10 @@ const ( PersistenceRangeCompleteVisibilityTasksScope = "RangeCompleteVisibilityTasks" // PersistenceGetReplicationTaskScope tracks GetReplicationTask calls made by service to persistence layer PersistenceGetReplicationTaskScope = "GetReplicationTask" + // PersistenceGetArchivalTaskScope tracks GetArchivalTask calls made by service to persistence layer + PersistenceGetArchivalTaskScope = "GetArchivalTask" + // PersistenceGetArchivalTasksScope tracks GetArchivalTasks calls made by service to persistence layer + PersistenceGetArchivalTasksScope = "GetArchivalTasks" // PersistenceGetReplicationTasksScope tracks GetReplicationTasks calls made by service to persistence layer PersistenceGetReplicationTasksScope = "GetReplicationTasks" // PersistenceCompleteReplicationTaskScope tracks CompleteReplicationTasks calls made by service to persistence layer diff --git a/common/persistence/persistenceMetricClients.go b/common/persistence/persistenceMetricClients.go index 3317ef539d51..a3d1bc8060eb 100644 --- a/common/persistence/persistenceMetricClients.go +++ b/common/persistence/persistenceMetricClients.go @@ -331,6 +331,8 @@ func (p *executionPersistenceClient) GetHistoryTask( operation = metrics.PersistenceGetVisibilityTaskScope case tasks.CategoryIDReplication: operation = metrics.PersistenceGetReplicationTaskScope + case tasks.CategoryIDArchival: + operation = metrics.PersistenceGetArchivalTaskScope default: return nil, serviceerror.NewInternal(fmt.Sprintf("unknown task category type: %v", request.TaskCategory)) } @@ -357,6 +359,8 @@ func (p *executionPersistenceClient) GetHistoryTasks( operation = metrics.PersistenceGetVisibilityTasksScope case tasks.CategoryIDReplication: operation = metrics.PersistenceGetReplicationTasksScope + case tasks.CategoryIDArchival: + operation = metrics.PersistenceGetArchivalTasksScope default: return nil, serviceerror.NewInternal(fmt.Sprintf("unknown task category type: %v", request.TaskCategory)) }