Skip to content

v0.2.54..v0.2.55 changeset RemoveFilesCommand.java

Garret Voltz edited this page Aug 14, 2020 · 1 revision
diff --git a/hoot-services/src/main/java/hoot/services/controllers/ingest/RemoveFilesCommand.java b/hoot-services/src/main/java/hoot/services/controllers/ingest/RemoveFilesCommand.java
new file mode 100644
index 0000000..546523b
--- /dev/null
+++ b/hoot-services/src/main/java/hoot/services/controllers/ingest/RemoveFilesCommand.java
@@ -0,0 +1,77 @@
+/*
+ * This file is part of Hootenanny.
+ *
+ * Hootenanny is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * --------------------------------------------------------------------
+ *
+ * The following copyright notices are generated automatically. If you
+ * have a new notice to add, please use the format:
+ * " * @copyright Copyright ..."
+ * This will properly maintain the copyright information. DigitalGlobe
+ * copyrights will be updated automatically.
+ *
+ * @copyright Copyright (C) 2020 DigitalGlobe (http://www.digitalglobe.com/)
+ */
+package hoot.services.controllers.ingest;
+
+import org.apache.commons.io.FileUtils;
+
+import java.io.File;
+import java.io.IOException;
+import java.time.LocalDateTime;
+import java.util.List;
+
+import hoot.services.command.CommandResult;
+import hoot.services.command.InternalCommand;
+
+
+public class RemoveFilesCommand implements InternalCommand {
+    private final String jobId;
+    private final List<File> deleteFiles;
+
+    RemoveFilesCommand(String jobId, List<File> deleteFiles) {
+        this.jobId = jobId;
+        this.deleteFiles = deleteFiles;
+    }
+
+    @Override
+    public CommandResult execute() {
+        CommandResult commandResult = new CommandResult();
+        commandResult.setJobId(jobId);
+        commandResult.setCommand("Delete list of files");
+        commandResult.setStart(LocalDateTime.now());
+
+        try {
+            for(File file : this.deleteFiles) {
+                if (file.isDirectory()) {
+                    FileUtils.deleteDirectory(file);
+                } else {
+                    file.delete();
+                }
+            }
+        } catch(IOException exc) {
+            String msg = "Failure to delete files: " + exc.getMessage();
+            throw new RuntimeException(msg, exc);
+        }
+
+
+        commandResult.setFinish(LocalDateTime.now());
+        commandResult.setExitCode(CommandResult.SUCCESS);
+
+        return commandResult;
+    }
+}
+
+
Clone this wiki locally