【SpringBoot】DEMO：实战②.⑤——登录注册功能实现，并加入Session

SpringBoot学习视频
最近看的小匠SpringBoot的教学视频，看了20多章的网课，跟着老师跑，感觉比较迷，希望巩固知识

小匠视频链接: https://www.bilibili.com/video/av65117012

• 今天刚刚复习了session和cookie，对上一篇博客：【SpringBoot】DEMO：实战②——登录注册功能实现 进行修改，加入session和cookie

• 根据我的上一篇博客进行修改，请根据 【SpringBoot】DEMO：实战②——登录注册功能实现结合观看

• 什么是cookie和session呢

• 请移步到 【Cookie和Session】初学，尝试理解总结

1、在注册时做好准备：为每个用户生成随机数

• 在注册的时候，除了存入 账号 和 密码，还需要使用 UUID.randomUUID().toString() 方法为每一个用户生成随机数token

• 在regiestController中：

@PostMapping("/regiest")
    public String regiest_show(
            @RequestParam("username") String username,
            @RequestParam("password") String password,
            @RequestParam("btn_regiest") String btn_regiest,
            Model model
    ) {
        //把前端的username和password写入user对象，方便后续操作
        User user = new User();
        user.setUsername(username);
        user.setPassword(password);
!____________________________________________________________________!
        //使用UUID，生成唯一的随机数，写入数据库，作为唯一标识码
        String token = UUID.randomUUID().toString();
        user.setToken(token);
!____________________________________________________________________!
        if(username == null || username == "" || password == "" || password == null ){
            //注册失败，出错提示
            model.addAttribute("btn_regiest",btn_regiest);
            return "error";
        }else{
            //注册成功
            //使用findByUsername（）方法，判断userCheck是否为空（在数据库中是否存在）
            User userCheck = userMapper.findByUsername(username);
            if (userCheck == null) {
                //账号还不存在数据库，执行插入操作，注册成功，返回ok_regiest页面
                userMapper.add(user);
                model.addAttribute("regiest_username", username);
                model.addAttribute("regiest_password", password);
                return "ok_regiest";
            } else {
                int CunZai = 1;
                //账号存在，返回error页面
                model.addAttribute("regiest_username", username);
                model.addAttribute("CunZai",CunZai);
                return "error";
            }
        }
    }

• 数据库展示：
  

2、在登录入写入cookie

• 登录成功之后，通过 用户名 查询其唯一的随机码token
• 在login（）方法中添加 HttpServletResponse response 参数，用于添加cookie
• 使用response.addCookie(new Cookie(“token”,token))，自定义cookie，名字为：token，value为 token的随机数

具体代码如下：

• 在loginController中：

@PostMapping("/login")
    //获取三个参数，账号，密码，按钮是否被点击
    public String login(@Param("username") String username,
                        @Param("password") String password,
                        !____________________________________________________________________!
                        HttpServletResponse response,
                        !____________________________________________________________________!
                        Model model) {
        User user = new User();
        user.setUsername(username);
        user.setPassword(password);
        //使用findByUsername_login（）方法，对数据库进行检查，判断账号密码是否正确
        User check_login = userMapper.findByUsername_login(username,password);
        //正确，返回index页面，错误，返回error页面,写cookie和session
        if(check_login != null){
            model.addAttribute("success_login_username",username);
            //创建userList对象去承接service层的getUserList()方法中return的数据，装进model
            List<User> userList = userlist.getUserList();
            model.addAttribute("userList", userList);

!____________________________________________________________________!
			//查询数据库中username对应的唯一token，
            User user_token = userMapper.username_to_setToken(username);
            String token = user_token.getToken();
!____________________________________________________________________!
            if(user != null){
!____________________________________________________________________!
				//自定义cokie，名字为：token，value为 token的随机数
                response.addCookie(new Cookie("token",token));
!____________________________________________________________________!
                return "redirect:/";
            }else{
                return "redirect:/";
            }

        }else{
            return "error";
        }
    }

效果展示：

3、登录成功，进入主页，取cookie验证，写session

• 当登录成功进入主页，获取Cookie中名为 token 的信息

• 把名为 token 的信息放入数据库进行查询，若存在，写入session，即可保持登录态

• 在 2、在登录入写入cookie 中使用了 ：HttpServletResponse 写入cookie，现在我们需要去cookie，要使用 HttpServletRequest 请求Cookie中的值进行查询

代码展示：

• 在indexController中：

@GetMapping("/")
    public String index(
    !_____________________________________________!
            HttpServletRequest request)
    !_____________________________________________!        
    {
        Cookie[] cookies = request.getCookies();
        for (Cookie cookie : cookies) {
            if(cookie.getName().equals("token")){
                String token = cookie.getValue();
                User user = userMapper.findByToken(token);
                if (user != null){
                	//查询到token中的value值存在，写入session，保持登录态
                    request.getSession().setAttribute("user",user);
                }
                break;
            }
        }
        return "index";
    }

完工！