/*
   Copyright 2013, 2016 Nationale-Nederlanden

   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 nl.nn.adapterframework.ejb;

import nl.nn.adapterframework.core.IAdapter;
import nl.nn.adapterframework.core.IListener;
import nl.nn.adapterframework.core.IPortConnectedListener;
import nl.nn.adapterframework.receivers.GenericReceiver;
import nl.nn.adapterframework.util.LogUtil;

import org.apache.log4j.Logger;
import org.springframework.jndi.JndiLookupFailureException;

/**
 *
 * @author Tim van der Leeuw
 */
abstract public class AbstractListenerConnectingEJB extends AbstractEJBBase {
	protected Logger log = LogUtil.getLogger(this);

	/**
	 * This value is set in the EJB Create method
	 */
	protected boolean containerManagedTransactions;

	protected IPortConnectedListener listener;
	protected ListenerPortPoller listenerPortPoller;

	/**
	 * Common code to be executed when an EJB is created which is derived
	 * from this abstract class.
	 */
	protected void onEjbCreate() {
		this.listener = retrieveListener();
		this.containerManagedTransactions = retrieveTransactionType();
		log.info("onEjbCreate: Connected to Listener [" + listener.getName() + "]");
	}

	/**
	 * Common code to be executed when an EJB is created which is derived
	 * from this abstract class.
	 */
	protected void onEjbRemove() {
		listenerPortPoller.unregisterEjbListenerPortConnector(
				listener.getListenerPortConnector());
	}

	protected boolean retrieveTransactionType() {
		try {
			Boolean txType = (Boolean) getContextVariable("containerTransactions");
			if (txType == null) {
				log.warn("Value of variable 'containerTransactions' in Bean JNDI context is null, assuming bean-managed transactions");
				return false;
			} else {
				return txType.booleanValue();
			}
		} catch (JndiLookupFailureException e) {
			log.error("Cannot look up variable 'containerTransactions' in Bean JNDI context; assuming bean-managed transactions", e);
			return false;
		}
	}

	protected IListener retrieveListener(String receiverName, String adapterName) {
		IAdapter adapter = ibisManager.getRegisteredAdapter(adapterName);
		GenericReceiver receiver = (GenericReceiver) adapter.getReceiverByName(receiverName);
		return receiver.getListener();
	}

	protected void rollbackTransaction() throws IllegalStateException {
		if (containerManagedTransactions) {
			getEJBContext().setRollbackOnly();
		} else {
			try {
				getEJBContext().getUserTransaction().setRollbackOnly();
			} catch (Exception ex) {
				log.error("Cannot roll back user-transactions, must be using container-managed transactions without being properly configured for it?", ex);
				// Try the container-maanged way
				try {
					getEJBContext().setRollbackOnly();
				} catch (IllegalStateException e) {
					log.error("After failing to rolll back user-transaction, also failing to roll back container-transaction.", e);
				}
				throw new IllegalStateException("Cannot roll back user-transaction; must be using container-managed transactions? Error-message: ["
						+ ex.getMessage() + "]");
			}
		}
	}

	public ListenerPortPoller getListenerPortPoller() {
		return listenerPortPoller;
	}

	public void setListenerPortPoller(ListenerPortPoller listenerPortPoller) {
		this.listenerPortPoller = listenerPortPoller;
	}

	protected IPortConnectedListener retrieveListener() {
		String adapterName = (String) getContextVariable("adapterName");
		String receiverName = (String) getContextVariable("receiverName");
		return (IPortConnectedListener) retrieveListener(receiverName, adapterName);
	}

}