JWT核心代码实现：轻松掌握用户身份认证！

2023-12-13 20:56:56

JWT核心代码详解：生成和校验Token

简介

在上一篇文章中，我们探讨了采用JSON Web Token (JWT)作为用户身份认证机制的决定。今天，我们将深入JWT的核心代码，了解如何实现两项关键功能：生成Token和校验Token。通过理解这些代码，我们可以更加全面地掌握JWT的认证机制。

生成Token

生成Token是JWT的一项基本功能，它使我们能够为认证的用户创建数字凭证。以下是用Java代码实现此功能的示例：

import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

public class TokenGenerator {

    // 定义密钥
    private static final String SECRET_KEY = "my-secret-key";

    // 生成Token
    public static String generateToken(String username) {
        // 设置主体（用户名）
        return Jwts.builder()
                .setSubject(username)
                // 设置签名算法和密钥
                .signWith(SignatureAlgorithm.HS512, SECRET_KEY)
                // 生成紧凑型Token字符串
                .compact();
    }
}

在这个示例中，我们首先创建一个Token构建器，然后设置Token的主体（即用户名）。接下来，我们指定签名算法和密钥，这是验证Token的关键元素。最后，我们生成紧凑的Token字符串，它包含有关用户身份和签名的所有必要信息。

校验Token

校验Token同样至关重要，因为它使我们能够验证用户的合法性和访问权限。以下是如何用Java代码实现此功能：

import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

public class TokenValidator {

    // 定义密钥
    private static final String SECRET_KEY = "my-secret-key";

    // 验证Token
    public static boolean validateToken(String token) {
        try {
            // 设置解析器和密钥
            Jwts.parser()
                    .setSigningKey(SECRET_KEY)
                    .parseClaimsJws(token);
            return true;
        } catch (Exception e) {
            return false;
        }
    }
}

在这个示例中，我们首先创建一个Token解析器并设置密钥。然后，我们尝试解析Token并获取它的声明。如果解析成功，这意味着Token是有效的，并且我们返回true。否则，我们返回false。

代码示例：使用Token

让我们举一个使用Token的示例：

public class AuthController {

    // 用户登录，获取Token
    @PostMapping("/login")
    public String login(@RequestBody LoginRequest loginRequest) {
        // 验证凭证，生成Token
        String token = TokenGenerator.generateToken(loginRequest.getUsername());
        return token;
    }

    // 受保护的API端点，需要有效的Token
    @GetMapping("/api/protected")
    public String protectedApi(@RequestHeader("Authorization") String token) {
        // 校验Token，如果无效则返回401
        if (!TokenValidator.validateToken(token)) {
            return "401 Unauthorized";
        }
        // 如果Token有效，则返回受保护的资源
        return "Protected API Response";
    }
}