br.com.anteros.persistence.session.context.ManagedSQLSessionContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of Anteros-Persistence Show documentation

Anteros Persistence for Java.

There is a newer version: 1.0.38

/*******************************************************************************
 * Copyright 2012 Anteros Tecnologia
 *  
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *  
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *******************************************************************************/
package br.com.anteros.persistence.session.context;

import java.util.HashMap;
import java.util.Map;

import br.com.anteros.persistence.session.SQLSession;
import br.com.anteros.persistence.session.SQLSessionFactory;
import br.com.anteros.persistence.session.exception.SQLSessionException;

public class ManagedSQLSessionContext implements CurrentSQLSessionContext {

	private static final long serialVersionUID = 1L;

	private static final ThreadLocal> context = new ThreadLocal>();
	
	private final SQLSessionFactory factory;

	public ManagedSQLSessionContext(SQLSessionFactory factory) {
		this.factory = factory;
	}

	public SQLSession currentSession() {
		SQLSession current = existingSession( factory );
		if ( current == null ) {
			throw new SQLSessionException( "No session currently bound to execution context" );
		}
		return current;
	}

	public static boolean hasBind(SQLSessionFactory factory) {
		return existingSession( factory ) != null;
	}

	public static SQLSession bind(SQLSession session) {
		return sessionMap( true ).put( session.getSQLSessionFactory(), session );
	}

	public static SQLSession unbind(SQLSessionFactory factory) {
		SQLSession existing = null;
		Map sessionMap = sessionMap();
		if ( sessionMap != null ) {
			existing = sessionMap.remove( factory );
			doCleanup();
		}
		return existing;
	}

	private static SQLSession existingSession(SQLSessionFactory factory) {
		Map sessionMap = sessionMap();
		if ( sessionMap == null ) {
			return null;
		}
		else {
			return sessionMap.get( factory );
		}
	}

	protected static Map sessionMap() {
		return sessionMap( false );
	}

	private static synchronized Map sessionMap(boolean createMap) {
		Map sessionMap = context.get();
		if ( sessionMap == null && createMap ) {
			sessionMap = new HashMap();
			context.set( sessionMap );
		}
		return sessionMap;
	}

	private static synchronized void doCleanup() {
		Map sessionMap = sessionMap( false );
		if ( sessionMap != null ) {
			if ( sessionMap.isEmpty() ) {
				context.set( null );
			}
		}
	}
}