1. import com.isyscore.rpc.file.config.DownloadTaskConfig;
    2. import com.isyscore.rpc.file.config.UploadTaskConfig;
    3. import com.isyscore.rpc.file.util.FileUtil;
    4. import com.isyscore.rpc.file.util.TimeUtils;
    6. import java.io.File;
    7. import java.util.concurrent.ScheduledThreadPoolExecutor;
    8. import java.util.concurrent.TimeUnit;
    10. import static com.isyscore.rpc.file.FileDownloadHandler.FILE_DOWNLOAD_TASK_PATH;
    11. import static com.isyscore.rpc.file.FileUploadHandler.FILE_UPLOAD_TASK_PATH;
    13. /**
    14. * @author shizi
    15. * @since 2021-08-25 18:19:49
    16. */
    17. public class DeamonTask {
    19.     /**
    20.     * 守护线程，用于清理
    21.     */
    22.     private final ScheduledThreadPoolExecutor scheduler = new ScheduledThreadPoolExecutor(1, r -> {
    23.         Thread thread = new Thread(r, "Thread-TaskCacheConfig-Deamon");
    24.         thread.setDaemon(true);
    25.         return thread;
    26.     });
    28.     private volatile boolean started = false;
    30.     private static final DeamonTask task = new DeamonTask();
    32.     public static DeamonTask getInstance() {
    33.         return task;
    34.     }
    36.     private DeamonTask(){}
    38.     public void start() {
    39.         if (started) {
    40.             return;
    41.         }
    42.         synchronized (DeamonTask.class) {
    43.             if (!started) {
    44.                 initDeamon();
    45.                 started = true;
    46.             }
    47.         }
    48.     }
    50.     private void cleanUploadTask() {
    51.         if (!FileUtil.exist(FILE_UPLOAD_TASK_PATH)) {
    52.             return;
    53.         }
    54.         File taskConfigFile = FileUtil.getDirectFile(FILE_UPLOAD_TASK_PATH);
    55.         File[] files = taskConfigFile.listFiles();
    56.         if (null == files) {
    57.             return;
    58.         }
    59.         for (File file : files) {
    60.             UploadTaskConfig taskConfig = FileUtil.readObject(UploadTaskConfig.class, file);
    61.             if (null == taskConfig) {
    62.                 continue;
    63.             }
    64.             Long lastTime = taskConfig.getLastUpdateTime();
    65.             if (null == lastTime) {
    66.                 continue;
    67.             }
    68.             // 如果超过5个小时，则任务的进度文件删除掉
    69.             if (TimeUtils.plusHours(lastTime, 5) < System.currentTimeMillis()) {
    70.                 FileUtil.delete(file);
    71.             }
    72.         }
    73.     }
    75.     private void cleanDownTask() {
    76.         if (!FileUtil.exist(FILE_DOWNLOAD_TASK_PATH)) {
    77.             return;
    78.         }
    79.         File taskConfigFile = FileUtil.getDirectFile(FILE_DOWNLOAD_TASK_PATH);
    80.         File[] files = taskConfigFile.listFiles();
    81.         if (null == files) {
    82.             return;
    83.         }
    84.         for (File file : files) {
    85.             DownloadTaskConfig taskConfig = FileUtil.readObject(DownloadTaskConfig.class, file);
    86.             if (null == taskConfig) {
    87.                 continue;
    88.             }
    89.             Long lastTime = taskConfig.getLastUpdateTime();
    90.             if (null == lastTime) {
    91.                 continue;
    92.             }
    93.             // 如果超过5个小时，则任务的进度文件删除掉
    94.             if (TimeUtils.plusHours(lastTime, 5) < System.currentTimeMillis()) {
    95.                 FileUtil.delete(file);
    96.             }
    97.         }
    98.     }
    100.     private void initDeamon() {
    101.         scheduler.scheduleWithFixedDelay(() -> {
    102.             cleanUploadTask();
    103.             cleanDownTask();
    104.         }, 1, 30, TimeUnit.MINUTES);
    105.     }
    106. }