Use AtomicHistogram instead of syncronized

Author: alex268

slo/src/main/java/tech/ydb/slo/Config.java

@@ -15,19 +15,22 @@
  */
 public final class Config {
     private final String connectionString;
+    private final String token;
     private final String ref;
     private final String workloadName;
     private final int durationSeconds;
     private final String otlpEndpoint;
 
     private Config(
             String connectionString,
+            String token,
             String ref,
             String workloadName,
             int durationSeconds,
             String otlpEndpoint
     ) {
         this.connectionString = connectionString;
+        this.token = token;
         this.ref = ref;
         this.workloadName = workloadName;
         this.durationSeconds = durationSeconds;
@@ -38,6 +41,10 @@ public String connectionString() {
         return connectionString;
     }
 
+    public String token() {
+        return token;
+    }
+
     public String ref() {
         return ref;
     }
@@ -57,6 +64,7 @@ public String otlpEndpoint() {
     /**
      * Loads configuration from environment variables.
      *
+     * @return configuration instance
      * @throws IllegalStateException if required variables are missing or invalid
      */
     public static Config fromEnv() {
@@ -67,12 +75,13 @@ public static Config fromEnv() {
             );
         }
 
+        String token = envOrDefault("YDB_TOKEN", "");
         String ref = envOrDefault("WORKLOAD_REF", "unknown");
         String workloadName = envOrDefault("WORKLOAD_NAME", "java-slo-workload");
         int durationSeconds = parseInt(envOrDefault("WORKLOAD_DURATION", "600"), 600);
         String otlpEndpoint = envOrDefault("OTEL_EXPORTER_OTLP_ENDPOINT", "");
 
-        return new Config(connectionString, ref, workloadName, durationSeconds, otlpEndpoint);
+        return new Config(connectionString, token, ref, workloadName, durationSeconds, otlpEndpoint);
     }
 
     private static String resolveConnectionString() {

slo/src/main/java/tech/ydb/slo/Main.java

@@ -5,6 +5,9 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import tech.ydb.auth.AuthProvider;
+import tech.ydb.auth.NopAuthProvider;
+import tech.ydb.auth.TokenAuthProvider;
 import tech.ydb.core.grpc.GrpcTransport;
 import tech.ydb.query.QueryClient;
 import tech.ydb.slo.kv.KvWorkload;
@@ -78,7 +81,12 @@ public static void main(String[] args) {
 
         int exitCode = 0;
         Metrics metrics = Metrics.create(config);
+        AuthProvider provider = NopAuthProvider.INSTANCE;
+        if (config.token() != null && !config.token().isEmpty()) {
+            provider = new TokenAuthProvider(config.token());
+        }
         GrpcTransport transport = GrpcTransport.forConnectionString(config.connectionString())
+                .withAuthProvider(provider)
                 .build();
         QueryClient queryClient = QueryClient.newClient(transport).build();
 

slo/src/main/java/tech/ydb/slo/Metrics.java

@@ -16,6 +16,7 @@
 import io.opentelemetry.sdk.metrics.SdkMeterProviderBuilder;
 import io.opentelemetry.sdk.metrics.export.PeriodicMetricReader;
 import io.opentelemetry.sdk.resources.Resource;
+import org.HdrHistogram.AtomicHistogram;
 import org.HdrHistogram.Histogram;
 
 /**
@@ -112,7 +113,7 @@ private Metrics(
         this.pendingOperations = pendingOperations;
     }
 
-    /**
+    /*
      * Builds a {@code Metrics} instance configured to push OTLP metrics every
      * second to the endpoint from {@code config.otlpEndpoint()}. If the
      * endpoint is empty, all metrics are still observable in-process but never
@@ -232,7 +233,7 @@ private static String metricsEndpoint(String otlpEndpoint) {
         return trimmed + "/v1/metrics";
     }
 
-    /**
+    /*
      * Records a started operation and returns a span used to record the
      * outcome.
      */
@@ -286,9 +287,7 @@ private void recordOutcome(
         if (status == OperationStatus.SUCCESS) {
             Histogram histogram = histograms.computeIfAbsent(type, k -> newHistogram());
             long clamped = Math.max(HDR_MIN_MICROS, Math.min(HDR_MAX_MICROS, latencyMicros));
-            synchronized (histogram) {
-                histogram.recordValue(clamped);
-            }
+            histogram.recordValue(clamped);
         } else {
             errorsTotal.add(1, Attributes.of(
                     ATTR_REF, ref,
@@ -319,15 +318,13 @@ private static void observeAndResetPercentiles(
             long p50Micros;
             long p95Micros;
             long p99Micros;
-            synchronized (histogram) {
-                if (histogram.getTotalCount() == 0) {
-                    continue;
-                }
-                p50Micros = histogram.getValueAtPercentile(50.0);
-                p95Micros = histogram.getValueAtPercentile(95.0);
-                p99Micros = histogram.getValueAtPercentile(99.0);
-                histogram.reset();
+            if (histogram.getTotalCount() == 0) {
+                continue;
             }
+            p50Micros = histogram.getValueAtPercentile(50.0);
+            p95Micros = histogram.getValueAtPercentile(95.0);
+            p99Micros = histogram.getValueAtPercentile(99.0);
+            histogram.reset();
 
             // Percentile gauges are always tagged with operation_status="success"
             // because we only record successful samples (see recordOutcome).
@@ -345,7 +342,7 @@ private static void observeAndResetPercentiles(
     }
 
     private static Histogram newHistogram() {
-        return new Histogram(HDR_MIN_MICROS, HDR_MAX_MICROS, HDR_SIGNIFICANT_DIGITS);
+        return new AtomicHistogram(HDR_MIN_MICROS, HDR_MAX_MICROS, HDR_SIGNIFICANT_DIGITS);
     }
 
     /**