feat(time): 尝试支持按日/周/月/总时长判别奖励。

src/main/java/cc/carm/plugin/timereward/manager/UserManager.java

@@ -1,132 +1,62 @@
 package cc.carm.plugin.timereward.manager;
 
+import cc.carm.lib.easyplugin.EasyPlugin;
+import cc.carm.lib.easyplugin.user.UserDataManager;
 import cc.carm.plugin.timereward.Main;
-import cc.carm.plugin.timereward.util.DataTaskRunner;
-import cc.carm.plugin.timereward.data.UserData;
+import cc.carm.plugin.timereward.data.IntervalType;
+import cc.carm.plugin.timereward.data.TimeRecord;
 import cc.carm.plugin.timereward.storage.database.MySQLStorage;
-import com.google.common.collect.ImmutableMap;
-import org.bukkit.Bukkit;
+import cc.carm.plugin.timereward.user.LockedRewardData;
+import cc.carm.plugin.timereward.user.UserRewardData;
 import org.bukkit.entity.Player;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
-import org.jetbrains.annotations.Unmodifiable;
 
-import java.util.HashMap;
-import java.util.Map;
+import java.time.Duration;
+import java.time.LocalDate;
+import java.util.LinkedHashMap;
 import java.util.UUID;
 
-public class UserManager {
+public class UserManager extends UserDataManager<UUID, UserRewardData> {
 
-    private final HashMap<UUID, UserData> userDataMap;
-
-    public UserManager() {
-        this.userDataMap = new HashMap<>();
+    public UserManager(@NotNull EasyPlugin plugin) {
+        super(plugin);
     }
 
-    @NotNull
-    public UserData readData(UUID userUUID) {
-        try {
-            long start = System.currentTimeMillis();
-            MySQLStorage storage = Main.getStorage();
-
-            UserData data = storage.loadData(userUUID);
-
-            if (data == null) {
-                Main.debugging("当前还不存在玩家 " + userUUID + " 的数据，视作新档。");
-                return new UserData(userUUID);
-            }
-
-            Main.debugging("读取 " + userUUID + " 的用户数据完成， 耗时 " + (System.currentTimeMillis() - start) + "ms。");
-
-            return data;
-        } catch (Exception e) {
-            Main.severe("无法正常读取玩家数据，玩家操作将不会被保存，请检查数据配置！");
-            Main.severe("Could not loadData user's data, please check the data conf!");
-            e.printStackTrace();
-            return new UserData(userUUID);
-        }
+    public @NotNull UserRewardData get(Player player) {
+        return get(player.getUniqueId());
     }
 
-    public void saveData(UserData data) {
-        try {
-            long start = System.currentTimeMillis();
-            MySQLStorage storage = Main.getStorage();
-
-            Main.debugging("正在保存 " + data.getUserUUID() + " 的用户数据...");
-            storage.saveClaimedData(data.getUserUUID(), data.getClaimedRewards());
-            storage.savePlayTime(data.getUserUUID(), data.getAllSeconds());
-            Main.debugging("保存 " + data.getUserUUID() + " 的用户数据完成，耗时 " + (System.currentTimeMillis() - start) + "ms。");
-
-        } catch (Exception e) {
-            Main.severe("无法正常保存玩家数据，请检查数据配置！");
-            Main.severe("Could not saveData user's data, please check the data conf!");
-            e.printStackTrace();
-        }
+    @Override
+    protected @Nullable UserRewardData loadData(@NotNull UUID key) throws Exception {
+        return Main.getStorage().loadData(key);
     }
 
-    public void loadData(UUID userUUID) {
-        getUserDataMap().put(userUUID, readData(userUUID));
+    @Override
+    public void saveData(@NotNull UserRewardData data) throws Exception {
+        if (data instanceof LockedRewardData) return; // 不保存
+
+        MySQLStorage storage = Main.getStorage();
+
+        // 只需要保存游玩时间数据，领取数据已经实时保存了
+        LocalDate date = LocalDate.now();
+        Duration daily = data.getOnlineDuration(IntervalType.DAILY);
+        Duration weekly = data.getOnlineDuration(IntervalType.WEEKLY);
+        Duration monthly = data.getOnlineDuration(IntervalType.MONTHLY);
+        Duration total = data.getOnlineDuration(IntervalType.TOTAL);
+        TimeRecord newRecord = new TimeRecord(date, daily, weekly, monthly, total);
+
+        storage.savePlayTime(data.getUserUUID(), newRecord);
     }
 
-    public void unloadData(UUID userUUID) {
-        unloadData(userUUID, true);
+    @Override
+    public @NotNull UserRewardData emptyUser(@NotNull UUID key) {
+        return new UserRewardData(key, TimeRecord.empty(), new LinkedHashMap<>());
     }
 
-    public void unloadData(UUID userUUID, boolean save) {
-        UserData data = getData(userUUID);
-        if (data == null) return;
-        if (save) saveData(data);
-        getUserDataMap().remove(userUUID);
-    }
-
-    public void loadAll() {
-        for (Player player : Bukkit.getOnlinePlayers()) {
-            if (getUserDataMap().containsKey(player.getUniqueId())) continue;
-            loadData(player.getUniqueId());
-        }
-    }
-
-    public void saveAll() {
-        getUserDataMap().values().forEach(this::saveData);
-    }
-
-    public void unloadAll(boolean save) {
-        if (save) saveAll();
-        getUserDataMap().clear();
-    }
-
-    @Nullable
-    public UserData getData(UUID userUUID) {
-        return getUserDataMap().get(userUUID);
-    }
-
-    @NotNull
-    public UserData getData(Player player) {
-        return getUserDataMap().get(player.getUniqueId());
-    }
-
-    public void editData(@NotNull DataTaskRunner task) {
-        try {
-            task.run(Main.getStorage());
-        } catch (Exception exception) {
-            Main.severe("无法正常更改玩家数据，请检查数据配置！");
-            Main.severe("Could not edit user's data, please check the data conf!");
-            exception.printStackTrace();
-        }
-    }
-
-    public void editDataAsync(@NotNull DataTaskRunner task) {
-        Main.getInstance().getScheduler().runAsync(() -> editData(task));
-    }
-
-    @NotNull
-    @Unmodifiable
-    public Map<UUID, UserData> listUserData() {
-        return ImmutableMap.copyOf(getUserDataMap());
-    }
-
-    protected @NotNull HashMap<UUID, UserData> getUserDataMap() {
-        return userDataMap;
+    @Override
+    public @NotNull UserRewardData errorUser(@NotNull UUID key) {
+        return new LockedRewardData(key); // 避免影响数据存储
     }
 
 }