用戶注冊及登錄功能

• 注冊思路：填寫注冊表單，向用戶發送郵箱，點擊郵箱中的激活鏈接激活賬號，提示激活成功並跳轉至首頁。

核心代碼如下，若存在不合理信息，則將錯誤提示封裝至map中。在controller中通過判斷返回的map是否為空，以及map中的值判斷數據和合理性，並返回給界面，顯示錯誤信息。map不為空時，生成激活碼，作為激活鏈接中的參數，並驗證激活碼是否正確。service核心方法如下。

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);

        // 激活郵件
        Context context = new Context();
        context.setVariable("email", user.getEmail());
        // http://localhost:8080/community/activation/101/code
        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);

        return map;
    }

 

通過JavaMailSender包實現了發送郵件功能時，有個細節要注意，配置類中的password不是郵箱密碼，而是郵箱授權碼！

• 登錄思路：通過Kaptcha實現驗證碼的生成，通過圖片流將圖片返回給瀏覽器，將驗證碼數字存入Session。登錄時進行驗證。新建LoginTicket類用來存儲用戶id，登錄憑證號(UUID)，目的是將uuid存入Cookie，方便其他模塊獲取當前登錄用戶信息。

  @RequestMapping(path = "/login", method = RequestMethod.POST)
      public String login(String username, String password, String code, boolean rememberme,
                          Model model, HttpSession session, HttpServletResponse response) {
          // 檢查驗證碼
          String kaptcha = (String) session.getAttribute("kaptcha");
          if (StringUtils.isBlank(kaptcha) || StringUtils.isBlank(code) || !kaptcha.equalsIgnoreCase(code)) {
              model.addAttribute("codeMsg", "驗證碼不正確!");
              return "/site/login";
          }
  
          // 檢查賬號,密碼
          int expiredSeconds = rememberme ? REMEMBER_EXPIRED_SECONDS : DEFAULT_EXPIRED_SECONDS;
          Map<String, Object> map = userService.login(username, password, expiredSeconds);
          if (map.containsKey("ticket")) {
              Cookie cookie = new Cookie("ticket", map.get("ticket").toString());
              cookie.setPath(contextPath);
              cookie.setMaxAge(expiredSeconds);
              response.addCookie(cookie);
              return "redirect:/index";
          } else {
              model.addAttribute("usernameMsg", map.get("usernameMsg"));
              model.addAttribute("passwordMsg", map.get("passwordMsg"));
              return "/site/login";
          }
      }

  public Map<String, Object> login(String username, String password, int expiredSeconds) {
          Map<String, Object> map = new HashMap<>();
  
          // 空值處理
          if (StringUtils.isBlank(username)) {
              map.put("usernameMsg", "賬號不能為空!");
              return map;
          }
          if (StringUtils.isBlank(password)) {
              map.put("passwordMsg", "密碼不能為空!");
              return map;
          }
  
          // 驗證賬號
          User user = userMapper.selectByName(username);
          if (user == null) {
              map.put("usernameMsg", "該賬號不存在!");
              return map;
          }
  
          // 驗證狀態
          if (user.getStatus() == 0) {
              map.put("usernameMsg", "該賬號未激活!");
              return map;
          }
  
          // 驗證密碼
          password = CommunityUtil.md5(password + user.getSalt());
          if (!user.getPassword().equals(password)) {
              map.put("passwordMsg", "密碼不正確!");
              return map;
          }
  
          // 生成登錄憑證
          LoginTicket loginTicket = new LoginTicket();
          loginTicket.setUserId(user.getId());
          loginTicket.setTicket(CommunityUtil.generateUUID());
          loginTicket.setStatus(0);
          loginTicket.setExpired(new Date(System.currentTimeMillis() + expiredSeconds * 1000));
          loginTicketMapper.insertLoginTicket(loginTicket);
  
          map.put("ticket", loginTicket.getTicket());
          return map;
      }