Skip to content

Commit 77d9b63

Browse files
Ability to configure the write throttling via config (#246)
1 parent 5200907 commit 77d9b63

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

  • hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/clients

hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/main/java/org/hypertrace/traceenricher/enrichment/clients/DefaultClientRegistry.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@ public class DefaultClientRegistry implements ClientRegistry {
2727
private static final String CONFIG_SERVICE_PORT_KEY = "config.service.config.port";
2828
private static final String ENTITY_SERVICE_HOST_KEY = "entity.service.config.host";
2929
private static final String ENTITY_SERVICE_PORT_KEY = "entity.service.config.port";
30+
private static final String TRACE_ENTITY_WRITE_THROTTLE_DURATION =
31+
"trace.entity.write.throttle.duration";
3032

3133
private final ManagedChannel attributeServiceChannel;
3234
private final ManagedChannel configServiceChannel;
@@ -67,6 +69,10 @@ public DefaultClientRegistry(Config config) {
6769
EntityTypeClient.builder(this.entityServiceChannel).build(),
6870
EntityDataClient.builder(this.entityServiceChannel).build(),
6971
this.cachingAttributeClient)
72+
.withEntityWriteThrottleDuration(
73+
config.hasPath(TRACE_ENTITY_WRITE_THROTTLE_DURATION)
74+
? config.getDuration(TRACE_ENTITY_WRITE_THROTTLE_DURATION)
75+
: Duration.ofSeconds(15))
7076
.build();
7177
}
7278

0 commit comments

Comments
 (0)