remove agent type receiver from trace agent obfuscateStatsGroup

DataDog · Dec 11, 2024 · df12504 · df12504
1 parent 3c882e7
commit df12504
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/pkg/trace/agent/agent.go b/pkg/trace/agent/agent.go
@@ -535,7 +535,8 @@ func (a *Agent) processStats(in *pb.ClientStatsPayload, lang, tracerVersion, con
 			if !a.Blacklister.AllowsStat(b) {
 				continue
 			}
-			a.obfuscateStatsGroup(b)
+			obfuscator := a.lazyInitObfuscator()
+			obfuscateStatsGroup(obfuscator, b)
 			a.Replacer.ReplaceStatsGroup(b)
 			group.Stats[n] = b
 			n++

diff --git a/pkg/trace/agent/obfuscate.go b/pkg/trace/agent/obfuscate.go
@@ -110,9 +110,7 @@ func (a *Agent) obfuscateSpan(span *pb.Span) {
 	}
 }
 
-func (a *Agent) obfuscateStatsGroup(b *pb.ClientGroupedStats) {
-	o := a.lazyInitObfuscator()
-
+func obfuscateStatsGroup(o *obfuscate.Obfuscator, b *pb.ClientGroupedStats) {
 	switch b.Type {
 	case "sql", "cassandra":
 		oq, err := o.ObfuscateSQLString(b.Resource)

diff --git a/pkg/trace/agent/obfuscate_test.go b/pkg/trace/agent/obfuscate_test.go
@@ -48,7 +48,8 @@ func TestObfuscateStatsGroup(t *testing.T) {
 	} {
 		agnt, stop := agentWithDefaults()
 		defer stop()
-		agnt.obfuscateStatsGroup(tt.in)
+		obfuscator := agnt.lazyInitObfuscator()
+		obfuscateStatsGroup(obfuscator, tt.in)
 		assert.Equal(t, tt.in.Resource, tt.out)
 	}
 }