diff --git a/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java b/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java index c699133..27c8a99 100644 --- a/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java +++ b/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java @@ -4,6 +4,8 @@ import java.io.InputStreamReader; import java.io.IOException; import java.io.FileNotFoundException; +import java.io.File; + import java.nio.file.DirectoryStream; import java.nio.file.Files; @@ -35,6 +37,22 @@ public static void main(String[] args) { for (Path entry : ds) { Job job = new Job(entry.toFile()); logger.info(String.format("Job %d yields %d\n", job.getInput(), job.processJob())); + // Delete the file since it has been processed + boolean isDeleted = job.deleteJob(); + if (isDeleted) { + logger.info(String.format("Job %d was successfully deleted", job.getInput())); + } else { + logger.info(String.format("Job %d was not possible to delete", job.getInput())); + } + } + + // Delete directory + File directory = dir.toFile(); + boolean isDeleted = directory.delete(); + if (isDeleted) { + logger.info(String.format("Directory %s was successfully deleted", dir.toString())); + } else { + logger.info(String.format("Directory %s was not possible to delete", dir.toString())); } } catch (IOException e) { e.printStackTrace();