From bbd2353d923f848f7495b262c7e00f9459335253 Mon Sep 17 00:00:00 2001
From: Michael Snowden <michaelosnowden@gmail.com>
Date: Mon, 7 Nov 2022 10:22:54 -0800
Subject: [PATCH] Add metrics scopes for the archival queue

---
 common/metrics/metric_defs.go                  | 4 ++++
 common/persistence/persistenceMetricClients.go | 4 ++++
 2 files changed, 8 insertions(+)

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))
 	}