@Api(tags = "用户相关接口")

@RestController

@RequestMapping("/user")

public class UserController {

@Autowired

private UserService userService;

@Autowired

private JwtProperties jwtProperties;

@ApiOperation("用户登录")

@PostMapping("/login")

public Result login(@RequestBody User user) {

user = userService.login(user);

//登录成功后，生成jwt令牌

Map<String, Object> claims = new HashMap<>();

claims.put("userId", user.getId());

String token = JwtUtil.createJWT(

jwtProperties.getUserSecretKey(),

jwtProperties.getUserTtl(),

claims);

UserLoginVo userLoginVo = UserLoginVo.builder()

.user(user)

.token(token)

.build();

return Result.okResult(userLoginVo);

}

@ApiOperation("注册用户")

@PostMapping

public Result addUser(@RequestBody UserDto userDto) {

userService.addUser(userDto);

return Result.okResult();

}

@ApiOperation("更新用户信息")

@PostMapping("/update")

public Result uploadAvatar(User user) {

userService.uploadAvatar(user);

return Result.okResult();

}

@GetMapping("/test")

public Result test() {

return Result.okResult("test");

}

}

写了几个常用的用户层接口用来测试，主要关注用户登录/login接口，其他的暂时无需理会。

配置JwtProperties 类

@Component

@ConfigurationProperties(prefix = "zwk.jwt")

@Data

public class JwtProperties {

/**

* 用户生成jwt令牌相关配置

*/

private String userSecretKey;

private long userTtl;

private String userTokenName;

}

JwtProperties 对应的配置文件

zwk:

jwt:

# 设置jwt签名加密时使用的秘钥

user-secret-key: zwkzwk

# 设置jwt过期时间

user-ttl: 7200000

# 设置前端传递过来的令牌名称

user-token-name: token

配置UserService 类

public interface UserService {

void addUser(UserDto userDto);

User login(User user);

void uploadAvatar(User user);

}

UserService 的实现类

@Service

public class UserServiceImpl implements UserService {

@Autowired

private UserMapper userMapper;

/**

* 新增用户

* @param userDto

*/

public void addUser(UserDto userDto) {

User user = new User();

BeanUtils.copyProperties(userDto, user);

//user.setEmail("123@qq.com");

user.setCreateTime(new Date());

user.setLoginTime(new Date());

userMapper.insert(user);

}

public User login(User user) {

String password = user.getPassword();

final User user1 = userMapper.getUserByName(user.getUsername());

if (user1 == null) {

throw new RuntimeException("该用户名不存在");

}

//对密码进行md5加密

//password = DigestUtils.md5DigestAsHex(password.getBytes());

if (!password.equals(user1.getPassword())){

throw new RuntimeException("密码错误");

}

return user1;

}

/**

* 更新用户信息

* @param user

* @return

*/

@Override

public void uploadAvatar(User user) {

userMapper.updateById(user);

}

}

这里主要是对用户登录时传过来的用户名和密码进行校验，校验通过后我们再重新回到控制层看看是怎么处理的。

@ApiOperation("用户登录")

@PostMapping("/login")

public Result login(@RequestBody User user) {

user = userService.login(user);

//登录成功后，生成jwt令牌

Map<String, Object> claims = new HashMap<>();

claims.put("userId", user.getId());

String token = JwtUtil.createJWT(

jwtProperties.getUserSecretKey(),

jwtProperties.getUserTtl(),

claims);

UserLoginVo userLoginVo = UserLoginVo.builder()

.user(user)

.token(token)

.build();

return Result.okResult(userLoginVo);

}

如果登录成功，代码将生成一个 JWT（JSON Web Token）令牌。JWT 是一种紧凑的、自包含的方式，用于在客户端和服务器之间传递安全信息。在这个例子中，JWT 令牌包含了用户的 ID 信息。claims 是一个 Map，用于存储 JWT 中的声明（Claims），这里存储了用户 ID。JwtUtil.createJWT 方法用于创建 JWT 令牌，它接收三个参数：用户的密钥（jwtProperties.getUserSecretKey()）、令牌的有效时间（jwtProperties.getUserTtl()）和声明信息（claims）。

导入User类

@Data

@AllArgsConstructor

@NoArgsConstructor

public class User implements Serializable {

private static final long serialVersionUID = 1L;

/**

* 主键

*/

@TableId

private Long id;

/**

* 用户名

*/

private String username;

private String password;

private String email;

@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")

private Date createTime;

@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss", timezone = "GMT+8")

private Date LoginTime;

/**

* 头像

*/

private String avatar;

}

