5、用户注册功能
5.1 需求分析
注册功能：保存用户信息到数据库
验证码功能
展示验证码：展示验证码图片，并可以点击切换
校验验证码：验证码填写不正确，则注册失败
5.2 实现流程分析
(1)前端通过表单发送请求和数据给Web层的RegisterServlet
(2)在RegisterServlet中接收请求和数据（用户名和密码）
(3)RegisterServlet接收到请求和数据后，调用Service层完成用户信息的保存
(4)在Service层需要编写UserService类，在类中实现register方法，需要判断用户是否已经存在，如果不存在，则完成用户数据的保存
(5)在UserMapper接口中，声明两个方法，一个是根据用户名查询用户信息方法，另一个是保存用户信息方法
(6)在UserService类中保存成功则返回true，失败则返回false,将数据返回给Web层
(7)Web层获取到结果后，如果返回的是true,则提示注册成功，并转发到登录页面，如果返回false则提示用户名已存在并转发到注册页面
5.3 具体实现
5.3.1 Dao层代码
package com.sun.mapper; import com.sun.pojo.User; import org.apache.ibatis.annotations.Insert; import org.apache.ibatis.annotations.Param; import org.apache.ibatis.annotations.Select; /** * @author JumperSun * @date 2022-09-22-16:54 */ public interface UserMapper { /** * 根据用户名和密码查询用户对象 * @param username * @param password * @return */ @Select("select * from user where username= #{username} and password = #{password}") User select(@Param("username") String username, @Param("password") String password); /** * 根据用户名查询用户对象 * @param username * @return */ @Select("select * from user where username = #{username}") User selectByUsername(@Param("username") String username); /** * 添加用户 * @param user * @return */ @Insert("insert into user values(null,#{username},#{password}))") boolean addUser(User user); }
5.3.2 编写Service层代码
public class UserService { //1.使用工具类获取SqlSessionFactory SqlSessionFactory factory = SqlSessionFactoryUtils.getSqlSessionFactory(); /** * 注册方法 * @return */ public boolean register(User user){ //2. 获取SqlSession SqlSession sqlSession = factory.openSession(); //3. 获取UserMapper UserMapper mapper = sqlSession.getMapper(UserMapper.class); //4. 判断用户名是否存在 User u = mapper.selectByUsername(user.getUsername()); if(u == null){ // 用户名不存在，注册 mapper.add(user); sqlSession.commit(); } sqlSession.close(); return u == null; } }
5.3.3 完成页面和Web层的代码编写
将register.html内容修改成register.jsp <%@ page contentType="text/html;charset=UTF-8" language="java" %> <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>欢迎注册</title> <link href="css/register.css" rel="stylesheet"> </head> <body> <div class="form-div"> <div class="reg-content"> <h1>欢迎注册</h1> <span>已有帐号？</span> <a href="login.html">登录</a> </div> <form id="reg-form" action="/login-demo/registerServlet" method="post"> <table> <tr> <td>用户名</td> <td class="inputs"> <input name="username" type="text" id="username"> <br> <span id="username_err" class="err_msg" style="display:none">用户名不太受欢迎</span> </td> </tr> <tr> <td>密码</td> <td class="inputs"> <input name="password" type="password" id="password"> <br> <span id="password_err" class="err_msg" style="display: none">密码格式有误</span> </td> </tr> <tr> <td>验证码</td> <td class="inputs"> <input name="checkCode" type="text" id="checkCode"> <img src="imgs/a.jpg"> <a href="#" id="changeImg" >看不清？</a> </td> </tr> </table> <div class="buttons"> <input value="注 册" type="submit" id="reg_btn"> </div> <br class="clear"> </form> </div> </body> </html>
编写RegisterServlet
@WebServlet("/registerServlet") public class RegisterServlet extends HttpServlet { private UserService service = new UserService(); @Override protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { //1. 获取用户名和密码数据 String username = request.getParameter("username"); String password = request.getParameter("password"); User user = new User(); user.setUsername(username); user.setPassword(password); //2. 调用service 注册 boolean flag = service.register(user); //3. 判断注册成功与否 if(flag){ //注册功能，跳转登陆页面 request.setAttribute("register_msg","注册成功，请登录"); request.getRequestDispatcher("/login.jsp").forward(request,response); }else { //注册失败，跳转到注册页面 request.setAttribute("register_msg","用户名已存在"); request.getRequestDispatcher("/register.jsp").forward(request,response); } } @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { this.doGet(request, response); } }
需要在页面上展示后台返回的错误信息，需要修改register.jsp
修改前:<span id="username_err" class="err_msg" style="display:none">用户名不太受欢迎</span> 修改后:<span id="username_err" class="err_msg">${register_msg}</span>
如果注册成功，需要把成功信息展示在登录页面，所以也需要修改login.jsp
修改前:<div id="errorMsg">${login_msg}</div> 修改后:<div id="errorMsg">${login_msg} ${register_msg}</div>
修改login.jsp，将注册跳转地址修改为register.jsp
修改前：<a href="register.html">没有账号？</a> 修改后: <a href="register.jsp">没有账号？</a>
5.2.4 启动测试
如果是注册的用户信息已经存在:
如果注册的用户信息不存在，注册成功:
6、验证码-展示
6.1 需求分析
展示验证码：展示验证码图片，并可以点击切换
验证码的生成是通过工具类来实现的，具体的工具类参考
登录注册案例\CheckCodeUtil.java
在该工具类中编写main方法进行测试:
public static void main(String[] args) throws IOException { //生成验证码的图片位置 OutputStream fos = new FileOutputStream("d://a.jpg"); //checkCode为最终验证码的数据 String checkCode = CheckCodeUtil.outputVerifyImage(100, 50, fos, 4); System.out.println(checkCode); }
生成完验证码以后，我们就可以知晓:
验证码就是使用Java代码生成的一张图片
验证码的作用:防止机器自动注册，攻击服务器
6.2 实现流程分析
(1)前端发送请求给CheckCodeServlet
(2)CheckCodeServlet接收到请求后，生成验证码图片，将图片用Reponse对象的输出流写回到前端
重点:如何将图片写回到前端浏览器呢?
(1)Java中已经有工具类生成验证码图片，测试类中只是把图片生成到磁盘上
(2)生成磁盘的过程中使用的是OutputStream流，如何把这个图片生成在页面呢?
(3)前面在将Reponse对象的时候，它有一个方法可以获取其字节输出流，getOutputStream()
(4)综上所述，我们可以把写往磁盘的流对象更好成Response的字节流，即可完成图片响应给前端
6.3 具体实现
(1)修改Register.jsp页面，将验证码的图片从后台获取
<tr> <td>验证码</td> <td class="inputs"> <input name="checkCode" type="text" id="checkCode"> <img id="checkCodeImg" src="/login-demo/checkCodeServlet"> <a href="#" id="changeImg" >看不清？</a> </td> </tr> <script> document.getElementById("changeImg").onclick = function () { //路径后面添加时间戳的目的是避免浏览器进行缓存静态资源 document.getElementById("checkCodeImg").src = "/login-demo/checkCodeServlet?"+new Date().getMilliseconds(); } </script>
(2)编写CheckCodeServlet类，用来接收请求生成验证码
@WebServlet("/checkCodeServlet") public class CheckCodeServlet extends HttpServlet { @Override protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { // 生成验证码 ServletOutputStream os = response.getOutputStream(); String checkCode = CheckCodeUtil.outputVerifyImage(100, 50, os, 4); } @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { this.doGet(request, response); } }
7、验证码-校验
7.1 需求分析
判断程序生成的验证码 和 用户输入的验证码 是否一样，如果不一样，则阻止注册
验证码图片访问和提交注册表单是两次请求，所以要将程序生成的验证码存入Session中
思考:为什么要把验证码数据存入到Session中呢?
生成验证码和校验验证码是两次请求，此处就需要在一个会话的两次请求之间共享数据
验证码属于安全数据类的，所以我们选中Session来存储验证码数据。
7.2 实现流程分析
(1)在CheckCodeServlet中生成验证码的时候，将验证码数据存入Session对象
(2)前端将验证码和注册数据提交到后台，交给RegisterServlet类
(3)RegisterServlet类接收到请求和数据后，其中就有验证码，和Session中的验证码进行对比
(4)如果一致，则完成注册，如果不一致，则提示错误信息
7.3 具体实现
(1)修改CheckCodeServlet类，将验证码存入Session对象
@WebServlet("/checkCodeServlet") public class CheckCodeServlet extends HttpServlet { @Override protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { // 生成验证码 ServletOutputStream os = response.getOutputStream(); String checkCode = CheckCodeUtil.outputVerifyImage(100, 50, os, 4); // 存入Session HttpSession session = request.getSession(); session.setAttribute("checkCodeGen",checkCode); } @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { this.doGet(request, response); } }
(2)在RegisterServlet中，获取页面的和session对象中的验证码，进行对比
package com.sun.web; import com.sun.pojo.User; import com.sun.service.UserService; import javax.servlet.ServletException; import javax.servlet.annotation.WebServlet; import javax.servlet.http.*; import java.io.IOException; @WebServlet("/registerServlet") public class RegisterServlet extends HttpServlet { private UserService service = new UserService(); @Override protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { //1. 获取用户名和密码数据 String username = request.getParameter("username"); String password = request.getParameter("password"); User user = new User(); user.setUsername(username); user.setPassword(password); // 获取用户输入的验证码 String checkCode = request.getParameter("checkCode"); // 程序生成的验证码，从Session获取 HttpSession session = request.getSession(); String checkCodeGen = (String) session.getAttribute("checkCodeGen"); // 比对 if(!checkCodeGen.equalsIgnoreCase(checkCode)){ request.setAttribute("register_msg","验证码错误"); request.getRequestDispatcher("/register.jsp").forward(request,response); // 不允许注册 return; } //2. 调用service 注册 boolean flag = service.register(user); //3. 判断注册成功与否 if(flag){ //注册功能，跳转登陆页面 request.setAttribute("register_msg","注册成功，请登录"); request.getRequestDispatcher("/login.jsp").forward(request,response); }else { //注册失败，跳转到注册页面 request.setAttribute("register_msg","用户名已存在"); request.getRequestDispatcher("/register.jsp").forward(request,response); } } @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { this.doGet(request, response); } }
至此，用户的注册登录功能就已经完成了。
