spring boot登录流程之登录成功返回token

最新推荐文章于 2025-01-24 16:00:24 发布

原创最新推荐文章于 2025-01-24 16:00:24 发布 · 2.1k 阅读

7 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#spring

Spring Boot 专栏收录该内容

13 篇文章

订阅专栏

该代码段展示了用户登录过程，包括解密用户名、密码和验证码，验证用户信息，比对密码，并通过JWT生成授权令牌。如果验证失败，返回错误信息；验证成功则将用户信息存入session并设置JWT token。

controller

@PostMapping("/login")
public Result<JSONObject> login(@RequestBody UserLoginModel userLoginModel) throws Exception {
    Result<JSONObject> result = new Result<>();
    // 解密
    String username = userLoginUtil.decrypt(userLoginModel.getUsername());
    String pwd = userLoginUtil.decrypt(userLoginModel.getPwd());
    String code = userLoginUtil.decrypt(userLoginModel.getCode());
    Object codeCheck = redisUtil.get(codeCheckKey);
    if (code == null || code.length() == 0) {
        result.error500("验证码无效");
        return result;
    }
    if (!codeCheck.toString().equals(code.toLowerCase())) {
        result.error500("验证码错误");
        return result;
    }
    LambdaQueryWrapper<AppUser> queryWrapper = new LambdaQueryWrapper<>();
    queryWrapper.eq(AppUser::getUsername, username);
    AppUser appUser = appUserService.getOne(queryWrapper);

    if (appUser == null) {
        result.error500("用户名不存在");
        return result;
    }
    // 根据用户名和密码盐获取真实密码
    String password = userLoginUtil.encryptPassword(username, pwd, appUser.getSalt());
    if (!password.equals(appUser.getPassword())) {
        result.error500("用户名或密码错误");
        return result;
    }
	// 将当前用户存入session
    SessionUtil.setUser(appUser);
	// 生成token
    generateToken(appUser, result);
    return result;
}

token

private Result<JSONObject> generateToken(AppUser appUser, Result<JSONObject> result) {
   String password = appUser.getPassword();
    String username = appUser.getUsername();
	// 根据用户名和密码生成token
    String token = JWTUtil.sign(username, password);
    // 将token存入redis
    redisUtil.set("prefix_user_token_" + token, token);
    // 过期时间半小时
    redisUtil.expire("prefix_user_token_" + token, 60 * 30);
    JSONObject obj = new JSONObject();
    obj.put("token", token);
    obj.put("userInfo", appUser);
    result.success(obj);
    return result;
}

JWTUtil.sign

public static String sign(String username, String secret) {
   Date date = new Date(System.currentTimeMillis() + 60 * 30);
   Algorithm algorithm = Algorithm.HMAC256(secret);
   // 附带username信息
   return JWT.create().withClaim("username", username).withExpiresAt(date).sign(algorithm);
}