package com.trafficaudit.security.utils;
|
|
import io.jsonwebtoken.Claims;
|
import io.jsonwebtoken.Jwts;
|
import io.jsonwebtoken.SignatureAlgorithm;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Component;
|
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
@Component
|
public class JwtUtils {
|
|
@Value("${jwt.secret}")
|
private String secret;
|
|
@Value("${jwt.expiration}")
|
private Long expiration;
|
|
public String generateToken(String username) {
|
Map<String, Object> claims = new HashMap<>();
|
claims.put("username", username);
|
return Jwts.builder()
|
.setClaims(claims)
|
.setSubject(username)
|
.setIssuedAt(new Date())
|
.setExpiration(new Date(System.currentTimeMillis() + expiration))
|
.signWith(SignatureAlgorithm.HS256, secret)
|
.compact();
|
}
|
|
public String getUsernameFromToken(String token) {
|
Claims claims = Jwts.parser().setSigningKey(secret)
|
.parseClaimsJws(token).getBody();
|
return claims.getSubject();
|
}
|
|
public boolean validateToken(String token) {
|
try {
|
Jwts.parser().setSigningKey(secret).parseClaimsJws(token);
|
return true;
|
} catch (Exception e) {
|
return false;
|
}
|
}
|
|
/** 从 Authorization: Bearer xxx 请求头解析用户名；无效返回 null */
|
public String resolveUsername(String authorizationHeader) {
|
if (authorizationHeader == null || !authorizationHeader.startsWith("Bearer ")) {
|
return null;
|
}
|
try {
|
return getUsernameFromToken(authorizationHeader.substring(7));
|
} catch (Exception e) {
|
return null;
|
}
|
}
|
}
|
