所有项目几乎都能用到获取User的类

来源：互联网发布：软件开发工程师要求编辑：程序博客网时间：2024/06/01 09:25

package com.oa.util;

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


}