SessionPoolConnector.java

package bc.connection;

import java.io.PrintStream;
import java.lang.reflect.Constructor;
import java.sql.Connection;
import java.util.HashMap;
import java.util.Set;

public class SessionPoolConnector {

	private void outError(Object information){
		System.out.print("SessionPoolConnector");
		System.out.print("  ERROR  ");
		System.out.println(information);
	}
	
	/** pool of Connection for user's [User Name, HibernateOracleConnect] (1:1) */
	private HashMap <String,PoolConnect> connectionUserPool=new HashMap<String,PoolConnect>();
	private UserSessionManager userSession=new UserSessionManager();
	private Class<? extends PoolConnect> connectorClass;
	private Integer poolSize=new Integer(10);
	
	/** îáúåêò, êîòîðûé ñîçäàåò Pool ñîåäèíåíèé ïî óêàçàííîìó êëèåíòó, à âîçâðàùàåò Connection ïî SessionId
	 * @param ïåðåäàåòñÿ êëàññ, êîòîðûé èìååò êîíñòðóêòîð ñëåäóþùåãî ñîäåðæàíèÿ 
	 * (String UserName, String Password, Integer poolSize)
	 * 
	 * 
	 * */
	public SessionPoolConnector(Class<? extends PoolConnect> connectorClass, Integer poolSize){
		this.connectorClass=connectorClass;
		if(poolSize!=null){
			this.poolSize=poolSize;
		}
	}
	

	/** ïîëó÷èòü Connection íà îñíîâàíèè ââåäåííûõ äàííûõ 
	 * @param userName èìÿ ïîëüçîâàòåëÿ
	 * @param password ïàðîëü
	 * @param sessionId óíèêàëüíûé íîìåð ñåññèè 
	 * @return Connection 
	 * @throws NoSuchMethodException 
	 * @throws SecurityException 
	 * */
	public Connection getConnection(String userName, 
								    String password, 
									String sessionId) {
		Connection returnValue;
		// ïðîâåðêà íàëè÷èÿ ñâÿçêè - "èìÿ ïîëüçîâàòåëÿ"-"POOL"
		if(connectionUserPool.containsKey(userName)==true){
			// Pool ñîçäàí ïî äàííîìó ïîëüçîâàòåëþ 
			if(connectionUserPool.get(userName).isPasswordEquals(password)){
				returnValue=connectionUserPool.get(userName).getConnection();
				this.userSession.put(userName, sessionId);
				if(returnValue==null){
					// âîçìîæíî ïðîèçîøëî èçìåíåíèå ïàðîëÿ - èëè æå âíåøíèå ôàêòîðû ïîâëèÿëè íà ñîåäèíåíèå (-ÿ)
					this.removeSessionByUser(userName);
					// ïîïûòàòüñÿ ñîçäàòü Pool ïî äàííîìó ïîëüçîâàòåëþ
					try{
						// ñîçäàòü îáúåêò è ïîëîæèòü åãî â õðàíèëèùå
						Constructor<? extends PoolConnect> constructor=this.connectorClass.getConstructor(String.class,String.class,Integer.class);
						PoolConnect newConnection=constructor.newInstance(userName, password, this.poolSize);
						// ïîëó÷èòü Connection èç óêàçàííîãî Pool
						returnValue=newConnection.getConnection();
						if(returnValue==null){
							throw new Exception();
						}
						this.connectionUserPool.put(userName, newConnection);
						this.userSession.put(userName, sessionId);
					}catch(Exception ex){
						returnValue=null;
						System.out.println("SessionPoolConnector Exception: "+ex.getMessage());
					}
				}
			}else{
				// Password is incorrect
				// âîçìîæíî ïðîèçîøëî èçìåíåíèå ïàðîëÿ - èëè æå âíåøíèå ôàêòîðû ïîâëèÿëè íà ñîåäèíåíèå (-ÿ)
				this.removeSessionByUser(userName);
				// ïîïûòàòüñÿ ñîçäàòü Pool ïî äàííîìó ïîëüçîâàòåëþ
				try{
					// ñîçäàòü îáúåêò è ïîëîæèòü åãî â õðàíèëèùå
					Constructor<? extends PoolConnect> constructor=this.connectorClass.getConstructor(String.class,String.class,Integer.class);
					PoolConnect newConnection=constructor.newInstance(userName, password, this.poolSize);
					// ïîëó÷èòü Connection èç óêàçàííîãî Pool
					returnValue=newConnection.getConnection();
					if(returnValue==null){
						throw new Exception();
					}
					this.connectionUserPool.put(userName, newConnection);
					this.userSession.put(userName, sessionId);
				}catch(Exception ex){
					returnValue=null;
					System.out.println("SessionPoolConnector Exception: "+ex.getMessage());
				}
			}
		}else{
			// Pool íå ñîçäàí ïî äàííîìó îáúåêòó
			try{
				// ñîçäàòü îáúåêò è ïîëîæèòü åãî â õðàíèëèùå
				Constructor<? extends PoolConnect> constructor=this.connectorClass.getConstructor(String.class,String.class,Integer.class);
				PoolConnect newConnection=constructor.newInstance(userName, password, this.poolSize);
				// ïîëó÷èòü Connection èç óêàçàííîãî Pool
				returnValue=newConnection.getConnection();
				if(returnValue==null){
					throw new Exception();
				}
				this.connectionUserPool.put(userName, newConnection);
				this.userSession.put(userName, sessionId);
			}catch(Exception ex){
				returnValue=null;
				System.out.println("SessionPoolConnector Exception: "+ex.getMessage());
			}
		}
		return returnValue;
	}

