程式師世界 >> 編程語言 >> JAVA編程 >> JAVA編程入門知識 >> Java注冊郵箱激活驗證實現代碼

Java注冊郵箱激活驗證實現代碼

編輯：JAVA編程入門知識

最近從項目分離出來的注冊郵箱激活功能，整理一下，方便下次使用

RegisterValidateService.java

代碼如下：

package com.app.service.impl;

import java.text.ParseException;
import java.util.Date;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.app.dao.UserDao;
import com.app.tools.MD5Tool;
import com.app.tools.MD5Util;
import com.app.tools.SendEmail;
import com.app.tools.ServiceException;
import com.code.model.UserModel;

/**
*
* @author Qixuan.Chen
*/
@Service
public class RegisterValidateService {

    @Autowired
    private UserDao userDao;

    /**
     * 處理注冊
     */

    public void processregister(String email){
        UserModel user=new UserModel();
        Long as=5480l;
        user.setId(as);
        user.setName("xiaoming");
        user.setPassword("324545");
        user.setEmail(email);
        user.setRegisterTime(new Date());
        user.setStatus(0);
        ///如果處於安全，可以將激活碼處理的更復雜點，這裡我稍做簡單處理
        //user.setValidateCode(MD5Tool.MD5Encrypt(email));
        user.setValidateCode(MD5Util.encode2hex(email));

        userDao.save(user);//保存注冊信息

        ///郵件的內容
        StringBuffer sb=new StringBuffer("點擊下面鏈接激活賬號，48小時生效，否則重新注冊賬號，鏈接只能使用一次，請盡快激活！</br>");
        sb.append("<a href=\"http://localhost:8080/springmvc/user/register?action=activate&email=");
        sb.append(email);
        sb.append("&validateCode=");
        sb.append(user.getValidateCode());
        sb.append("\">http://localhost:8080/springmvc/user/register?action=activate&email=");
        sb.append(email);
        sb.append("&validateCode=");
        sb.append(user.getValidateCode());
        sb.append("</a>");

        //發送郵件
        SendEmail.send(email, sb.toString());
        System.out.println("發送郵件");

    }

    /**
     * 處理激活
     * @throws ParseException
     */
      ///傳遞激活碼和email過來
    public void processActivate(String email , String validateCode)throws ServiceException, ParseException{
         //數據訪問層，通過email獲取用戶信息
        UserModel user=userDao.find(email);
        //驗證用戶是否存在
        if(user!=null) {
            //驗證用戶激活狀態
            if(user.getStatus()==0) {
                ///沒激活
                Date currentTime = new Date();//獲取當前時間
                //驗證鏈接是否過期
                currentTime.before(user.getRegisterTime());
                if(currentTime.before(user.getLastActivateTime())) {
                    //驗證激活碼是否正確
                    if(validateCode.equals(user.getValidateCode())) {
                        //激活成功， //並更新用戶的激活狀態，為已激活
                        System.out.println("==sq==="+user.getStatus());
                        user.setStatus(1);//把狀態改為激活
                        System.out.println("==sh==="+user.getStatus());
                        userDao.update(user);
                    } else {
                       throw new ServiceException("激活碼不正確");
                    }
                } else { throw new ServiceException("激活碼已過期！");
                }
            } else {
               throw new ServiceException("郵箱已激活，請登錄！");
            }
        } else {
            throw new ServiceException("該郵箱未注冊（郵箱地址不存在）！");
        }

    }

}

RegisterController.java

代碼如下：

package com.app.web.controller;

import java.text.ParseException;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.servlet.ModelAndView;

import com.app.service.impl.RegisterValidateService;
import com.app.tools.ServiceException;

@Controller
public class RegisterController {

    @Resource
    private RegisterValidateService service;

