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..f529e06 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,7 @@ 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 +36,15 @@ 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())); + // Remove job after job has been processed + // Reference: https://www.w3schools.com/java/java_files_delete.asp + // Github username: apyw + File jobPath = entry.toFile(); + if (jobPath.delete()) { + logger.info(String.format("Deleted the job: %s\n", jobPath.getName())); + } else { + logger.info(String.format("Failed to delete the job: %s\n", jobPath.getName())); + } } } catch (IOException e) { e.printStackTrace();