1 files changed, 80 insertions, 0 deletions
diff --git a/src/main/java/com/mesasoft/cn/config/TokenConfig.java b/src/main/java/com/mesasoft/cn/config/TokenConfig.java
new file mode 100644
index 0000000..7a83572
--- /dev/null
+++ b/src/main/java/com/mesasoft/cn/config/TokenConfig.java
@@ -0,0 +1,80 @@
+package com.mesasoft.cn.config;
+
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import com.mesasoft.cn.SketchApplication;
+import com.mesasoft.cn.modules.constant.ConfigConsts;
+import com.zhazhapan.modules.constant.ValueConsts;
+import com.zhazhapan.util.Checker;
+import com.zhazhapan.util.FileExecutor;
+import com.zhazhapan.util.Formatter;
+import com.zhazhapan.util.RandomUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.Hashtable;
+
+/**
+ * @author pantao
+ * @since 2018/1/26
+ */
+public class TokenConfig {
+
+    private static Logger logger = LoggerFactory.getLogger(TokenConfig.class);
+
+    public static String generateToken(String token, int userId) {
+        if (Checker.isNotEmpty(token)) {
+            SketchApplication.tokens.remove(token);
+        }
+        return generateToken(userId);
+    }
+
+    public static String generateToken(int userId) {
+        String token = RandomUtils.getRandomStringOnlyLetter(ValueConsts.THIRTY_TWO_INT);
+        SketchApplication.tokens.put(token, userId);
+        saveToken();
+        return token;
+    }
+
+    public static void saveToken() {
+        String tokens = Formatter.mapToJson(SketchApplication.tokens);
+        try {
+            FileExecutor.saveFile(SettingConfig.getStoragePath(ConfigConsts.TOKEN_OF_SETTINGS), tokens);
+        } catch (Exception e) {
+            logger.error("save token error： " + e.getMessage());
+        }
+    }
+
+    public static Hashtable<String, Integer> loadToken() {
+        Hashtable<String, Integer> tokens = new Hashtable<>(ValueConsts.SIXTEEN_INT);
+        try {
+            String token = FileExecutor.readFile(SettingConfig.getStoragePath(ConfigConsts.TOKEN_OF_SETTINGS));
+            JSONArray array = JSON.parseArray(token);
+            array.forEach(object -> {
+                JSONObject jsonObject = (JSONObject) object;
+                tokens.put(jsonObject.getString(ValueConsts.KEY_STRING), jsonObject.getInteger(ValueConsts
+                        .VALUE_STRING));
+            });
+        } catch (Exception e) {
+            logger.error("load token error: " + e.getMessage());
+        }
+        return tokens;
+    }
+
+    public static void removeTokenByValue(int userId) {
+        if (userId > 0) {
+            String removeKey = "";
+            for (String key : SketchApplication.tokens.keySet()) {
+                if (SketchApplication.tokens.get(key) == userId) {
+                    removeKey = key;
+                    break;
+                }
+            }
+            if (Checker.isNotEmpty(removeKey)) {
+                SketchApplication.tokens.remove(removeKey);
+                TokenConfig.saveToken();
+            }
+        }
+    }
+}