diff --git a/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/enrichers/backend/AbstractBackendEntityEnricher.java b/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/enrichers/backend/AbstractBackendEntityEnricher.java index 386f0cd80..be630e10d 100644 --- a/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/enrichers/backend/AbstractBackendEntityEnricher.java +++ b/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/enrichers/backend/AbstractBackendEntityEnricher.java @@ -309,6 +309,11 @@ protected Map getIdentifyingAttributes( return Collections.unmodifiableMap(identifyingAttributes); } + protected Map getEnrichedEntityAttributes( + StructuredTrace trace, Event event, BackendType type, String backendURI) { + return Collections.emptyMap(); + } + @VisibleForTesting public Optional resolve( Event event, StructuredTrace trace, StructuredTraceGraph structuredTraceGraph) { @@ -337,6 +342,9 @@ public Optional resolve( final Builder entityBuilder = getEntityBuilder(trace, event, type, backendUri); backendProvider.getEntityAttributes(event).forEach(entityBuilder::putAttributes); + Map entityAttributes = + getEnrichedEntityAttributes(trace, event, type, backendUri); + entityBuilder.putAllAttributes(entityAttributes); Map enrichedAttributes = new HashMap<>(); Optional backendOperation = backendProvider.getBackendOperation(event);