1. import java.util.ArrayList;
  2. import java.util.List;
  3. import java.util.concurrent.ExecutorService;
  4. import java.util.concurrent.Executors;
  6. public class TokenBatchProcessor {
  8. private final AuthenticationTokenProcessor processor;
  9. private final int batchSize;
  10. private final ExecutorService executorService;
  12. public TokenBatchProcessor(AuthenticationTokenProcessor processor, int batchSize) {
  13. this.processor = processor;
  14. this.batchSize = batchSize;
  15. this.executorService = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
  16. }
  18. public List<String> processTokens(List<String> tokens) {
  19. List<String> results = new ArrayList<>();
  20. int tokensProcessed = 0;
  22. for (int i = 0; i < tokens.size(); i += batchSize) {
  23. List<String> batch = tokens.subList(i, Math.min(i + batchSize, tokens.size()));
  24. executorService.submit(() -> {
  25. try {
  26. List<String> batchResults = processor.processBatch(batch);
  27. results.addAll(batchResults);
  28. } catch (Exception e) {
  29. // Handle exceptions appropriately. Log, retry, etc.
  30. System.err.println("Error processing batch: " + e.getMessage());
  31. }
  32. });
  33. tokensProcessed += batchSize;
  34. }
  36. executorService.shutdown();
  37. try {
  38. if (!executorService.awaitTermination(60, java.util.concurrent.TimeUnit.SECONDS)) {
  39. System.err.println("ExecutorService did not terminate in time.");
  40. }
  41. } catch (InterruptedException e) {
  42. System.err.println("ExecutorService termination interrupted.");
  43. }
  45. return results;
  46. }
  48. public interface AuthenticationTokenProcessor {
  49. List<String> processBatch(List<String> tokens) throws Exception;
  50. }
  53. public static void main(String[] args) {
  54. // Example Usage (replace with your actual processor and tokens)
  56. AuthenticationTokenProcessor exampleProcessor = tokens -> {
  57. List<String> results = new ArrayList<>();
  58. for (String token : tokens) {
  59. // Simulate token processing
  60. try {
  61. Thread.sleep(100); // Simulate some work
  62. if (token.equals("invalid_token")) {
  63. throw new Exception("Invalid token encountered");
  64. }
  65. results.add("Result for " + token);
  66. } catch (Exception e) {
  67. results.add("Error processing " + token + ": " + e.getMessage());
  68. }
  69. }
  70. return results;
  71. };
  73. TokenBatchProcessor processor = new TokenBatchProcessor(exampleProcessor, 5);
  74. List<String> tokens = List.of("token1", "token2", "token3", "token4", "token5", "token6", "token7", "token8", "token9", "invalid_token");
  76. List<String> results = processor.processTokens(tokens);
  77. System.out.println(results);
  78. }
  79. }

Add your comment