导入UserLoginVo类

@Data

@NoArgsConstructor

@AllArgsConstructor

@Builder

public class UserLoginVo {

private String token;

private User user;

}

编写JwtUtil工具类，该类用来生成jwt令牌

public class JwtUtil {

/**

* 生成jwt

* 使用Hs256算法, 私匙使用固定秘钥

*

* @param secretKey jwt秘钥

* @param ttlMillis jwt过期时间(毫秒)

* @param claims 设置的信息

* @return

*/

public static String createJWT(String secretKey, long ttlMillis, Map<String, Object> claims) {

// 指定签名的时候使用的签名算法，也就是header那部分

SignatureAlgorithm signatureAlgorithm = SignatureAlgorithm.HS256;

// 生成JWT的时间

long expMillis = System.currentTimeMillis() + ttlMillis;

Date exp = new Date(expMillis);

// 设置jwt的body

JwtBuilder builder = Jwts.builder()

// 如果有私有声明，一定要先设置这个自己创建的私有的声明，这个是给builder的claim赋值，一旦写在标准的声明赋值之后，就是覆盖了那些标准的声明的

.setClaims(claims)

// 设置签名使用的签名算法和签名使用的秘钥

.signWith(signatureAlgorithm, secretKey.getBytes(StandardCharsets.UTF_8))

// 设置过期时间

.setExpiration(exp);

return builder.compact();

}

/**

* Token解密

*

* @param secretKey jwt秘钥 此秘钥一定要保留好在服务端, 不能暴露出去, 否则sign就可以被伪造, 如果对接多个客户端建议改造成多个

* @param token 加密后的token

* @return

*/

public static Claims parseJWT(String secretKey, String token) {

// 得到DefaultJwtParser

Claims claims = Jwts.parser()

// 设置签名的秘钥

.setSigningKey(secretKey.getBytes(StandardCharsets.UTF_8))

// 设置需要解析的jwt

.parseClaimsJws(token).getBody();

return claims;

}

}

以上就是我们前期准备工作，然后发现好像还是没用，因为我们还没有做自定义拦截处理。我们首先对除了/user/login接口进行放行，其他接口全部拦截。

编写JwtTokenAdminInterceptor 类重写HandlerInterceptor方法

@Component

@Slf4j

public class JwtTokenAdminInterceptor implements HandlerInterceptor {

@Autowired

private JwtProperties jwtProperties;

/**

* 校验jwt

*

* @param request

* @param response

* @param handler

* @return

* @throws Exception

*/

public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

//判断当前拦截到的是Controller的方法还是其他资源

if (!(handler instanceof HandlerMethod)) {

//当前拦截到的不是动态方法，直接放行

return true;

}

//1、从请求头中获取令牌

String token = request.getHeader(jwtProperties.getUserTokenName());

//2、校验令牌

try {

log.info("jwt校验:{}", token);

Claims claims = JwtUtil.parseJWT(jwtProperties.getUserSecretKey(), token);

Long userId = Long.valueOf(claims.get("userId").toString());

log.info("当前用户id：{}", userId);

//3、通过，放行

return true;

} catch (Exception ex) {

//4、不通过，响应401状态码

response.setStatus(401);

return false;

}

}

}

自定义拦截器WebMvcConfiguration

@Configuration

@Slf4j

public class WebMvcConfiguration extends WebMvcConfigurationSupport {

@Autowired

private JwtTokenAdminInterceptor jwtTokenAdminInterceptor;

/**

* 注册自定义拦截器

* @param registry

*/

protected void addInterceptors(InterceptorRegistry registry) {

log.info("开始注册自定义拦截器...");

registry.addInterceptor(jwtTokenAdminInterceptor)

.addPathPatterns("/user/**") //表示拦截所以前缀带/user的请求

.excludePathPatterns("/user/login"); //排除特定路径：excludePathPatterns("/user/login") 方法用于排除某些路径，

//即使它们匹配前面指定的模式。在这个例子中，/user/login 路径不会被 jwtTokenAdminInterceptor 拦截。

}

/**

* 设置静态资源映射

* @param registry

*/

protected void addResourceHandlers(ResourceHandlerRegistry registry) {

registry.addResourceHandler("/doc.html").addResourceLocations("classpath:/META-INF/resources/");

registry.addResourceHandler("/webjars/**").addResourceLocations("classpath:/META-INF/resources/webjars/");

}

}

然后对接口进行登录测试

登录测试