com.alogic.auth.LocalSession Maven / Gradle / Ivy

Go to download
package com.alogic.auth;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import org.apache.commons.lang3.BooleanUtils;
import org.apache.commons.lang3.StringUtils;
import org.w3c.dom.Element;

import com.anysoft.util.JsonTools;
import com.anysoft.util.Pair;
import com.anysoft.util.StringMatcher;
import com.anysoft.util.XmlTools;
import com.alogic.load.Loadable;

/**
 * 本地服务的Session
 * 
 * 
 * LocalSession会在HttpSession中构造一个HashMap对象和一个HashSet对象，用于实现HashObject和SetObject接口。
 * 
 * @author duanyy
 * @since 1.6.10.10
 * 
 * @version 1.6.11.7 [20180107 duanyy] 

 * - 优化Session管理 

 *
 * @version 1.6.14.14 [20210715 duanyy] 

 *  - 登录时可以设置会话的ttl;
 */
public class LocalSession extends Loadable.Abstract implements Session{
	
	/**
	 * Session Id
	 */
	protected String id;	
	
	/**
	 * Session管理器
	 */
	protected SessionManager sm = null;
	
	/**
	 * Set组
	 */
	protected Map> setGroups = null;
	
	/**
	 * Hash组
	 */
	protected Map> hashGroups = null;
	
	public LocalSession(String id,SessionManager sm){
		this.id = id;
		this.sm = sm;
	}
	
	@Override
	public boolean isLoggedIn() {
		return BooleanUtils.toBoolean(hGet(DEFAULT_GROUP,LOGIN_KEY, "false"));
	}
	
	@Override
	public void setLoggedIn(boolean loggedIn){
		this.hSet(DEFAULT_GROUP,LOGIN_KEY, BooleanUtils.toStringTrueFalse(loggedIn), true);
	}

	@Override
	public void setLoggedIn(boolean loggedIn, long ttl) {
		this.hSet(DEFAULT_GROUP,LOGIN_KEY, BooleanUtils.toStringTrueFalse(loggedIn), true);
	}

	protected Map> getSetGroup(boolean create){
		if (setGroups == null && create){
			synchronized (this){
				if (setGroups == null){
					setGroups = new ConcurrentHashMap>();
				}
			}
		}
		
		return setGroups;
	}
	
	protected Map> getHashGroup(boolean create){
		if (hashGroups == null && create){
			synchronized (this){
				if (hashGroups == null){
					hashGroups = new ConcurrentHashMap>();
				}
			}
		}
		
		return hashGroups;
	}

	/**
	 * 获取内置的setObject
	 * 
	 * 

	 * 用于获取内置的setObject, 当该对象不存在时，如果create为true，则创建一个。
	 * 
	 * @param group 分组
	 * @param create 当不存在时是否创建
	 * @return 内置的setObject
	 */
	protected Set getSetObject(String group,boolean create){		
		Set setObject = null;
		
		Map> groups = getSetGroup(create);
		if (groups != null){
			setObject = groups.get(group);
			if (setObject == null && create){
				synchronized(this){
					setObject = groups.get(group);
					if (setObject == null){
						setObject = Collections.newSetFromMap(new ConcurrentHashMap());
						groups.put(group, setObject);
					}
				}
			}
		}
		return setObject;
	}
	
	/**
	 * 获取内置的mapObject
	 * 
	 * 
	 * 用于获取内置的mapObject,当该对象不存在时，如果create为true，则创建一个。
	 * 
	 * @param group 分组
	 * @param create 当不存在时是否创建
	 * @return 内置的mapObject
	 */
	protected  Map getMapObject(String group,boolean create){
		Map mapObject = null;
		
		Map> groups = getHashGroup(create);
		if (groups != null){
			mapObject = groups.get(group);
			if (mapObject == null){
				synchronized(this){
					mapObject = groups.get(group);
					if (mapObject == null){
						mapObject = new ConcurrentHashMap();
						groups.put(group, mapObject);
					}
				}
			}
		}
		return (Map) mapObject;
	}
	
	
	@Override
	public void sAdd(String group,String... members) {
		Set obj = getSetObject(group,true);
		for (String m:members){
			obj.add(m);
		}
	}

	@Override
	public void sDel(String group,String... members) {
		Set obj = getSetObject(group,false);
		if (obj != null){
			for (String m:members){
				obj.remove(m);
			}
		}
	}

	@Override
	public void sDel(String group){
		Set obj = getSetObject(group,false);
		if (obj != null){
			obj.clear();
		}
	}
	
	@Override
	public int sSize(String group) {
		Set obj = getSetObject(group,false);
		return obj == null ? 0 : obj.size();
	}
	
	private boolean isConditionValid(String condition){
		return StringUtils.isNotEmpty(condition) && !condition.equals("*");
	}

