项目比较复杂的时候,需要进行换一个思路,对于问题进行拆开,功能主要是分为三个方面:第一个是访问注册页面,第二个是提交注册数据,第三个是激活注册账号。还是按照三层的逻辑进行。进一步凝练是:
- 访问注册页面,
点击顶部区域内的链接,打开注册页面。 - 提交注册数据
通过表单提交数据。
服务端验证账号是否已存在、邮箱是否已注册。
服务端发送激活邮件。 - 激活注册账号
点击邮件中的链接,访问服务端的激活服务。
访问注册页面
在controller包下新建一个LoginController,返回一下注册页面,在首页index里,点击注册默认跳转的是register页面。
package com.newcoder.community.controller;
import com.newcoder.community.entity.User;
import com.newcoder.community.service.UserService;
import com.newcoder.community.util.CommunityConstant;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import java.util.Map;
@Controller
public class LoginController implements CommunityConstant {
@Autowired
private UserService userService;
@RequestMapping(path = "/register",method = RequestMethod.GET)
public String getResisterPage(){
return "/site/register";
}
@RequestMapping(path = "/login",method = RequestMethod.GET)
public String getLoginPage(){
return "/site/login";
}
@RequestMapping(path = "/register",method = RequestMethod.POST)
public String register(Model model ,User user){
Map<String,Object> map = userService.register(user);
if (map == null || map.isEmpty()){
// 成功进行调转首页,
model.addAttribute("msg","注册成功,我们已经向您的邮箱发送一封激活邮件,请尽快激活");
model.addAttribute("target","/index");
return "/site/operate-result";
}else {
model.addAttribute("usernameMsg",map.get("usernameMsg"));
model.addAttribute("passwordMsg",map.get("passwordMsg"));
model.addAttribute("emailMsg",map.get("emailMsg"));
return "/site/register";
}
}
@RequestMapping(path = "/activation/{userId}/{code}",method = RequestMethod.GET)
public String activation(Model model, @PathVariable("userId") int userId, @PathVariable("code") String code){
int result = userService.activation(userId, code);
if (result == ACTIVATION_SUCCESS){
model.addAttribute("msg","激活成功,您的账号已经可以正常使用了");
model.addAttribute("target","/login");
} else if (result == ACTIVATION_REPEAT){
model.addAttribute("msg","无效的操作,该账号已经被激活");
model.addAttribute("target","/index");
}else {
model.addAttribute("msg","激活失败,您提供的激活码不正确");
model.addAttribute("target","/index");
}
return "/site/operate-result";
}
}
注册页面
封装一个工具类
第一个是生成随机字符串,利用UUID,为了简单,把生成的字符串里的短横线去掉。
第二个是MD5加密,对密码进行加密处理,同一个密码计算出来的MD5值是一样的,所以还是不够安全,我们对密码进行加盐,也就是在用户的密码后面拼接一个随机字符串后再进行MD5计算。
package com.newcoder.community.util;
import org.apache.commons.lang3.StringUtils;
import org.springframework.util.DigestUtils;
import java.util.UUID;
public class CommunityUtil {
// 生成随机字符串
// 提供上传的随机的名字
public static String generateUUid(){
return UUID.randomUUID().toString().replace("-","");
}
// 加密算法
// 加密只是加密 没法解密 加上随机的字符串
public static String md5(String key){
if (StringUtils.isBlank(key)){
return null;
}
return DigestUtils.md5DigestAsHex(key.getBytes());
}
}
注册服务
properties里配置一个项目地址localhost:80/community/index
community.path.domain = http://localhost:80
- 首先用注解引入项目路径和域名
- 写注册方法,返回值用map返回对应的消息。
- 首先对传进来的user对象做一个空值的判断
- 然后再验证账号和邮箱是否已注册过了。
再补全user对象的值,生成盐然后对密码进行加密。注册用户都是普通用户,type设为0,状态设置为未激活0,再生成一个激活码随机字符串头像使用牛客网的默认头像,0-1000号头像都可以使用注册时间生成后插入对象就可以了。用模板引擎生成html邮件,进行发送。
userService
package com.newcoder.community.service;
import com.newcoder.community.dao.UserMapper;
import com.newcoder.community.entity.User;
import com.newcoder.community.service.imp.MailClient;
import com.newcoder.community.util.CommunityConstant;
import com.newcoder.community.util.CommunityUtil;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.thymeleaf.TemplateEngine;
import org.thymeleaf.context.Context;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
@Service
public class UserService implements CommunityConstant {
@Autowired
private UserMapper userMapper;
@Autowired
private MailClient mailClient;
@Autowired
private TemplateEngine templateEngine;
@Value("${community.path.domain}")
private String domain;
@Value("${server.servlet.context-path}")
private String contextPath;
// 生成激活码
public User findUserById(int id){
return userMapper.selectById(id);
}
public Map<String,Object> register(User user){
Map<String,Object> map = new HashMap<>();
// 空值处理
if (user == null){
throw new IllegalArgumentException("参数不为空");
}
if (StringUtils.isBlank(user.getUsername())){
map.put("usernameMsg","账号不能为空");
return map;
}
if (StringUtils.isBlank(user.getPassword())){
map.put("passwordMsg","密码不能为空");
return map;
}
if (StringUtils.isBlank(user.getEmail())){
map.put("emailMsg","邮箱不能为空");
return map;
}
// 判断账号
User u = userMapper.selectByName(user.getUsername());
if (u!=null){
map.put("usernameMsg","该账号已经存在");
return map;
}
// 验证邮箱
u = userMapper.selectByEmail(user.getEmail());
if (u!=null){
map.put("emailMsg","该邮箱已经存在");
return map;
}
//开始注册。所谓的注册就是把用户放在库里,生成随机的字符串
user.setSalt(CommunityUtil.generateUUid().substring(0,5));
user.setPassword(CommunityUtil.md5(user.getPassword()+user.getSalt()));
user.setType(0);
user.setStatus(0);
user.setActivationCode(CommunityUtil.generateUUid());
// 随机的图像
user.setHeaderUrl(String.format("http://images.nowcoder.com/head/%dt.png",new Random().nextInt(1000)));
user.setCreateTime(new Date());
userMapper.insertUser(user);
// 需要发html文件
Context context = new Context();
context.setVariable("email",user.getEmail());
// url路径
String url = domain+contextPath+"/activation/"+user.getId()+"/"+user.getActivationCode();
context.setVariable("url",url);
//
String content = templateEngine.process("/mail/activation",context);
mailClient.senMail(user.getEmail(),"激活账号",content);
// 激活的状态是
return map;
}
public int activation(int userId,String code){
// 首先查到用户,激活码之后判断
User user = userMapper.selectById(userId);
if (user.getStatus() == 1){
// 表示激活重复
return ACTIVATION_REPEAT;
}else if(user.getActivationCode().equals(code)){
// 表示没问题
userMapper.updateStatus(userId,1);
return ACTIVATION_SUCCESS;
}
else {
return ACTIVATION_FAILURE;
}
}
}
然后在LoginController里写控制方法,首先调用service的register方法返回map消息。如果map为空,就返回注册成功的页面.
最后重新启动一下启动类,注册成功!