java用什么存Token

原创

mob64ca12ecb6c5 2023-12-08 08:36:42 ©著作权

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

存储和管理Java中的Token

1. 流程图

flowchart TD
  A[生成Token] --> B[存储Token]
  B -- Token有效? --> C{是}
  C --> D[使用Token]
  C -- Token失效? --> E{是}
  E --> F[重新生成Token]
  F --> B

2. 详细步骤和代码示例

步骤1：生成Token

首先，我们需要生成一个Token。Token是一个唯一的字符串，用于验证用户身份和授权访问。在Java中，通常使用JWT（JSON Web Tokens）来生成和验证Token。

代码示例：

import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import java.util.Date;

public class TokenUtils {
    private static final String SECRET_KEY = "your_secret_key";
    private static final long EXPIRATION_TIME = 86400000; // Token过期时间，单位为毫秒

    public static String generateToken(String userId) {
        Date now = new Date();
        Date expiration = new Date(now.getTime() + EXPIRATION_TIME);

        return Jwts.builder()
                .setSubject(userId)
                .setIssuedAt(now)
                .setExpiration(expiration)
                .signWith(SignatureAlgorithm.HS512, SECRET_KEY)
                .compact();
    }
}

步骤2：存储Token

生成Token后，我们需要将其存储起来，以便后续使用。Token的存储方式可以根据具体需求选择，常见的方式包括数据库、缓存（如Redis）或内存存储。

代码示例：

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

public class TokenStorage {
    private static final Map<String, String> tokenMap = new HashMap<>();

    public static void saveToken(String userId, String token) {
        tokenMap.put(userId, token);
    }

    public static String getToken(String userId) {
        return tokenMap.get(userId);
    }

    public static void removeToken(String userId) {
        tokenMap.remove(userId);
    }
}

步骤3：使用Token

在需要验证用户身份和授权访问的地方，我们可以使用Token来进行身份验证。具体的验证逻辑可以根据业务需求进行定制。

代码示例：

import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureException;

public class TokenUtils {
    // ...

    public static boolean validateToken(String token) {
        try {
            Claims claims = Jwts.parser()
                    .setSigningKey(SECRET_KEY)
                    .parseClaimsJws(token)
                    .getBody();

            // 根据claims中的信息进行身份验证

            return true;
        } catch (SignatureException e) {
            // Token验证失败
            return false;
        }
    }
}

步骤4：Token失效处理

在一定时间后，Token可能会失效。我们需要对失效的Token进行处理，通常的做法是重新生成一个新的Token，并将其替换原有的Token。

代码示例：

import java.util.Date;

public class TokenUtils {
    // ...

    public static String refreshExpiredToken(String expiredToken, String userId) {
        if (validateToken(expiredToken)) {
            String newToken = generateToken(userId);
            TokenStorage.saveToken(userId, newToken);
            return newToken;
        }
        return null;
    }
}