牛客网社区项目（开发注册功能）

项目比较复杂的时候，需要进行换一个思路，对于问题进行拆开，功能主要是分为三个方面：第一个是访问注册页面，第二个是提交注册数据，第三个是激活注册账号。还是按照三层的逻辑进行。进一步凝练是：

访问注册页面，
点击顶部区域内的链接，打开注册页面。
提交注册数据
通过表单提交数据。
服务端验证账号是否已存在、邮箱是否已注册。
服务端发送激活邮件。
激活注册账号
点击邮件中的链接，访问服务端的激活服务。

访问注册页面

在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 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 register(User user){
        Map 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为空，就返回注册成功的页面.
最后重新启动一下启动类，注册成功！

牛客网社区项目（开发注册功能）

Java相关栏目本月热门文章