1. import java.util.Collection;
  2. import java.util.HashSet;
  3. import java.util.Set;
  5. public class CollectionGuard {
  7. private final Set<String> guardedCollections = new HashSet<>();
  8. private boolean dryRunMode = false;
  10. public CollectionGuard() {
  11. // Add collections to be guarded. Can be dynamically updated.
  12. }
  14. public void addCollection(String collectionName) {
  15. guardedCollections.add(collectionName);
  16. }
  18. public void setDryRunMode(boolean dryRunMode) {
  19. this.dryRunMode = dryRunMode;
  20. }
  22. public void executeTask(String collectionName, Runnable task) {
  23. if (dryRunMode) {
  24. System.out.println("Dry run: Would execute task on collection: " + collectionName);
  25. return;
  26. }
  28. if (!guardedCollections.contains(collectionName)) {
  29. System.out.println("Collection '" + collectionName + "' is not guarded. Skipping.");
  30. return;
  31. }
  33. try {
  34. task.run(); // Execute the task on the collection
  35. } catch (Exception e) {
  36. System.err.println("Error executing task on collection '" + collectionName + "': " + e.getMessage());
  37. e.printStackTrace();
  38. }
  39. }
  41. public static void main(String[] args) {
  42. CollectionGuard guard = new CollectionGuard();
  43. guard.addCollection("myCollection");
  45. // Example task
  46. Runnable myTask = () -> {
  47. System.out.println("Executing task on myCollection");
  48. };
  50. guard.setDryRunMode(false);
  51. guard.executeTask("myCollection", myTask);
  53. guard.setDryRunMode(true);
  54. guard.executeTask("myCollection", myTask);
  55. guard.executeTask("anotherCollection", myTask); // should skip
  57. }
  58. }

Add your comment