diff --git a/src/main/java/com/kms/App.java b/src/main/java/com/kms/App.java index 9d5c6e1a3276d18b143208a8621ec14ad6c08a35..e4adca08cc701e4f5228af684b2f2ab31900e193 100644 --- a/src/main/java/com/kms/App.java +++ b/src/main/java/com/kms/App.java @@ -11,6 +11,7 @@ import com.kms.model.Payment; import com.kms.service.PaymentHistoryService; import com.kms.service.PaymentService; import com.kms.task.PaymentCompleterTask; +import com.kms.task.PdfCleanerTask; import javax.xml.ws.Endpoint; import java.sql.SQLException; @@ -36,6 +37,10 @@ public class App { PaymentCompleterTask task = new PaymentCompleterTask(paymentDao); time.schedule(task, 0, 25000); + final Timer timePdf = new Timer(); + PdfCleanerTask pdfCleanerTask = new PdfCleanerTask(); + timePdf.schedule(pdfCleanerTask, 0, 25000); + Endpoint.publish("http://0.0.0.0:8080/paymentservice", new PaymentService(paymentDao)); Endpoint.publish("http://0.0.0.0:8080/paymenthistory", new PaymentHistoryService(paymentDao)); } catch (SQLException e) { diff --git a/src/main/java/com/kms/task/PdfCleanerTask.java b/src/main/java/com/kms/task/PdfCleanerTask.java index 6e4833405abff7fc346f4fa1912f189c7059d2e1..24ed986a3970c60d34ccf7066e0c86645b304f55 100644 --- a/src/main/java/com/kms/task/PdfCleanerTask.java +++ b/src/main/java/com/kms/task/PdfCleanerTask.java @@ -1,5 +1,6 @@ package com.kms.task; +import java.io.File; import java.util.TimerTask; public class PdfCleanerTask extends TimerTask { @@ -8,6 +9,17 @@ public class PdfCleanerTask extends TimerTask { @Override public void run() { + File dir = new File(PDF_FILE_LOCATION); + if (!dir.isDirectory() || dir.listFiles() == null) { + return; + } + try { + for(File file: dir.listFiles()) + if (!file.isDirectory()) + file.delete(); + } catch (Exception e) { + System.out.println("error while cleansing pdf. err: " + e.getMessage()); + } } }