	@Override
	public List sMembers(String group,String condition) {
		List result = new ArrayList();
		Set obj = getSetObject(group,false);
		if (obj != null){
			if (isConditionValid(condition)){
				StringMatcher matcher = new StringMatcher(condition);			
				Iterator iter = obj.iterator();			
				while (iter.hasNext()){
					String member = iter.next();				
					if (matcher.match(member)){
						result.add(member);
					}
				}
			}else{
				Iterator iter = obj.iterator();			
				while (iter.hasNext()){
					result.add(iter.next());
				}
			}
		}
		return result;
	}

	@Override
	public boolean sExist(String group,String member) {
		Set obj = getSetObject(group,false);
		return obj != null ? obj.contains(member) : false;
	}

	@Override
	public String getId() {
		return id;
	}

	@Override
	public void expire() {
		super.expire();
		sm.delSession(getId());
	}

	@Override
	public void report(Element xml) {
		if (xml != null){
			XmlTools.setString(xml,"module",getClass().getName());
			XmlTools.setString(xml, "id", getId());
			XmlTools.setString(xml,"timestamp", String.valueOf(getTimestamp()));
		}
	}

	@Override
	public void report(Map json) {
		if (json != null){
			JsonTools.setString(json, "module", getClass().getName());
			JsonTools.setString(json, "id", getId());
			JsonTools.setLong(json, "timestamp", getTimestamp());
		}
	}

	@Override
	public String getValue(String varName, Object context, String defaultValue) {
		return getRawValue(varName,context,defaultValue);
	}

	@Override
	public String getRawValue(String varName, Object context, String dftValue) {
		@SuppressWarnings("unchecked")
		Map mapObject = context != null ?
				(Map)context
				:getMapObject(DEFAULT_GROUP,false);
				
		if (mapObject != null){
			String value = mapObject.get(varName);
			return StringUtils.isEmpty(value)?dftValue:value;
		}else{
			return dftValue;
		}
	}

	@Override
	public Object getContext(String varName) {
		return getMapObject(DEFAULT_GROUP,false);
	}

	@Override
	public void hSet(String group,String key, String value, boolean overwrite) {
		Map mapObject = getMapObject(group,true);
		if (mapObject.containsKey(key)){
			if (overwrite){
				mapObject.put(key, value);
			}
		}else{
			mapObject.put(key, value);
		}
	}

	@Override
	public String hGet(String group,String key, String dftValue) {
		Map mapObject = getMapObject(group,false);
		
		if (mapObject != null){
			String value = mapObject.get(key);
			return StringUtils.isEmpty(value)?dftValue:value;
		}else{
			return dftValue;
		}
	}

	@Override
	public boolean hExist(String group,String key) {
		Map mapObject = getMapObject(group,false);
		
		return mapObject == null ? false:mapObject.containsKey(key);
	}

	@Override
	public List> hGetAll(String group,String condition) {
		List> result = new ArrayList>();
		Map mapObject = getMapObject(group,false);
		
		if (mapObject != null){
			if (isConditionValid(condition)){
				StringMatcher matcher = new StringMatcher(condition);
				Iterator> iter = mapObject.entrySet().iterator();		
				while (iter.hasNext()){
					Entry entry = iter.next();
					if (matcher.match(entry.getKey())){
						result.add(new Pair.Default(entry.getKey(), entry.getValue()));
					}
				}
			}else{
				Iterator> iter = mapObject.entrySet().iterator();		
				while (iter.hasNext()){
					Entry entry = iter.next();
					result.add(new Pair.Default(entry.getKey(), entry.getValue()));
				}
			}
		}
		return result;
	}

	@Override
	public int hLen(String group) {
		Map mapObject = getMapObject(group,false);
		return mapObject == null ? 0 : mapObject.size();
	}

	@Override
	public List hKeys(String group,String condition) {
		List result = new ArrayList();
		Map mapObject = getMapObject(group,false);
		
		if (mapObject != null){
			if (isConditionValid(condition)){
				StringMatcher matcher = new StringMatcher(condition);
				Iterator iter = mapObject.keySet().iterator();
				while (iter.hasNext()){
					String key = iter.next();
					if (matcher.match(key)){
						result.add(key);
					}
				}
			}else{
				Iterator iter = mapObject.keySet().iterator();
				while (iter.hasNext()){
					result.add(iter.next());
				}
			}
		}
		return result;
	}

	@Override
	public void hDel(String group,String key) {
		Map mapObject = getMapObject(group,false);
		if (mapObject != null){
			mapObject.remove(key);
		}
	}

	@Override
	public void hDel(String group) {
		Map mapObject = getMapObject(group,false);
		if (mapObject != null){
			mapObject.clear();
		}
	}
}