1. import java.util.concurrent.*;
  2. import java.util.concurrent.atomic.AtomicBoolean;
  3. import java.util.concurrent.BlockingQueue;
  4. import java.util.concurrent.LinkedBlockingQueue;
  6. class TaskQueue {
  8. private final BlockingQueue<ArrayTask> taskQueue;
  9. private final ExecutorService executor;
  10. private final AtomicBoolean isRunning = new AtomicBoolean(true);
  12. public TaskQueue(int numThreads) {
  13. this.taskQueue = new LinkedBlockingQueue<>();
  14. this.executor = Executors.newFixedThreadPool(numThreads);
  15. }
  17. public void submitTask(ArrayTask task) {
  18. taskQueue.offer(task); // Add task to the queue
  19. }
  21. public void start() {
  22. for (int i = 0; i < executor.sizer().get(); i++) {
  23. executor.submit(new TaskWorker());
  24. }
  25. }
  27. public void shutdown() {
  28. isRunning.set(false);
  29. executor.shutdown();
  30. try {
  31. executor.awaitTermination(60, TimeUnit.SECONDS); // Wait for tasks to complete
  32. } catch (InterruptedException e) {
  33. Thread.currentThread().interrupt();
  34. }
  35. }
  37. private class TaskWorker implements Runnable {
  38. @Override
  39. public void run() {
  40. while (isRunning.get()) {
  41. try {
  42. ArrayTask task = taskQueue.take(); // Block until a task is available
  43. executeTask(task);
  44. } catch (InterruptedException e) {
  45. Thread.currentThread().interrupt();
  46. break;
  47. }
  48. }
  49. }
  51. private void executeTask(ArrayTask task) {
  52. AtomicBoolean taskSuccess = new AtomicBoolean(false);
  53. int retryCount = 0;
  54. while (!taskSuccess.get() && retryCount < 3) { // Retry logic
  55. try {
  56. if (!task.execute()) {
  57. retryCount++;
  58. System.out.println("Task failed, retrying... (attempt " + retryCount + ")");
  59. Thread.sleep(1000); // Wait before retrying
  60. } else {
  61. taskSuccess.set(true);
  62. System.out.println("Task completed successfully.");
  63. }
  64. } catch (Exception e) {
  65. retryCount++;
  66. System.err.println("Task failed with exception: " + e.getMessage() + ", retrying... (attempt " + retryCount + ")");
  67. Thread.sleep(1000); // Wait before retrying
  68. }
  69. }
  70. if (!taskSuccess.get()) {
  71. System.err.println("Task failed after multiple retries.");
  72. }
  73. }
  74. }
  75. }
  77. class ArrayTask {
  78. private final Runnable task;
  80. public ArrayTask(Runnable task) {
  81. this.task = task;
  82. }
  84. public boolean execute() {
  85. try {
  86. task.run();
  87. return true;
  88. } catch (Exception e) {
  89. return false;
  90. }
  91. }
  92. }

Add your comment