邮件注册功能业务实现

注册账号时，通过邮件功能发送给注册者一封邮件，使其点击链接激活该账户，以供后续使用。

1、导入commons-lang3包，用于一些字符、集合空值检查。

<dependency>
    <groupId>org.apache.commons</groupId>
    <artifactId>commons-lang3</artifactId>
</dependency>

2、配置访问路径以及域名

server.servlet.context-path=/community
community.path.domain=http://localhost:8888

3、编写Utils工具类

public class CommunityUtil {

    // 生成随机的字符串
    public static String generateUUID() {

        return UUID.randomUUID().toString().replaceAll("-", "");
    }

    //MD5加密
    public static String md5(String key) {

        //判断是否为空
        if (StringUtils.isBlank(key)) {
            return null;
        }
        return DigestUtils.md5DigestAsHex(key.getBytes());
    }
}

4、代码实现

4.1Service

//注册功能
    @Override
    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.queryByName(user.getUsername());
        if (u != null) {
            map.put("usernameMsg","该账号已存在");
        }
        //验证邮箱
        u = userMapper.queryByEmail(user.getEmail());
        if (u != null) {
            map.put("emailMsg","该邮箱已被注册");
        }

        //注册用户
        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.addUser(user);

        //激活邮件
        Context context = new Context();
        context.setVariable("email", user.getEmail());
        //http://localhost:8080/community/activation/userId/ActivationCode
        String url = domain + contextPath + "/activation/" + user.getId() + "/" + user.getActivationCode();
        context.setVariable("url",url);
        String content = templateEngine.process("/mail/activation", context);
        mailClient.sendMail(user.getEmail(), "激活账号", content);

        //成功时map返回空
        return map;
    }
}

4.2 Controller

@PostMapping("/register")
    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("passwoedMsg", map.get("passwordMsg"));
            model.addAttribute("emailMsg", map.get("emailMsg"));
            return "/site/register";
        }
    }

4.3 前端实现

5、实现激活功能

5.1 定义激活状态常量

/**
 * Created by Monologue_zsj Luna on 2020/11/13 9:53
 * Description：激活状态常量声明
 */
public interface CommunityConstant {

    /**
     * 激活成功
     */
    int ACTIVATION_SUCCESS = 0;

    /**
     * 重复激活
     */
    int ACTIVATION_REPEAT = 1;

    /**
     * 激活失败
     */
    int ACTIVATION_FAILURE = 2;
}

5.2 Service激活业务处理

    /**
     * 使用常量需要实现CommunityConstant接口   
     * 激活状态处理
     * @param userId
     * @param activationCode
     * @return
     */
    public int activation(int userId, String activationCode) {

        User user = userMapper.queryById(userId);
        if (user.getStatus() == 1) {
            return ACTIVATION_REPEAT;
        }else if (user.getActivationCode().equals(activationCode)) {
            userMapper.updateStatus(userId, 1);
            return ACTIVATION_SUCCESS;
        }else {
            return ACTIVATION_FAILURE;
        }
    }

5.2 Controller处理

    /**
     * http://localhost:8080/community/activation/userId/activationCode
     */
    @GetMapping("/activation/{userId}/{activationCode}")
    public String activation(Model model, @PathVariable("userId") int userId, @PathVariable("activationCode") String activationCode) {

        int result = userService.activation(userId, activationCode);
        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";
    }

6、实现效果