import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import org.apache.struts2.ServletActionContext;
import com.oa.apps.limit.entity.User;
public class UserContext{
private static ThreadLocal<User> user = new ThreadLocal<User>();
public static HttpSession getSession(){
HttpServletRequest request = ServletActionContext.getRequest();
HttpSession session = request.getSession();
return session;
}
public static User getUser()
{
Object user = null;
HttpServletRequest request = ServletActionContext.getRequest();
if (getSession() != null){
Cookie cookie = CookieContext.getCookieByName(request, "OA_USER");
if(cookie!=null){
user = getSession().getAttribute(cookie.getValue());
}
}
if ((user != null) && (user instanceof User)){
return ((User)user);
}
return null;
}
public static void setSecurityUser(User user) {
HttpServletResponse response = ServletActionContext.getResponse();
HttpSession session = getSession();
session.setAttribute("OA_SECURITY_USER_"+user.getId(), user);
CookieContext.addCookie(response, "OA_USER", "OA_SECURITY_USER_"+user.getId(), 0);
CookieContext.addCookie(response, "USER_NAME", String.valueOf(user.getId()), 0);
}
public static boolean checkSecurityUser(User user) {
HttpServletRequest request = ServletActionContext.getRequest();
HttpSession session = getSession();
Cookie cookie = CookieContext.getCookieByName(request, "OA_USER");
if(cookie != null){
User u =(User)session.getAttribute(cookie.getValue());
if(u != null){
UserContext.user.set(u);
return true;
}
}
return false;
}
public static void removeUser() {
HttpServletRequest request = ServletActionContext.getRequest();
Cookie cookie = CookieContext.getCookieByName(request, "OA_USER");
getSession().removeAttribute(cookie.getValue());
}
}
设置cookie的类
package com.oa.util;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
public class CookieContext {
/**
* 设置cookie
* @param response
* @param name cookie名字
* @param value cookie值
* @param maxAge cookie生命周期 以秒为单位
*/
public static void addCookie(HttpServletResponse response,String name,String value,int maxAge){
Cookie cookie = new Cookie(name,value);
cookie.setPath("/");
if(maxAge>0) cookie.setMaxAge(maxAge);
response.addCookie(cookie);
}
/**
* 根据名字获取cookie
* @param request
* @param name cookie名字
* @return
*/
public static Cookie getCookieByName(HttpServletRequest request,String name){
Map<String,Cookie> cookieMap = ReadCookieMap(request);
if(cookieMap.containsKey(name)){
Cookie cookie = (Cookie)cookieMap.get(name);
return cookie;
}else{
return null;
}
}
/**
* 将cookie封装到Map里面
* @param request
* @return
*/
private static Map<String,Cookie> ReadCookieMap(HttpServletRequest request){
Map<String,Cookie> cookieMap = new HashMap<String,Cookie>();
Cookie[] cookies = request.getCookies();
if(null!=cookies){
for(Cookie cookie : cookies){
cookieMap.put(cookie.getName(), cookie);
}
}
return cookieMap;
}
}