/*

   * Hibernate, Relational Persistence for Idiomatic Java

   *

   * Copyright (c) 2011, Red Hat Inc. or third-party contributors as

   * indicated by the @author tags or express copyright attribution

   * statements applied by the authors.  All third-party contributions are

   * distributed under license by Red Hat Inc.

   *

   * This copyrighted material is made available to anyone wishing to use, modify,

  * copy, or redistribute it subject to the terms and conditions of the GNU

  * Lesser General Public License, as published by the Free Software Foundation.

  *

  * This program 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 Lesser General Public License

  * for more details.

  *

  * You should have received a copy of the GNU Lesser General Public License

  * along with this distribution; if not, write to:

  * Free Software Foundation, Inc.

  * 51 Franklin Street, Fifth Floor

  * Boston, MA  02110-1301  USA

  */

 package org.hibernate.service.jta.platform.internal;

 import org.hibernate.cfg.AvailableSettings;

 import org.hibernate.internal.util.config.ConfigurationHelper;

 import org.hibernate.service.jndi.spi.JndiService;

 import org.hibernate.service.jta.platform.spi.JtaPlatform;

 import org.hibernate.service.spi.Configurable;

 import org.hibernate.service.ServiceRegistry;

 import org.hibernate.service.spi.ServiceRegistryAwareService;

 import org.hibernate.service.spi.ServiceRegistryImplementor;

 import javax.transaction.Synchronization;

 import javax.transaction.SystemException;

 import javax.transaction.Transaction;

 import javax.transaction.TransactionManager;

 import javax.transaction.UserTransaction;

 import java.util.Map;

 public abstract class AbstractJtaPlatform

 		implements JtaPlatform, Configurable, ServiceRegistryAwareService, TransactionManagerAccess {

 	private boolean cacheTransactionManager;

 	private boolean cacheUserTransaction;

 	private ServiceRegistryImplementor serviceRegistry;

 	private final JtaSynchronizationStrategy tmSynchronizationStrategy = new TransactionManagerBasedSynchronizationStrategy( this );

 	@Override

 	public void injectServices(ServiceRegistryImplementor serviceRegistry) {

 		this.serviceRegistry = serviceRegistry;

 	}

 	protected ServiceRegistry serviceRegistry() {

 		return serviceRegistry;

 	}

 	protected JndiService jndiService() {

 		return serviceRegistry().getService( JndiService.class );

 	}

 	protected abstract TransactionManager locateTransactionManager();

 	protected abstract UserTransaction locateUserTransaction();

 	public void configure(Map configValues) {

 		cacheTransactionManager = ConfigurationHelper.getBoolean( AvailableSettings.JTA_CACHE_TM, configValues, true );

 		cacheUserTransaction = ConfigurationHelper.getBoolean( AvailableSettings.JTA_CACHE_UT, configValues, false );

 	}

 	protected boolean canCacheTransactionManager() {

 		return cacheTransactionManager;

 	}

 	protected boolean canCacheUserTransaction() {

 		return cacheUserTransaction;

 	}

 	private TransactionManager transactionManager;

 	@Override

 	public TransactionManager retrieveTransactionManager() {

 		if ( canCacheTransactionManager() ) {

 			if ( transactionManager == null ) {

 				transactionManager = locateTransactionManager();

 			}

 			return transactionManager;

 		}

 		else {

 			return locateTransactionManager();

 		}

 	}

 	@Override

 	public TransactionManager getTransactionManager() {

 		return retrieveTransactionManager();

	}

	private UserTransaction userTransaction;

	@Override

	public UserTransaction retrieveUserTransaction() {

		if ( canCacheUserTransaction() ) {

			if ( userTransaction == null ) {

				userTransaction = locateUserTransaction();

			}

			return userTransaction;

		}

		return locateUserTransaction();

	}

	@Override

	public Object getTransactionIdentifier(Transaction transaction) {

		// generally we use the transaction itself.

		return transaction;

	}

	protected JtaSynchronizationStrategy getSynchronizationStrategy() {

		return tmSynchronizationStrategy;

	}

	@Override

	public void registerSynchronization(Synchronization synchronization) {

		getSynchronizationStrategy().registerSynchronization( synchronization );

	}

	@Override

	public boolean canRegisterSynchronization() {

		return getSynchronizationStrategy().canRegisterSynchronization();

	}

	@Override

	public int getCurrentStatus() throws SystemException {

		return retrieveTransactionManager().getStatus();

	}

}