    @RequestMapping(value="/user/register",method={RequestMethod.GET,RequestMethod.POST})
    public ModelAndView load(HttpServletRequest request,HttpServletResponse response) throws ParseException{
        String action = request.getParameter("action");
        System.out.println("-----r----"+action);
        ModelAndView mav=new ModelAndView();

        if("register".equals(action)) {
            //注冊
            String email = request.getParameter("email");
            service.processregister(email);//發郵箱激活
            mav.addObject("text","注冊成功");
            mav.setViewName("register/register_success");
        }
        else if("activate".equals(action)) {
            //激活
            String email = request.getParameter("email");//獲取email
            String validateCode = request.getParameter("validateCode");//激活碼

            try {
                service.processActivate(email , validateCode);//調用激活方法
                mav.setViewName("register/activate_success");
            } catch (ServiceException e) {
                request.setAttribute("message" , e.getMessage());
                mav.setViewName("register/activate_failure");
            }

        }
        return mav;
    }

}

UserDao.java（這裡個人沒有做入庫操作，只是利用集合，做過效果出來0_0）
代碼如下：

package com.app.dao;

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;

import org.springframework.stereotype.Repository;

import com.code.model.UserModel;

/**
*
* @author Qixuan.Chen
*/
@Repository
public class UserDao {

   public HashMap<String, String> map=new HashMap<String, String>();
    /**
     * @保存注冊信息
     * private Long id;
        private String name;
        private String password;
        private String email;//注冊賬號
        private int status;//激活狀態
        private String validateCode;//激活碼
        private Date registerTime;//注冊時間
     */
    public void save(UserModel user){
        System.out.println("cicicici");
        map.put("id", String.valueOf(user.getId()));
        map.put("email", user.getEmail());
        map.put("validateCode", user.getValidateCode());
        SimpleDateFormat sdf=new SimpleDateFormat("yyyyMMddhhmmss");
        String time=sdf.format(user.getRegisterTime());
        map.put("registerTime", time);
        int status=user.getStatus();
        map.put("status", String.valueOf(status));
        map.put("name", user.getName());
        String t2=sdf.format(user.getLastActivateTime());
        map.put("activeLastTime", String.valueOf(t2));
        System.out.println("=======s========="+status);

    }

    /**
     * @更新 user
     */
    public void update(UserModel user){
        map.put("email", user.getEmail());
        map.put("validateCode", user.getValidateCode());
        Date time=user.getRegisterTime();
        map.put("registerTime", String.valueOf(time));
        int status=user.getStatus();
        map.put("status", String.valueOf(status));
        System.out.println("=======st========="+status);
    }

    /**
     * @throws ParseException
     * @查找信息
     */
    public UserModel find(String email) throws ParseException{
        UserModel user=new UserModel();
        user.setEmail(map.get("email"));
        user.setName(map.get("name"));
        SimpleDateFormat sdf=new SimpleDateFormat("yyyyMMddhhmmss");
        Date day=sdf.parse(map.get("registerTime"));
        user.setRegisterTime(day);
        user.setStatus(Integer.valueOf(map.get("status")));
        user.setValidateCode(map.get("validateCode"));

return user;
}

}

UserModel.java
代碼如下：

package com.code.model;

import java.beans.Transient;
import java.util.Calendar;
import java.util.Date;

public class UserModel {
    private Long id;
private String name;
private String password;
private String email;//注冊賬號
private int status=0;//激活狀態
private String validateCode;//激活碼
private Date registerTime;//注冊時間


    /////////////////

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password;
    }

    public String getEmail() {
        return email;
    }

    public void setEmail(String email) {
        this.email = email;
    }

    public int getStatus() {
        return status;
    }

    public void setStatus(int status) {
        this.status = status;
    }

    public String getValidateCode() {
        return validateCode;
    }

    public void setValidateCode(String validateCode) {
        this.validateCode = validateCode;
    }

    public Date getRegisterTime() {
        return registerTime;
    }

    public void setRegisterTime(Date registerTime) {
        this.registerTime = registerTime;
    }

    /////////////////////////
    @Transient
    public Date getLastActivateTime() {
        Calendar cl = Calendar.getInstance();
        cl.setTime(registerTime);
        cl.add(Calendar.DATE , 2);

        return cl.getTime();
    }

}

SendEmail.java
代碼如下：

package com.app.tools;

import java.util.Date;
import java.util.Properties;

