1. import java.util.concurrent.ExecutorService;
  2. import java.util.concurrent.Executors;
  3. import java.util.concurrent.ScheduledExecutorService;
  4. import java.util.concurrent.TimeUnit;
  6. public class TaskScheduler {
  8. private final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1); // Single thread for simplicity
  10. /**
  11. * Schedules a task to run with a specified delay and period.
  12. *
  13. * @param task The task to execute. Should be a Runnable.
  14. * @param initialDelay The delay in seconds before the first execution.
  15. * @param period The time in seconds between subsequent executions.
  16. */
  17. public void scheduleTask(Runnable task, long initialDelay, long period) {
  18. scheduler.scheduleAtFixedRate(task, initialDelay, period, TimeUnit.SECONDS);
  19. }
  21. /**
  22. * Schedules a task to run after a specified delay, once.
  23. *
  24. * @param task The task to execute. Should be a Runnable.
  25. * @param delay The delay in seconds before the task is executed.
  26. */
  27. public void scheduleTaskOnce(Runnable task, long delay) {
  28. scheduler.schedule(task, delay, TimeUnit.SECONDS);
  29. }
  31. /**
  32. * Shuts down the scheduler gracefully.
  33. */
  34. public void shutdown() {
  35. scheduler.shutdown();
  36. try {
  37. if (!scheduler.awaitTermination(60, TimeUnit.SECONDS)) {
  38. scheduler.shutdownNow();
  39. }
  40. } catch (InterruptedException e) {
  41. scheduler.shutdownNow();
  42. Thread.currentThread().interrupt();
  43. }
  44. }
  46. public static void main(String[] args) throws InterruptedException {
  47. TaskScheduler scheduler = new TaskScheduler();
  49. // Example task that might fail
  50. Runnable myTask = () -> {
  51. try {
  52. //Simulate a potential error
  53. if (Math.random() < 0.2) {
  54. throw new RuntimeException("Simulated error!");
  55. }
  56. System.out.println("Task executed at: " + System.currentTimeMillis());
  57. } catch (Exception e) {
  58. System.err.println("Error executing task: " + e.getMessage()); // Simple error message
  59. }
  60. };
  63. // Schedule a task to run every 5 seconds, starting after 2 seconds
  64. scheduler.scheduleTask(myTask, 2, 5);
  66. // Schedule a task to run once after 10 seconds
  67. scheduler.scheduleTaskOnce(() -> System.out.println("One time task executed"), 10);
  69. Thread.sleep(30000); // Run for 30 seconds
  70. scheduler.shutdown();
  71. System.out.println("Scheduler shutdown.");
  72. }
  73. }

Add your comment