1. import java.io.IOException;
  2. import java.nio.file.Files;
  3. import java.nio.file.Path;
  4. import java.nio.file.Paths;
  5. import java.util.concurrent.Executors;
  6. import java.util.concurrent.ScheduledExecutorService;
  7. import java.util.concurrent.TimeUnit;
  8. import java.util.concurrent.atomic.AtomicBoolean;
  10. public class CookieBootstrapper {
  12. private final Path scriptPath;
  13. private final int retryCount;
  14. private final long retryDelayMillis;
  15. private final AtomicBoolean bootstrapped = new AtomicBoolean(false);
  17. public CookieBootstrapper(Path scriptPath, int retryCount, long retryDelayMillis) {
  18. this.scriptPath = scriptPath;
  19. this.retryCount = retryCount;
  20. this.retryDelayMillis = retryDelayMillis;
  21. }
  23. public void bootstrap() throws IOException {
  24. for (int i = 0; i < retryCount; i++) {
  25. try {
  26. executeScript();
  27. bootstrapped.set(true);
  28. return; // Success, exit loop
  29. } catch (IOException e) {
  30. if (i == retryCount - 1) {
  31. throw e; // Re-throw on last retry
  32. }
  33. System.err.println("Bootstrap failed (attempt " + (i + 1) + "): " + e.getMessage());
  34. try {
  35. Thread.sleep(retryDelayMillis);
  36. } catch (InterruptedException e2) {
  37. Thread.currentThread().interrupt();
  38. throw new IOException("Interrupted during retry", e2);
  39. }
  40. }
  41. }
  42. }
  44. private void executeScript() throws IOException {
  45. // Read the script file
  46. String scriptContent = new String(Files.readAllBytes(scriptPath));
  48. // Execute the script (replace with your actual execution logic)
  49. System.out.println("Executing script: " + scriptPath);
  50. // Example: Run the script using a shell command. Important: sanitize input!
  51. // ProcessBuilder pb = new ProcessBuilder("bash", "-c", scriptContent);
  52. // Process p = pb.start();
  53. // p.waitFor();
  55. //Simulate success
  56. System.out.println("Script executed successfully.");
  57. }
  59. public boolean isBootstrapped() {
  60. return bootstrapped.get();
  61. }
  64. public static void main(String[] args) throws IOException {
  66. Path scriptPath = Paths.get("cookie_script.sh"); // Replace with your script path
  67. int retryCount = 3;
  68. long retryDelayMillis = 1000;
  70. // Create a dummy script for testing
  71. String scriptContent = "echo 'Running cookie bootstrap script' > /tmp/cookie_boot_test.txt";
  72. Files.write(scriptPath, scriptContent.getBytes());
  76. CookieBootstrapper bootstrapper = new CookieBootstrapper(scriptPath, retryCount, retryDelayMillis);
  78. try {
  79. bootstrapper.bootstrap();
  80. if (bootstrapper.isBootstrapped()) {
  81. System.out.println("Cookie bootstrap completed successfully.");
  82. } else {
  83. System.err.println("Cookie bootstrap failed after multiple retries.");
  84. }
  85. } catch (IOException e) {
  86. System.err.println("Bootstrap failed permanently: " + e.getMessage());
  87. } finally {
  88. // Clean up the dummy script
  89. try {
  90. Files.deleteIfExists(scriptPath);
  91. } catch (IOException e) {
  92. System.err.println("Error deleting dummy script: " + e.getMessage());
  93. }
  94. }
  95. }
  96. }

Add your comment