diff --git a/src/main/java/com/yingjun/ssm/service/UserService.java b/src/main/java/com/yingjun/ssm/service/UserService.java index 9dd19fae5004690cfef6f14f6e9223d0a160c4d2..ec88520bf644a0c78680cfb3eca42c906aab1430 100644 --- a/src/main/java/com/yingjun/ssm/service/UserService.java +++ b/src/main/java/com/yingjun/ssm/service/UserService.java @@ -1,6 +1,9 @@ package com.yingjun.ssm.service; import java.util.List; import com.yingjun.ssm.entity.User; + +import javax.servlet.http.HttpSession; + /** * @author liyunfeng * 增加必要注释 @@ -77,5 +80,5 @@ public interface UserService { * 处理用户注册 * @param user */ - public void userRegister(User user); + public void userRegister(User user, String code, HttpSession session); } \ No newline at end of file diff --git a/src/main/java/com/yingjun/ssm/service/impl/UserServiceImpl.java b/src/main/java/com/yingjun/ssm/service/impl/UserServiceImpl.java index 9c74a15281011716b60450707ab08ea4c4df50f6..00f50cd9f4ec6432a0ad7f2a845ee97c186e5783 100644 --- a/src/main/java/com/yingjun/ssm/service/impl/UserServiceImpl.java +++ b/src/main/java/com/yingjun/ssm/service/impl/UserServiceImpl.java @@ -9,6 +9,8 @@ import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; +import javax.servlet.http.HttpSession; + /** * @author liyunfeng * 普通用户 @@ -86,8 +88,22 @@ public class UserServiceImpl implements UserService { * @param user */ @Override - public void userRegister(User user) { - + public void userRegister(User user, String code,HttpSession session) { + try{ + String emailCode = (String)session.getAttribute("code"); + if(!emailCode.equals(code)){ + throw new BizException("验证码错误"); + } + if(1 == userDao.insertTUser(user)){ + throw new Exception("插入数据影响函数不唯一"); + } + }catch (BizException biz){ + LOG.error("验证码错误",biz); + throw new BizException("验证码错误"); + }catch (Exception e){ + LOG.error("插入数据影响函数不唯一",e); + throw new BizException("插入数据影响函数不唯一"); + } } public UserDao getTUserDao() { diff --git a/src/main/java/com/yingjun/ssm/web/UserController.java b/src/main/java/com/yingjun/ssm/web/UserController.java index e61cb1a8225f58a94654beb1382a7f137a6efe5b..20b849decb4d922e902887802d1aa3f452090546 100644 --- a/src/main/java/com/yingjun/ssm/web/UserController.java +++ b/src/main/java/com/yingjun/ssm/web/UserController.java @@ -73,9 +73,9 @@ public class UserController { */ @ResponseBody @RequestMapping(value = "/register",produces = {"application/json;charset=UTF-8"}) - public BaseResult register(User user) { + public BaseResult register(User user,String code,HttpSession session) { try { - userService.userRegister(user); + userService.userRegister(user,code,session); } catch (BizException e) { return new BaseResult<>(false, e.getMessage()); } catch (Exception e) {