Init

src/main/java/com/ycwl/basic/utils/JwtAnalysisUtil.java

@@ -0,0 +1,82 @@
+package com.ycwl.basic.utils;
+
+
+import com.ycwl.basic.model.jwt.JwtInfo;
+import io.jsonwebtoken.Claims;
+import io.jsonwebtoken.JwtBuilder;
+import io.jsonwebtoken.Jwts;
+import io.jsonwebtoken.SignatureAlgorithm;
+
+import java.time.Instant;
+import java.time.LocalDateTime;
+import java.time.ZoneId;
+import java.util.Iterator;
+import java.util.Map;
+
+/**
+ * @author yangchen
+ */
+public class JwtAnalysisUtil {
+
+    /**
+     * 生成 Token
+     *
+     * @param jwtInfo
+     * @param priKey
+     * @param expireTime
+     * @return
+     * @throws Exception
+     */
+    public static String generateToken(JwtInfo jwtInfo, byte[] priKey, LocalDateTime expireTime) throws Exception {
+        JwtBuilder builder = Jwts.builder().setSubject(jwtInfo.getAccount())
+                .claim("userId", jwtInfo.getUserId())
+                .claim("roleId", jwtInfo.getRoleId())
+                .claim("phone", jwtInfo.getPhone())
+                .claim("name", jwtInfo.getName())
+                // 返回从1970 00：00：00 到现在的毫秒差（实际的过期时间）
+                .claim("expire", expireTime.atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
+        return builder.signWith(SignatureAlgorithm.RS256, RsaKeyUtil.getPrivateKey(priKey)).compact();
+    }
+
+    /**
+     * 解析 Token
+     *
+     * @param token
+     * @param pubKey
+     * @return
+     * @throws Exception
+     */
+    public static JwtInfo getInfoFromToken(String token, byte[] pubKey) throws Exception {
+        Claims body = (Claims) RsaKeyUtil.parserToken(token, pubKey).getBody();
+        Iterator var3 = body.entrySet().iterator();
+        while (var3.hasNext()) {
+            Map.Entry entry = (Map.Entry) var3.next();
+            if (!"sub".equals(entry.getKey())
+                    && !"userId".equals(entry.getKey())
+                    && !"phone".equals(entry.getKey())
+                    && !"roleId".equals(entry.getKey())
+                    && !"account".equals(entry.getKey())
+                    && !"name".equals(entry.getKey())
+                    && !"roleName".equals(entry.getKey())
+                    && !"expire".equals(entry.getKey()));
+        }
+
+        // convert
+        LocalDateTime expireTime = null;
+        try {
+            // expire time
+            Object expire = body.get("expire");
+            if (expire != null) {
+                expireTime = LocalDateTime.ofInstant(Instant.ofEpochMilli((Long) expire), ZoneId.systemDefault());
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return new JwtInfo(StringUtil.a(body.get("name")),
+                StringUtil.a(body.get("userId")),
+                StringUtil.a(body.get("roleId")),
+                body.getSubject(),
+                StringUtil.a(body.get("phone")),
+                expireTime);
+    }
+}