最近從項目分離出來的注冊郵箱激活功能,整理一下,方便下次使用
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
2
3
4
5