diff --git a/core/src/main/java/io/opentelemetry/android/export/RateLimitedExporter.java b/core/src/main/java/io/opentelemetry/android/export/RateLimitedExporter.java index b4e5372a3..bbbc4cea3 100644 --- a/core/src/main/java/io/opentelemetry/android/export/RateLimitedExporter.java +++ b/core/src/main/java/io/opentelemetry/android/export/RateLimitedExporter.java @@ -59,7 +59,7 @@ public CompletableResultCode export(Collection spans) { } private boolean canExport(long spanSize) { - long currentTime = System.currentTimeMillis(); + long currentTime = SystemTime.get().getCurrentTimeMillis(); if (currentTime - lastExportTime > timeWindowInMillis) { // Reset the window bytesExportedInWindow = 0; @@ -87,26 +87,26 @@ public CompletableResultCode shutdown() { static class Builder { final SpanExporter delegate; - CategoryFunction categoryFunction = span -> "default"; - long maxBytesPerSecond = 1024; // Default to 1 KB/s - long timeWindowInMillis = 1000; // Default to 1 second - + private CategoryFunction categoryFunction = span -> "default"; + private long maxBytesPerSecond = 1024; // Default to 1 KB/s + private long timeWindowInMillis = 1000; // Default to 1 second + private Builder(SpanExporter delegate) { this.delegate = delegate; } - + Builder maxBytesPerSecond(long maxBytesPerSecond) { this.maxBytesPerSecond = maxBytesPerSecond; return this; } - + Builder timeWindow(Duration timeWindow) { this.timeWindow = timeWindow; return this; } - + BandwidthThrottlingExporter build() { return new BandwidthThrottlingExporter(this); } - } + } }