STAC-21255: include attributes processor

StackVista · May 10, 2024 · ba6fb3f · ba6fb3f
1 parent ce51add
commit ba6fb3f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/connector/stsservicegraphconnector/connector_test.go b/connector/stsservicegraphconnector/connector_test.go
@@ -289,7 +289,7 @@ func TestUpdateDurationMetrics(t *testing.T) {
 	}
 	for _, tc := range testCases {
 		t.Run(tc.caseStr, func(_ *testing.T) {
-			p.updateDurationMetrics(metricKey, tc.duration, tc.duration)
+			p.updateDurationMetrics(metricKey, tc.duration, tc.duration, 1)
 		})
 	}
 }

diff --git a/connector/stsservicegraphconnector/internal/store/store.go b/connector/stsservicegraphconnector/internal/store/store.go
@@ -149,10 +149,10 @@ func (s *Store) tryEvictHead() bool {
 		delete(s.m, headEdge.Key)
 		s.l.Remove(head)
 	} else {
-		// TODO: update weight of edge to compensate for expiration
-		// this keeps the metrics (in expectation) correct
-		headEdge.expiration = headEdge.expiration.Add(s.ttl)
+		headEdge.expiration = time.Now().Add(s.ttl)
 		headEdge.generation++
+		// update weight of edge to compensate for expiration
+		// this keeps the metrics (in expectation) correct
 		headEdge.logp += math.Log(1.0 - float64(s.l.Len()-1)/float64(s.maxItems))
 		s.l.MoveToBack(head)
 	}

diff --git a/sts-otel-builder.yaml b/sts-otel-builder.yaml
@@ -31,6 +31,7 @@ extensions:
 processors:
   - gomod: go.opentelemetry.io/collector/processor/memorylimiterprocessor v0.99.0
   - gomod: go.opentelemetry.io/collector/processor/batchprocessor v0.99.0
+  - gomod: github.com/open-telemetry/opentelemetry-collector-contrib/processor/attributesprocessor v0.99.0
   - gomod: github.com/open-telemetry/opentelemetry-collector-contrib/processor/resourceprocessor v0.99.0
   - gomod: github.com/open-telemetry/opentelemetry-collector-contrib/processor/k8sattributesprocessor v0.99.0