Java tutorial
/* * Copyright (c) 2013, OpenCloudDB/MyCAT and/or its affiliates. All rights reserved. * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * This code is free software;Designed and Developed mainly by many Chinese * opensource volunteers. you can redistribute it and/or modify it under the * terms of the GNU General Public License version 2 only, as published by the * Free Software Foundation. * * This code is distributed in the hope that it will be useful, but WITHOUT * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * version 2 for more details (a copy is included in the LICENSE file that * accompanied this code). * * You should have received a copy of the GNU General Public License version * 2 along with this work; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. * * Any questions about this component can be directed to it's project Web address * https://code.google.com/p/opencloudb/. * */ package org.hx.rainbow.common.web.session; import java.util.Arrays; import java.util.Enumeration; import java.util.List; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpSession; import org.apache.commons.lang.StringUtils; import org.hx.rainbow.common.security.login.RainbowUser; import org.hx.rainbow.common.util.PropertiesUtil; public class RainbowSession implements java.io.Serializable { private static final long serialVersionUID = 8242424936268432231L; private static final String THREAD_LOACL_FILE = "threadLoacl"; private static final List<String> SESSION_KEYS = Arrays.asList(ThreadConstants.RAINBOW_USER, ThreadConstants.RAINBOW_USERNAME, ThreadConstants.RAINBOW_LOGINID); private static final ThreadLocal<Map<String, Object>> threadLocal = new ThreadLocal<Map<String, Object>>(); private static Map<String, Object> getSessionData() { Map<String, Object> map = threadLocal.get(); if (map == null) { map = new ConcurrentHashMap<String, Object>(); threadLocal.set(map); } return map; } private static Object getProperty(String keyName) { Map<String, Object> map = getSessionData(); return map.get(keyName); } private static void setProperty(String keyName, Object value) { if (keyName != null && value != null) { Map<String, Object> map = getSessionData(); map.put(keyName, value); } } public static Object getUserName() { return (String) getProperty(ThreadConstants.RAINBOW_USERNAME); } public static String getLoginId() { return (String) getProperty(ThreadConstants.RAINBOW_LOGINID); } public static HttpServletRequest getHttpRequest() { return (HttpServletRequest) getProperty(ThreadConstants.RAINBOW_REQUEST); } public static String getClientIp() { return (String) getProperty(ThreadConstants.CONSTMER_IPADDRESS); } public static String getClientHost() { return (String) getProperty(ThreadConstants.CONSTMER_HOST); } public static String getClientPort() { return (String) getProperty(ThreadConstants.CONSTMER_PORT); } public static String getServiceIP() { return (String) getProperty(ThreadConstants.SERVICE_IPADDRESS); } public static String getServiceHost() { return (String) getProperty(ThreadConstants.SERVICE_HOST); } public static RainbowUser getUser() { return (RainbowUser) getProperty(ThreadConstants.RAINBOW_SESSION); } public static void web2Service(HttpServletRequest request) { if (request == null) { return; } setProperty(ThreadConstants.CONSTMER_IPADDRESS, request.getRemoteAddr()); setProperty(ThreadConstants.CONSTMER_HOST, request.getRemoteHost()); setProperty(ThreadConstants.CONSTMER_PORT, request.getRemotePort()); setProperty(ThreadConstants.SERVICE_IPADDRESS, request.getLocalAddr()); setProperty(ThreadConstants.SERVICE_HOST, request.getLocalName()); setProperty(ThreadConstants.RAINBOW_REQUEST, request); HttpSession session = request.getSession(); RainbowUser rainbowUser = (RainbowUser) session.getAttribute(ThreadConstants.RAINBOW_USER); if (rainbowUser != null) { setProperty(ThreadConstants.RAINBOW_SESSION, rainbowUser); setProperty(ThreadConstants.RAINBOW_LOGINID, rainbowUser.getUsername()); setProperty(ThreadConstants.RAINBOW_USERNAME, rainbowUser.getSessionData().get("name")); String sessionKeys = (String) PropertiesUtil.getInstance().read(THREAD_LOACL_FILE) .get(ThreadConstants.SESSION_KEYS); if (sessionKeys != null) { if (sessionKeys.equals("*")) { Enumeration<String> attrNames = session.getAttributeNames(); while (attrNames.hasMoreElements()) { String attrName = (String) attrNames.nextElement(); if (SESSION_KEYS.contains(attrName)) { continue; } if (attrName != null) { Object session_attr = session.getAttribute(attrName); if (session_attr != null) { rainbowUser.getSessionData().put(attrName, session_attr); } } } } else { String[] s_sessionkey = StringUtils.split(sessionKeys, ","); for (int i = 0; i < s_sessionkey.length; i++) { if (s_sessionkey[i] != null) { Object session_attr = session.getAttribute(s_sessionkey[i]); if (session_attr != null) { rainbowUser.getSessionData().put(s_sessionkey[i], session_attr); } } } } } } // // // // // Map<String, Object> inCookie = new ConcurrentHashMap<String, Object>(); // String cookieKeys = (String)PropertiesUtil.get(ThreadConstants.resource_cookieKeys); // // if (cookieKeys != null) { // Cookie[] cookies = request.getCookies(); // if (cookies != null) { // if (cookieKeys.equals("*")) { // for (int i = 0; i < cookies.length; i++) { // Cookie cookie = cookies[i]; // String cookieName = cookie.getName(); // String cookieValue = cookie.getValue(); // if(cookieName != null && cookieValue != null){ // inCookie.put(cookieName, cookieValue); // } // } // } else { // cookieKeys = cookieKeys + ","; // for (int i = 0; i < cookies.length; i++) { // Cookie cookie = cookies[i]; // String cookieName = cookie.getName(); // if (cookieKeys.indexOf(cookieName + ",") > -1) { // String cookieValue = cookie.getValue(); // if(cookieName != null && cookieValue != null){ // inCookie.put(cookieName, cookieValue); // } // } // } // } // } // setProperty(ThreadConstants.IN_COOKIE, inCookie); // } } public static void service2Web(HttpServletRequest request) { if (request == null) { return; } request = getHttpRequest(); } }