java token保存工具类

原创

mob64ca12ed4084 2024-08-10 06:48:41 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12ed4084的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java Token保存工具类

在现代应用程序中，安全性和用户体验都是至关重要的。随着分布式系统和微服务架构的兴起，Token认证方式越来越多地被应用于系统中。为了方便管理Token，我们可以创建一个Token保存工具类。本文将探讨如何使用Java编写这样一个工具类，帮助开发者更好地管理身份认证Token。

1. 什么是Token？

Token是一种认证机制，它通常用于验证用户身份。用户在登录时会获得一个Token，而在随后的请求中，用户需要在请求头中包含此Token。服务器则通过解析Token来确认用户的身份。

2. 为何需要Token保存工具类？

集中管理：通过工具类集中管理Token，便于修改和维护。
降低耦合：将Token逻辑与业务逻辑分离，提高代码的可读性和可维护性。
增强安全性：可以在工具类中实现Token的加密与解密，提升安全性。

3. 策划类图

首先，来看看这个Token保存工具类的类图：

classDiagram
    class TokenManager {
        +String generateToken(User user)
        +boolean validateToken(String token)
        +User getUserFromToken(String token)
        +void saveToken(String token, User user)
        +void deleteToken(String token)
    }
    
    class User {
        +String username
        +String role
    }
    
    TokenManager --> User

TokenManager 类

TokenManager类负责处理Token的生成、验证以及与User对象的交互。

4. 代码示例

现在我们来实现这个工具类的代码。

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.JwtBuilder;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

public class TokenManager {
    private static final String SECRET_KEY = "your-secret-key";
    private static final long EXPIRATION_TIME = 3600000; // 1 hour

    // 生成Token
    public String generateToken(User user) {
        JwtBuilder builder = Jwts.builder()
                .setSubject(user.getUsername())
                .setClaims(createClaims(user))
                .setIssuedAt(new Date())
                .setExpiration(new Date(System.currentTimeMillis() + EXPIRATION_TIME))
                .signWith(SignatureAlgorithm.HS256, SECRET_KEY);
        return builder.compact();
    }

    // 验证Token
    public boolean validateToken(String token) {
        try {
            Jwts.parser().setSigningKey(SECRET_KEY).parseClaimsJws(token);
            return true;
        } catch (Exception e) {
            return false;
        }
    }

    // 从Token获取用户信息
    public User getUserFromToken(String token) {
        Claims claims = Jwts.parser().setSigningKey(SECRET_KEY).parseClaimsJws(token).getBody();
        return new User(claims.getSubject(), claims.get("role").toString());
    }

    // 保存Token至某个存储（如数据库、内存等）
    public void saveToken(String token, User user) {
        // Placeholder for saving logic
    }

    // 删除Token
    public void deleteToken(String token) {
        // Placeholder for deleting logic
    }

    // 创建Claim
    private Map<String, Object> createClaims(User user) {
        Map<String, Object> claims = new HashMap<>();
        claims.put("role", user.getRole());
        return claims;
    }
}

User 类

用户类，它包含了用户名和角色等信息。

public class User {
    private String username;
    private String role;

    public User(String username, String role) {
        this.username = username;
        this.role = role;
    }

    public String getUsername() {
        return username;
    }

    public String getRole() {
        return role;
    }
}

5. 旅行图

在实现Token保存工具类的过程中，用户的身份认证和Token的处理流程可以形象化为一个旅行图，展示了用户如何获取Token以及Token如何被验证。

journey
    title 用户Token验证流程
    section 登录
      用户输入用户名和密码         : 5: 用户
      服务器验证用户名和密码       : 5: 服务器
      服务器返回Token              : 5: 服务器
    section 使用Token访问资源
      用户在请求头中加入Token     : 5: 用户
      服务器验证Token              : 5: 服务器
      服务器返回请求的资源        : 5: 服务器