feat；增加角色模块

2024-06-25 18:32:08 +08:00
parent 80cd48b97c
commit d2229e0028
24 changed files with 350 additions and 111 deletions
--- a/springboot-util/src/main/java/com/tiesheng/util/config/TsTokenConfig.java
+++ b/springboot-util/src/main/java/com/tiesheng/util/config/TsTokenConfig.java
@@ -0,0 +1,162 @@
+package com.tiesheng.util.config;
+
+
+import cn.hutool.core.map.MapUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.extra.servlet.ServletUtil;
+import cn.hutool.extra.spring.SpringUtil;
+import cn.hutool.jwt.JWT;
+import cn.hutool.jwt.JWTValidator;
+import com.tiesheng.util.ServletKit;
+import com.tiesheng.util.exception.ApiException;
+import com.tiesheng.util.pojos.TokenBean;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
+
+import java.util.Map;
+
+
+/**
+ * @author hao
+ */
+@Configuration
+@ConfigurationProperties("tiesheng.token")
+public class TsTokenConfig {
+
+    private Map<String, TokenBean> testMap = MapUtil.newHashMap();
+    private String encryptKey = "%kIp9frQCu";
+    private Integer expireHours = 48;
+    private String[] ignorePaths;
+    private boolean validLoginSign = false;
+
+
+    /**
+     * 获取当前登录的token
+     *
+     * @return
+     */
+    public static TokenBean get() {
+        TsTokenConfig tokenConfig = SpringUtil.getBean(TsTokenConfig.class);
+        return tokenConfig.validToken(true);
+    }
+
+
+    /**
+     * 获取当前登录的token
+     *
+     * @return
+     */
+    public static TokenBean getWithoutThr() {
+        TsTokenConfig tokenConfig = SpringUtil.getBean(TsTokenConfig.class);
+        return tokenConfig.validToken(false);
+    }
+
+
+    /**
+     * 验证token
+     *
+     * @param token
+     * @return
+     */
+    public TokenBean isTestToken(String token) {
+        if (testMap == null) {
+            return null;
+        }
+        return testMap.get(token);
+    }
+
+
+    /**
+     * 验证token
+     *
+     * @param thrExp
+     * @return
+     */
+    public TokenBean validToken(boolean thrExp) {
+        String token = ServletUtil.getHeader(ServletKit.getRequest(), "token", "utf-8");
+        return validToken(token, thrExp);
+    }
+
+
+    /**
+     * 验证token
+     *
+     * @return
+     */
+    public TokenBean validToken(String token, boolean thrExp) {
+        TokenBean tokenBean = isTestToken(token);
+        if (tokenBean != null) {
+            return tokenBean;
+        }
+
+        try {
+            JWT decode = JWT.of(token);
+            JWTValidator.of(decode).validateDate();
+            String id = decode.getPayload("id").toString();
+            String environmentType = decode.getPayload("environmentType").toString();
+            String service = decode.getPayload("service").toString();
+            String extra = StrUtil.toStringOrNull(decode.getPayload("extra"));
+            String roleId = StrUtil.toStringOrNull(decode.getPayload("roleId"));
+            tokenBean = new TokenBean(id, environmentType, service);
+            tokenBean.setRoleId(roleId);
+            tokenBean.setExtra(extra);
+        } catch (Exception ignored) {
+        }
+
+        if (tokenBean == null && thrExp) {
+            throw new ApiException(StrUtil.isEmpty(token) ? 110 : 112,
+                    StrUtil.isEmpty(token) ? "请先登录" : "登录过期，请重新登陆");
+        }
+
+        if (tokenBean == null) {
+            tokenBean = new TokenBean();
+            tokenBean.setId("");
+        }
+
+        return tokenBean;
+    }
+
+    ///////////////////////////////////////////////////////////////////////////
+    // setter\getter
+    ///////////////////////////////////////////////////////////////////////////
+
+    public Map<String, TokenBean> getTestMap() {
+        return testMap;
+    }
+
+    public void setTestMap(Map<String, TokenBean> testMap) {
+        this.testMap = testMap;
+    }
+
+    public String getEncryptKey() {
+        return encryptKey;
+    }
+
+    public void setEncryptKey(String encryptKey) {
+        this.encryptKey = encryptKey;
+    }
+
+    public Integer getExpireHours() {
+        return expireHours;
+    }
+
+    public void setExpireHours(Integer expireHours) {
+        this.expireHours = expireHours;
+    }
+
+    public String[] getIgnorePaths() {
+        return ignorePaths;
+    }
+
+    public void setIgnorePaths(String[] ignorePaths) {
+        this.ignorePaths = ignorePaths;
+    }
+
+    public boolean isValidLoginSign() {
+        return validLoginSign;
+    }
+
+    public void setValidLoginSign(boolean validLoginSign) {
+        this.validLoginSign = validLoginSign;
+    }
+}