Added failture count

Author: alex268

jdbc/ydb-token-app/pom.xml

@@ -87,6 +87,5 @@
                 <version>2.0.12</version>
             </dependency>
         </dependencies>
-
     </dependencyManagement>
 </project>

jdbc/ydb-token-app/src/main/java/tech/ydb/apps/AppMetrics.java

@@ -147,6 +147,7 @@ private void printTotal(Logger logger) {
     private final Method batchUpdate;
 
     private final AtomicInteger executionsCount = new AtomicInteger(0);
+    private final AtomicInteger failturesCount = new AtomicInteger(0);
     private final AtomicInteger retriesCount = new AtomicInteger(0);
 
     private final ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor(
@@ -177,6 +178,10 @@ public Method getBatchUpdate() {
         return this.batchUpdate;
     }
 
+    public void incrementFaiture() {
+        failturesCount.incrementAndGet();
+    }
+
     public void runWithMonitor(Runnable runnable) {
         Arrays.asList(load, fetch, update, batchUpdate).forEach(Method::reset);
         final ScheduledFuture<?> future = scheduler.scheduleAtFixedRate(this::print, 1, 10, TimeUnit.SECONDS);
@@ -197,9 +202,16 @@ private void print() {
     }
 
     public void printTotal() {
-        logger.info("=========== TOTAL ==============");
-        Arrays.asList(load, fetch, update, batchUpdate).forEach(m -> m.printTotal(logger));
-        logger.info("Executed {} transactions with {} retries", executionsCount.get(), retriesCount.get());
+        if (failturesCount.get() > 0) {
+            logger.error("=========== TOTAL ==============");
+            Arrays.asList(load, fetch, update, batchUpdate).forEach(m -> m.printTotal(logger));
+            logger.error("Executed {} transactions with {} retries and {} failtures", executionsCount.get(),
+                    retriesCount.get() - failturesCount.get(), failturesCount.get());
+        } else {
+            logger.info("=========== TOTAL ==============");
+            Arrays.asList(load, fetch, update, batchUpdate).forEach(m -> m.printTotal(logger));
+            logger.info("Executed {} transactions with {} retries", executionsCount.get(), retriesCount.get());
+        }
     }
 
     public RetryListener getRetryListener() {
@@ -218,7 +230,6 @@ public <T, E extends Throwable> void onError(RetryContext ctx, RetryCallback<T,
                 if (m != null) {
                     m.retriesCounter.apply(extractStatusCode(th)).increment();
                 }
-
             }
 
             @Override

jdbc/ydb-token-app/src/main/java/tech/ydb/apps/Application.java

@@ -197,6 +197,7 @@ private void workload(RateLimiter rt, long finishAt) {
                     executeBatchUpdate(rnd, recordCount); // 10 percents
                 }
             } catch (RuntimeException ex) {
+                ticker.incrementFaiture();
                 logger.debug("got exception {}", ex.getMessage());
             }
         }

jdbc/ydb-token-app/src/main/java/tech/ydb/apps/Config.java

@@ -1,5 +1,7 @@
 package tech.ydb.apps;
 
+
+import io.micrometer.core.instrument.MeterRegistry;
 import org.springframework.boot.context.properties.ConfigurationProperties;
 import org.springframework.boot.context.properties.ConstructorBinding;
 import org.springframework.boot.context.properties.bind.Name;
@@ -20,7 +22,7 @@ public class Config {
     private final int rpsLimit;
 
     public Config(String connection, int threadsCount, int recordsCount, @Name("load.batchSize") int loadBatchSize,
-            @Name("workload.duration") int workloadDuration, int rpsLimit) {
+            @Name("workload.duration") int workloadDuration, int rpsLimit, MeterRegistry registy) {
         this.connection = connection;
         this.threadsCount = threadsCount <= 0 ? Runtime.getRuntime().availableProcessors() : threadsCount;
         this.recordsCount = recordsCount;
@@ -52,5 +54,4 @@ public int getWorkloadDurationSec() {
     public RateLimiter getRpsLimiter() {
         return rpsLimit <= 0 ? RateLimiter.noLimit() : RateLimiter.withRps(rpsLimit);
     }
-
 }

jdbc/ydb-token-app/src/main/java/tech/ydb/apps/annotation/YdbRetryable.java

@@ -19,7 +19,7 @@
 @Retention(RetentionPolicy.RUNTIME)
 @Retryable(
         retryFor = { SQLRecoverableException.class, SQLTransientException.class },
-        maxAttempts = 5,
+        maxAttempts = 15,
         backoff = @Backoff(delay = 100, multiplier = 2.0, maxDelay = 5000, random = true)
 )
 public @interface YdbRetryable {

jdbc/ydb-token-app/src/main/java/tech/ydb/apps/service/WorkloadService.java

@@ -7,6 +7,8 @@
 import java.util.UUID;
 import java.util.stream.Collectors;
 
+import javax.persistence.EntityNotFoundException;
+
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Service;
@@ -54,7 +56,7 @@ public Token fetchToken(int id) {
 
         if (!token.isPresent()) {
             logger.warn("token {} is not found", id);
-            return null;
+            throw new EntityNotFoundException("token " + id + " is not found");
         }
 
         return token.get();
@@ -71,6 +73,7 @@ public void updateToken(int id, long counter) {
             logger.trace("updated token {} -> {}", id, token.getVersion());
         } else {
             logger.warn("token {} is not found", id);
+            throw new EntityNotFoundException("token " + id + " is not found");
         }
     }
 

jdbc/ydb-token-app/src/main/resources/application.properties

@@ -26,6 +26,8 @@ management.metrics.enable.jdbc=true
 management.metrics.enable.sdk=true
 management.metrics.enable.grpc=true
 management.metrics.distribution.percentiles-histogram.all=true
+management.metrics.distribution.sla.all=true
+
 management.metrics.export.prometheus.pushgateway.enabled=${app.pushMetrics}
 management.metrics.export.prometheus.pushgateway.push-rate=15s
 management.metrics.export.prometheus.pushgateway.base-url=${app.prometheusUrl}