TRANSFER-779: Missed metering agent init

doublecloud · Aug 13, 2024 · 4cec91d · 4cec91d
1 parent a295662
commit 4cec91d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/transfer_manager/go/pkg/metering/initializer_os.go b/transfer_manager/go/pkg/metering/initializer_os.go
@@ -18,6 +18,13 @@ func InitializeWithTags(transfer *server.Transfer, task *server.TransferOperatio
 	return
 }
 
+func WithAgent(agent MeteringAgent) MeteringAgent {
+	commonAgentMu.Lock()
+	defer commonAgentMu.Unlock()
+	commonAgent = agent
+	return commonAgent
+}
+
 func Initialize(transfer *server.Transfer, task *server.TransferOperation) {
 	InitializeWithTags(transfer, task, map[string]interface{}{})
 }