import javax.mail.Authenticator;
import javax.mail.Message;
import javax.mail.MessagingException;
import javax.mail.PasswordAuthentication;
import javax.mail.Session;
import javax.mail.Transport;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeMessage;

/**
*
* @author Qixuan.Chen
*/
public class SendEmail {

    public static final String HOST = "smtp.163.com";
    public static final String PROTOCOL = "smtp";
    public static final int PORT = 25;
    public static final String FROM = "[email protected]";//發件人的email
    public static final String PWD = "123456";//發件人密碼

    /**
     * 獲取Session
     * @return
     */
    private static Session getSession() {
        Properties props = new Properties();
        props.put("mail.smtp.host", HOST);//設置服務器地址
        props.put("mail.store.protocol" , PROTOCOL);//設置協議
        props.put("mail.smtp.port", PORT);//設置端口
        props.put("mail.smtp.auth" , true);

        Authenticator authenticator = new Authenticator() {

            @Override
            protected PasswordAuthentication getPasswordAuthentication() {
                return new PasswordAuthentication(FROM, PWD);
            }

        };
        Session session = Session.getDefaultInstance(props , authenticator);

        return session;
    }

    public static void send(String toEmail , String content) {
        Session session = getSession();
        try {
            System.out.println("--send--"+content);
            // Instantiate a message
            Message msg = new MimeMessage(session);

            //Set message attributes
            msg.setFrom(new InternetAddress(FROM));
            InternetAddress[] address = {new InternetAddress(toEmail)};
            msg.setRecipients(Message.RecipientType.TO, address);
            msg.setSubject("賬號激活郵件");
            msg.setSentDate(new Date());
            msg.setContent(content , "text/html;charset=utf-8");

            //Send the message
            Transport.send(msg);
        }
        catch (MessagingException mex) {
            mex.printStackTrace();
        }
    }

}

MD5Util.java
代碼如下：

package com.app.tools;

import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

public class MD5Util {

/**
* 將源字符串使用MD5加密為字節數組
* @param source
* @return
*/
public static byte[] encode2bytes(String source) {
  byte[] result = null;
  try {
   MessageDigest md = MessageDigest.getInstance("MD5");
   md.reset();
   md.update(source.getBytes("UTF-8"));
   result = md.digest();
  } catch (NoSuchAlgorithmException e) {
   e.printStackTrace();
  } catch (UnsupportedEncodingException e) {
   e.printStackTrace();
  }

  return result;
}

/**
* 將源字符串使用MD5加密為32位16進制數
* @param source
* @return
*/
public static String encode2hex(String source) {
  byte[] data = encode2bytes(source);

  StringBuffer hexString = new StringBuffer();
  for (int i = 0; i < data.length; i++) {
   String hex = Integer.toHexString(0xff & data[i]);

   if (hex.length() == 1) {
    hexString.append('0');
   }

   hexString.append(hex);
  }

  return hexString.toString();
}

/**
* 驗證字符串是否匹配
* @param unknown 待驗證的字符串
* @param okHex 使用MD5加密過的16進制字符串
* @return 匹配返回true，不匹配返回false
*/
public static boolean validate(String unknown , String okHex) {
  return okHex.equals(encode2hex(unknown));
}

}

ServiceException.java
代碼如下：

package com.app.tools;

public class ServiceException extends Exception {
private static final long serialVersionUID = -1708015121235851228L;

public ServiceException(String message) {
super(message);
}
}

jsp頁面

registerEmailValidae.jsp
代碼如下：

   <h2>注冊激活</h2>
    <form action="user/register?action=register" method="post">
        Email:<input type="text" id="email" name="email" value="" >
        <input type="submit" value="提交">
    </form>

register_success.jsp
代碼如下：

<body>
恭喜你注冊成功！請到注冊的郵箱點擊鏈接激活！
</body>

activate_success.jsp:
代碼如下：

<body>
賬號激活成功，點擊這裡去登錄！
</body>

activate_failure.jsp:

<body>
激活失敗！錯誤信息：${message }
</body>

效果圖：
1