From b69ec0313e2a8876fe3251093257ec71ddec4421 Mon Sep 17 00:00:00 2001
From: johnniang <johnniang@riseup.net>
Date: Sat, 2 Mar 2019 17:26:39 +0800
Subject: [PATCH] Refactor post sync scheduler

---
 src/main/java/cc/ryanc/halo/Application.java         |  2 ++
 .../java/cc/ryanc/halo/listener/StartedListener.java |  4 ++--
 src/main/java/cc/ryanc/halo/task/PostSyncTask.java   | 12 ++++++++++--
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/main/java/cc/ryanc/halo/Application.java b/src/main/java/cc/ryanc/halo/Application.java
index 691f25df1..7c0805a08 100755
--- a/src/main/java/cc/ryanc/halo/Application.java
+++ b/src/main/java/cc/ryanc/halo/Application.java
@@ -8,6 +8,7 @@ import org.springframework.cache.annotation.EnableCaching;
 import org.springframework.context.ApplicationContext;
 import org.springframework.data.jpa.repository.config.EnableJpaAuditing;
 import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
+import org.springframework.scheduling.annotation.EnableScheduling;
 
 /**
  * <pre>
@@ -21,6 +22,7 @@ import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
 @SpringBootApplication
 @EnableCaching
 @EnableJpaAuditing
+@EnableScheduling
 @EnableJpaRepositories(basePackages = "cc.ryanc.halo.repository", repositoryBaseClass = BaseRepositoryImpl.class)
 public class Application {
     public static void main(String[] args) {
diff --git a/src/main/java/cc/ryanc/halo/listener/StartedListener.java b/src/main/java/cc/ryanc/halo/listener/StartedListener.java
index 0a0507bb3..8b5a5604e 100644
--- a/src/main/java/cc/ryanc/halo/listener/StartedListener.java
+++ b/src/main/java/cc/ryanc/halo/listener/StartedListener.java
@@ -50,8 +50,8 @@ public class StartedListener implements ApplicationListener<ApplicationStartedEv
         this.loadThemes();
         this.loadOwo();
         //启动定时任务
-        CronUtil.start();
-        log.info("The scheduled task starts successfully!");
+//        CronUtil.start();
+//        log.info("The scheduled task starts successfully!");
     }
 
     @PreDestroy
diff --git a/src/main/java/cc/ryanc/halo/task/PostSyncTask.java b/src/main/java/cc/ryanc/halo/task/PostSyncTask.java
index ac6e52cc9..f6632c689 100644
--- a/src/main/java/cc/ryanc/halo/task/PostSyncTask.java
+++ b/src/main/java/cc/ryanc/halo/task/PostSyncTask.java
@@ -2,8 +2,9 @@ package cc.ryanc.halo.task;
 
 import cc.ryanc.halo.model.domain.Post;
 import cc.ryanc.halo.service.PostService;
-import cc.ryanc.halo.utils.SpringUtil;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.scheduling.annotation.Scheduled;
+import org.springframework.stereotype.Component;
 
 import static cc.ryanc.halo.model.dto.HaloConst.POSTS_VIEWS;
 
@@ -12,13 +13,20 @@ import static cc.ryanc.halo.model.dto.HaloConst.POSTS_VIEWS;
  * @date : 2018/12/5
  */
 @Slf4j
+@Component
 public class PostSyncTask {
 
+    private final PostService postService;
+
+    public PostSyncTask(PostService postService) {
+        this.postService = postService;
+    }
+
     /**
      * 将缓存的图文浏览数写入数据库
      */
+    @Scheduled(cron = "0 0 * * * *")
     public void postSync() {
-        final PostService postService = SpringUtil.getBean(PostService.class);
         int count = 0;
         for (Long key : POSTS_VIEWS.keySet()) {
             Post post = postService.getByIdOfNullable(key);