From 2174fb96efa899b9435569d9f6a2734dabbdce34 Mon Sep 17 00:00:00 2001
From: LamGC <lam827@lamgc.net>
Date: Tue, 31 Mar 2020 13:02:14 +0800
Subject: [PATCH] =?UTF-8?q?[Add]=20=E5=A2=9E=E5=8A=A0=E6=8E=92=E8=A1=8C?=
 =?UTF-8?q?=E6=A6=9C=E7=BC=93=E5=AD=98=E6=9B=B4=E6=96=B0=E7=9A=84=E5=AE=9A?=
 =?UTF-8?q?=E6=97=B6=E4=BB=BB=E5=8A=A1;?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/net/lamgc/cgj/CQConfig.java     |  2 +
 src/main/java/net/lamgc/cgj/CQProcess.java    | 15 +++-
 .../net/lamgc/cgj/RankingUpdateTimer.java     | 75 +++++++++++++++++++
 3 files changed, 91 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/net/lamgc/cgj/RankingUpdateTimer.java

diff --git a/src/main/java/net/lamgc/cgj/CQConfig.java b/src/main/java/net/lamgc/cgj/CQConfig.java
index 801fda4..b0b31e6 100644
--- a/src/main/java/net/lamgc/cgj/CQConfig.java
+++ b/src/main/java/net/lamgc/cgj/CQConfig.java
@@ -22,6 +22,8 @@ public class CQConfig {
                         .setNameFormat("Plugin-ProcessThread-%d")
                         .build()
         );
+
+        CQProcess.initialize();
     }
 
 }
diff --git a/src/main/java/net/lamgc/cgj/CQProcess.java b/src/main/java/net/lamgc/cgj/CQProcess.java
index ea881dc..57c4b30 100644
--- a/src/main/java/net/lamgc/cgj/CQProcess.java
+++ b/src/main/java/net/lamgc/cgj/CQProcess.java
@@ -74,12 +74,25 @@ public class CQProcess {
             new ThreadPoolExecutor.DiscardOldestPolicy()
     ));
 
-    static {
+    private final static RankingUpdateTimer updateTimer = new RankingUpdateTimer();
+
+    public static void initialize() {
+        log.info("正在初始化...");
         try {
             imageCacheExecutor.addHandler(new ImageCacheHandler());
         } catch (IllegalAccessException e) {
             e.printStackTrace();
         }
+        updateTimer.schedule(null);
+        log.info("初始化完成.");
+    }
+
+    @Command
+    public static String runUpdateTimer(@Argument(force = false, name = "date") Date queryTime) {
+        log.info("正在手动触发排行榜更新任务...");
+        updateTimer.now(queryTime);
+        log.info("任务执行结束.");
+        return "操作已完成.";
     }
 
     @Command(defaultCommand = true)
diff --git a/src/main/java/net/lamgc/cgj/RankingUpdateTimer.java b/src/main/java/net/lamgc/cgj/RankingUpdateTimer.java
new file mode 100644
index 0000000..beac501
--- /dev/null
+++ b/src/main/java/net/lamgc/cgj/RankingUpdateTimer.java
@@ -0,0 +1,75 @@
+package net.lamgc.cgj;
+
+import net.lamgc.cgj.pixiv.PixivURL;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.IOException;
+import java.time.LocalDate;
+import java.util.Calendar;
+import java.util.Date;
+import java.util.Timer;
+import java.util.TimerTask;
+
+public class RankingUpdateTimer {
+
+    private final Timer timer = new Timer("PixivRankingUpdate@" + Integer.toHexString(this.hashCode()), true);
+    private final Logger log = LoggerFactory.getLogger("RankingUpdateTimer@" + Integer.toHexString(this.hashCode()));
+
+    /**
+     * 启动定时任务.
+     * 本方法在设置后立即返回
+     * @param firstRunDate 首次运行时间, 只需要设置日期, 时间为自动设置.
+     */
+    public void schedule(Date firstRunDate) {
+        Calendar cal = Calendar.getInstance();
+        cal.setTime(firstRunDate == null ? new Date() : firstRunDate);
+        LocalDate currentLocalDate = LocalDate.now();
+        if(cal.get(Calendar.DAY_OF_YEAR) <= currentLocalDate.getDayOfYear() && cal.get(Calendar.HOUR_OF_DAY) >= 12) {
+            cal.set(Calendar.DAY_OF_YEAR, cal.get(Calendar.DAY_OF_YEAR) + 1);
+        }
+        cal.set(Calendar.HOUR_OF_DAY, 12);
+        cal.set(Calendar.MINUTE, 30);
+        cal.set(Calendar.SECOND, 0);
+        cal.set(Calendar.MILLISECOND, 0);
+
+        log.info("已设置排行榜定时更新, 首次运行时间: {}", cal.getTime());
+        timer.schedule(new TimerTask() {
+            @Override
+            public void run() {
+                now(null);
+            }
+        }, cal.getTime(), 86400000); // 1 Day
+    }
+
+    public void now(Date queryDate) {
+        log.info("当前时间 {}, 定时任务开始执行...", new Date());
+        Calendar calendar = Calendar.getInstance();
+        calendar.setTime(queryDate == null ? new Date() : queryDate);
+        if(queryDate == null) {
+            calendar.add(Calendar.DATE, -1);
+        }
+        log.info("正在获取 {} 期排行榜数据...", calendar.getTime());
+        for (PixivURL.RankingMode rankingMode : PixivURL.RankingMode.values()) {
+            for (PixivURL.RankingContentType contentType : PixivURL.RankingContentType.values()) {
+                log.info("当前排行榜类型: {}.{}, 正在更新...", rankingMode.name(), contentType.name());
+                try {
+                    CQProcess.getRankingInfoByCache(contentType, rankingMode, calendar.getTime(), 1, 0);
+                    log.info("排行榜 {}.{} 更新完成.", rankingMode.name(), contentType.name());
+                } catch (IOException e) {
+                    log.error("排行榜 {}.{} 更新时发生异常", rankingMode.name(), contentType.name());
+                    log.error("异常信息如下", e);
+                }
+            }
+        }
+    }
+
+    /**
+     * 取消任务.
+     */
+    public void stop() {
+        timer.cancel();
+        log.warn("排行榜更新任务已取消.");
+    }
+
+}