Spring Boot整合JWT的實現(xiàn)步驟

更新時間：2021年08月06日 08:33:47 作者：定格&不褪

本文主要介紹了Spring Boot整合JWT，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

springboot整合jwt步驟：

1、登錄時，驗證賬號和密碼成功后，生成jwt，返回給前端；
2、前端接收后保存，再做其他操作，比如增刪改查時，同時將jwt傳給后端進行驗證，如果jwt當做參數(shù)一起傳給后端，那么每個操作都會有jwt，為了方便，把jwt放到請求頭中，通過攔截器來驗證。

代碼

代碼結(jié)構(gòu)圖如下，除了常規(guī)的controller、entity、mapper和service層，還有兩個攔截器和注冊攔截器，圖中用紅字進行注釋（這篇文章稍微有點長，因為我把代碼都放上來了，結(jié)合下邊的思路和結(jié)構(gòu)圖就能理解個大概）。

思路：有請求過來，通過攔截器進行攔截，但放行登錄請求，如果登錄成功，生成JWT令牌，返回給前端，當前端再有其他請求過來時，攔截器會攔截并解析token，如果通過就允許業(yè)務操作，否則就返回相應提示信息。

在這里插入圖片描述

1、創(chuàng)建jwt數(shù)據(jù)庫，然后創(chuàng)建user表

CREATE TABLE `user` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `username` varchar(255) DEFAULT NULL,
  `password` varchar(255) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=3 DEFAULT CHARSET=utf8mb4;

插入數(shù)據(jù)

INSERT INTO `user` VALUES (1, 'zhangsan', '123');
INSERT INTO `user` VALUES (2, 'lisi', '123');

2、創(chuàng)建Spring Boot項目，在pom文件中，添加jwt等一些依賴和properties：

        <!--jwt-->
        <dependency>
            <groupId>com.auth0</groupId>
            <artifactId>java-jwt</artifactId>
            <version>3.4.0</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>

        <dependency>
            <groupId>org.mybatis.spring.boot</groupId>
            <artifactId>mybatis-spring-boot-starter</artifactId>
            <version>2.1.4</version>
        </dependency>
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>druid</artifactId>
            <version>1.2.4</version>
        </dependency>
        <!--我的mysql是5.6，所以這里是5.1版本-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>5.1.47</version>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-tx</artifactId>
            <version>5.3.8</version>
        </dependency>

編寫application.properties

server.port=8899
spring.application.name=jwt

spring.datasource.type=com.alibaba.druid.pool.DruidDataSource
spring.datasource.driver-class-name=com.mysql.jdbc.Driver
spring.datasource.url=jdbc:mysql://localhost:3306/jwt?useSSL=true&characterEncoding=utf-8&serverTimezone=UTC
spring.datasource.username=root
spring.datasource.password=root

mybatis.type-aliases-package=com.jwt.entity
mybatis.mapper-locations=classpath:mapper/*Mapper.xml

#打印sql
logging.level.com.jwt.mapper=debug

3、編寫代碼

3.1、編寫entity包下的User類

public class User {
    private String id;
    private String username;
    private String password;
    //省略了get和set方法
}

3.2、編寫util包下的JWTUtils類

public class JWTUtils {

    private static final String SING = "lu123456";

    /**
     * 生成token
     */
    public static String getToken(Map<String,String> map){
        Calendar instance = Calendar.getInstance();
        //默認7天過期
        instance.add(Calendar.DATE,7);
        //創(chuàng)建jwt builder
        JWTCreator.Builder builder = JWT.create();

        map.forEach((k,v)->{
            builder.withClaim(k,v);
        });

        String token = builder.withExpiresAt(instance.getTime())
                .sign(Algorithm.HMAC256(SING));
        return token;
    }

    /**
     * 驗證token合法性
     */
    public static DecodedJWT verify(String token){
        return JWT.require(Algorithm.HMAC256(SING)).build().verify(token);
    }
}

3.3、編寫controller層下UserController類

@RestController
@Slf4j
public class UserController {
    @Autowired
    private UserService userService;

    @GetMapping("/user/login")
    public Map<String,Object> login(User user){
        log.info("用戶名：[{}]",user.getUsername());
        log.info("密碼：[{}]",user.getPassword());

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

        try {
            User userDB = userService.login(user);
            Map<String,String> payload = new HashMap<>();
            payload.put("id",userDB.getId());
            payload.put("username",userDB.getUsername());
            //生成JWT令牌
            String token = JWTUtils.getToken(payload);
            map.put("state",true);
            map.put("token",token);
            map.put("msg","認證成功");
        }catch (Exception e){
            map.put("state",false);
            map.put("msg",e.getMessage());
        }
        return map;
    }

    @PostMapping("/user/test")
    public Map<String,Object> test(String token){
        Map<String, Object> map = new HashMap<>();
        map.put("state",true);
        map.put("msg","請求成功");
        return map;
    }
}

3.4、編寫service層下UserService類和UserServiceImpl 類

public interface UserService {
    User login (User user);
}

實現(xiàn)類：

@Service
public class UserServiceImpl implements UserService {

    @Autowired
    private UserMapper userMapper;

    @Override
    @Transactional(propagation = Propagation.SUPPORTS)
    public User login(User user) {
        //查詢數(shù)據(jù)庫
        User userDB = userMapper.login(user);
        if (userDB != null){
            return userDB;
        }
        throw new RuntimeException("登錄失敗");
    }
}

3.5、編寫mapper層的UserMapper類

@Mapper
public interface UserMapper {
    User login(User user);
}

3.6、編寫與UserMapper 對應的UserMapper.xml

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="com.jwt.mapper.UserMapper">

    <select id="login" parameterType="com.jwt.entity.User" resultType="com.jwt.entity.User">
        SELECT * FROM user WHERE username = #{username} AND password = #{password}
    </select>

</mapper>

3.7、配置攔截器，這樣得到jwt后，再次請求時把jwt放到請求頭中，就可以不用當參數(shù)傳遞。
編寫JWTInterceptor類：

/**
 * 配置攔截器
 */
public class JWTInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        Map<String, Object> map = new HashMap<>();

        //獲取請求頭中的令牌
        String token = request.getHeader("token");

        try {
            //驗證令牌
            DecodedJWT verify = JWTUtils.verify(token);
            return true;
        } catch (SignatureVerificationException e){
            e.printStackTrace();
            map.put("msg","無效簽名");
        } catch (TokenExpiredException e){
            e.printStackTrace();
            map.put("msg","token過期");
        } catch (AlgorithmMismatchException e){
            e.printStackTrace();
            map.put("msg","token算法不一致");
        } catch (Exception e){
            e.printStackTrace();
            map.put("msg","token無效");
        }
        map.put("state","flase");

        //將map轉(zhuǎn)為json
        String json = new ObjectMapper().writeValueAsString(map);
        response.setContentType("application/json;charset=UTF-8");
        response.getWriter().println(json);
        return false;
    }
}

編寫InterceptorConfig類：

/**
 * 注冊攔截器
 */
@Configuration
public class InterceptorConfig implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new JWTInterceptor())
                //攔截
                .addPathPatterns("/user/test")
                //放行
                .excludePathPatterns("/user/login");
    }
}