Add metrics scopes for the archival queue

temporalio · Dec 1, 2022 · bbd2353 · bbd2353
1 parent 79a572d
commit bbd2353
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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))
 	}