1. import java.util.HashMap;
  2. import java.util.Map;
  3. import java.util.concurrent.CompletableFuture;
  4. import java.util.concurrent.ExecutorService;
  5. import java.util.concurrent.Executors;
  6. import java.util.concurrent.TimeUnit;
  8. public class TokenMetricsCollector {
  10. private final ExecutorService executor = Executors.newFixedThreadPool(5); // Adjust thread pool size as needed
  11. private final Map<String, Long> tokenMetrics = new HashMap<>(); // Token -> Metrics (e.g., validation attempts, success/failure)
  13. public CompletableFuture<Void> collectTokenMetrics(String token, String validationEndpoint, int maxRetries) {
  14. return CompletableFuture.supplyAsync(() -> {
  15. long retries = 0;
  16. boolean success = false;
  18. while (retries < maxRetries) {
  19. try {
  20. // Simulate token validation call
  21. boolean isValid = validateToken(token, validationEndpoint);
  23. if (isValid) {
  24. success = true;
  25. break; // Token is valid, exit the retry loop
  26. } else {
  27. retries++;
  28. System.out.println("Validation failed, retry: " + retries);
  29. //Introduce a delay between retries
  30. try {
  31. TimeUnit.SECONDS.sleep(1);
  32. } catch (InterruptedException e) {
  33. Thread.currentThread().interrupt();
  34. return null;
  35. }
  36. }
  37. } catch (Exception e) {
  38. retries++;
  39. System.err.println("Error during validation, retry: " + retries + ". Error: " + e.getMessage());
  40. try {
  41. TimeUnit.SECONDS.sleep(2); // Longer delay on error
  42. } catch (InterruptedException ie) {
  43. Thread.currentThread().interrupt();
  44. return null;
  45. }
  46. }
  47. }
  49. if (success) {
  50. recordSuccess(token);
  51. return true;
  52. } else {
  53. recordFailure(token, retries);
  54. return false;
  55. }
  56. });
  57. }
  59. private boolean validateToken(String token, String endpoint) {
  60. // Replace with your actual token validation logic
  61. // This is a placeholder for a network call or other validation process
  62. // Simulate validation success or failure based on the token.
  63. if (token.equals("validToken")) {
  64. return true;
  65. } else {
  66. return false;
  67. }
  68. }
  71. private void recordSuccess(String token) {
  72. tokenMetrics.put(token, tokenMetrics.getOrDefault(token, 0L) + 1);
  73. System.out.println("Token " + token + " validated successfully.");
  74. }
  76. private void recordFailure(String token, int retries) {
  77. tokenMetrics.put(token, tokenMetrics.getOrDefault(token, 0L) + retries);
  78. System.out.println("Token " + token + " failed validation after " + retries + " retries.");
  79. }
  81. public Map<String, Long> getTokenMetrics() {
  82. return tokenMetrics;
  83. }
  85. public void shutdown() {
  86. executor.shutdown();
  87. try {
  88. executor.awaitTermination(5, TimeUnit.SECONDS); //Give tasks a chance to finish
  89. } catch (InterruptedException e) {
  90. Thread.currentThread().interrupt();
  91. }
  92. }
  94. public static void main(String[] args) throws Exception {
  95. TokenMetricsCollector collector = new TokenMetricsCollector();
  96. String token1 = "validToken";
  97. String token2 = "invalidToken";
  98. String token3 = "invalidToken";
  100. CompletableFuture<Void> future1 = collector.collectTokenMetrics(token1, "/validate", 3);
  101. CompletableFuture<Void> future2 = collector.collectTokenMetrics(token2, "/validate", 2);
  102. CompletableFuture<Void> future3 = collector.collectTokenMetrics(token3, "/validate", 5);
  105. future1.join();
  106. future2.join();
  107. future3.join();
  109. collector.shutdown();
  111. System.out.println("Final Token Metrics: " + collector.getTokenMetrics());
  112. }
  113. }

Add your comment