package com.rongyichuang.auth.util;
|
|
import io.jsonwebtoken.*;
|
import io.jsonwebtoken.security.Keys;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Component;
|
|
import javax.crypto.SecretKey;
|
import java.util.Date;
|
|
/**
|
* JWT工具类
|
*/
|
@Component
|
public class JwtUtil {
|
|
private static final Logger logger = LoggerFactory.getLogger(JwtUtil.class);
|
|
@Value("${app.jwt.secret}")
|
private String jwtSecret;
|
|
@Value("${app.jwt.expiration:86400000}") // 默认24小时
|
private long jwtExpiration;
|
|
/**
|
* 生成JWT token（旧版本，保持兼容性）
|
*/
|
public String generateToken(Long userId, String phone) {
|
return generateToken(userId, phone, null);
|
}
|
|
/**
|
* 生成JWT token（新版本，支持wxopenid）
|
*/
|
public String generateToken(Long userId, String phone, String wxopenid) {
|
Date now = new Date();
|
Date expiryDate = new Date(now.getTime() + jwtExpiration);
|
|
SecretKey key = Keys.hmacShaKeyFor(jwtSecret.getBytes());
|
|
JwtBuilder builder = Jwts.builder()
|
.setSubject(userId.toString())
|
.setIssuedAt(now)
|
.setExpiration(expiryDate);
|
|
// 只有当phone不为null时才添加phone claim
|
if (phone != null) {
|
builder.claim("phone", phone);
|
}
|
|
// 只有当wxopenid不为null时才添加wxopenid claim
|
if (wxopenid != null) {
|
builder.claim("wxopenid", wxopenid);
|
}
|
|
return builder.signWith(key, SignatureAlgorithm.HS256).compact();
|
}
|
|
/**
|
* 从token中获取用户ID
|
*/
|
public Long getUserIdFromToken(String token) {
|
Claims claims = getClaimsFromToken(token);
|
return Long.parseLong(claims.getSubject());
|
}
|
|
/**
|
* 从token中获取手机号
|
*/
|
public String getPhoneFromToken(String token) {
|
Claims claims = getClaimsFromToken(token);
|
return claims.get("phone", String.class);
|
}
|
|
/**
|
* 从token中获取微信openid
|
*/
|
public String getWxOpenidFromToken(String token) {
|
Claims claims = getClaimsFromToken(token);
|
return claims.get("wxopenid", String.class);
|
}
|
|
/**
|
* 验证token是否有效
|
*/
|
public boolean validateToken(String token) {
|
try {
|
getClaimsFromToken(token);
|
return true;
|
} catch (JwtException | IllegalArgumentException e) {
|
logger.error("JWT token验证失败: {}", e.getMessage());
|
return false;
|
}
|
}
|
|
/**
|
* 检查token是否过期
|
*/
|
public boolean isTokenExpired(String token) {
|
try {
|
Claims claims = getClaimsFromToken(token);
|
return claims.getExpiration().before(new Date());
|
} catch (JwtException | IllegalArgumentException e) {
|
return true;
|
}
|
}
|
|
/**
|
* 从token中解析Claims
|
*/
|
private Claims getClaimsFromToken(String token) {
|
SecretKey key = Keys.hmacShaKeyFor(jwtSecret.getBytes());
|
return Jwts.parserBuilder()
|
.setSigningKey(key)
|
.build()
|
.parseClaimsJws(token)
|
.getBody();
|
}
|
}
|