	/** âåðíóòü Connection â POOL */
	public void closeConnection(Connection connection){
		try{
			connection.close();
		}catch(Exception ex){
			outError("closeConnection dropped:"+ex.getMessage());
		}
	}

	/** Ïîëó÷èòü Connection èç POOL ïî óêàçàííîìó ñåññèîííîìó èäåíòèôèêàòîðó 
	 * @param sessionId - óíèêàëüíûé íîìåð ñåññèè, ïî êîòîðîìó íóæíî ïîëó÷èòü Connection
	 * */
	public Connection getConnection(String sessionId){
		String user=this.userSession.getUser(sessionId);
		
		if(user==null){
			return null;
		}else{
			return this.connectionUserPool.get(user).getConnection();
		}
	}

	public void removeSessionId(String sessionId){
		String user=this.userSession.getUser(sessionId);
		this.userSession.clear(sessionId);
		if(this.userSession.hasSessionsByUser(user)==false){
			try{
				this.connectionUserPool.get(user).reset();
			}catch(Exception ex){
				// user:Pool is not finded
			}
		}
	}
	
	/** óäàëèòü ïî óêàçàííîìó ïîëüçîâàòåëþ âñå ñîåäèíåíèÿ */
	public boolean removeSessionByUser(String userName){
		boolean returnValue=true;
		try{
			this.connectionUserPool.get(userName).reset();
		}catch(Exception ex){
			System.err.println("SessionPoolConnector#removeSessionByUser UserName:"+userName+"  close Exception:"+ex.getMessage());
			returnValue=false;
		}
		return returnValue;
	}
	
	/** ïîëó÷èòü êîë-âî ïîëüçîâàòåëåé, ïî êîòîðûì ñîçäàí PoolConnection */
	public int getUserPoolCount(){
		return this.connectionUserPool.size();
	}
	
	/** ïîëó÷èòü èìåíà ïîëüçîâàòåëåé, ïî êîòîðûì ñîçäàíû ïóëû */
	public Set<String> getUsersIntoPool(){
		return this.connectionUserPool.keySet();
	}
	
	/** ïîëó÷èòü ïî óêàçàííîìó ïîëüçîâàòåëþ êîë-âî îòêðûòûõ ñîåäèíåíèé */
	public int getConnectionCountByUser(String user){
		return this.connectionUserPool.get(user).getConnectionCount();
	}
	
	public void printConnection(PrintStream out){
		if(out==null){
			out=System.out;
		}
		out.println("Connection:");
		Set<String> users=getUsersIntoPool();
		for(String user:users){
			out.println("User:"+user+"    ConnectionCount:"+getConnectionCountByUser(user));
		}
		out.println("-----------");
	}
}