Package org.jboss.resource.connectionmanager

Source Code of org.jboss.resource.connectionmanager.TransactionSynchronizer

/*
* JBoss, Home of Professional Open Source.
* Copyright 2006, Red Hat Middleware LLC, and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software 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 software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.resource.connectionmanager;

import java.util.ArrayList;

import javax.transaction.RollbackException;
import javax.transaction.Synchronization;
import javax.transaction.SystemException;
import javax.transaction.Transaction;
import javax.transaction.TransactionManager;

import org.jboss.logging.Logger;
import org.jboss.tm.TransactionLocal;
import org.jboss.util.NestedRuntimeException;

/**
* Organizes transaction synchronization done by JCA.<p>
*
* This class exists to make sure all Tx synchronizations
* are invoked before the cached connection manager closes any
* closed connections.
*
* @author <a href="mailto:adrian@jboss.org">Adrian Brock</a>
* @version $Revision: 89154 $
*/
public class TransactionSynchronizer implements Synchronization
{
   /** The logger */
   private static final Logger log = Logger.getLogger(TransactionSynchronizer.class);

   /** The transaction synchronizations */
   protected static TransactionLocal txSynchs;
  
   /** The transaction */
   protected Transaction tx;
  
   /** The enlisting thread */
   protected Thread enlistingThread;
  
   /** Unenlisted */
   protected ArrayList unenlisted;
  
   /** Enlisted */
   protected ArrayList enlisted;
  
   /** The cached connection manager synchronization */
   protected Synchronization ccmSynch;

   /** Initialization */
   public static void setTransactionManager(TransactionManager tm)
   {
      txSynchs = new TransactionLocal(tm);
   }
  
   /**
    * Create a new transaction synchronizer
    *
    * @param tx the transaction to synchronize with
    */
   private TransactionSynchronizer(Transaction tx)
   {
      this.tx = tx;
   }
  
   /**
    * Add a new Tx synchronization that has not been enlisted
    *
    * @param synch the synchronization
    */
   synchronized void addUnenlisted(Synchronization synch)
   {
      if (unenlisted == null)
         unenlisted = new ArrayList();
      unenlisted.add(synch);
   }
  
   /**
    * Get the unenlisted synchronizations
    * and say we are enlisting if some are returned.
    *
    * @return the unenlisted synchronizations
    */
   synchronized ArrayList getUnenlisted()
   {
      Thread currentThread = Thread.currentThread();
      boolean interrupted = false;
      try
      {
         while (enlistingThread != null && enlistingThread != currentThread)
         {
            try
            {
               wait();
            }
            catch (InterruptedException e)
            {
               interrupted = true;
            }
         }
         ArrayList result = unenlisted;
         unenlisted = null;
         if (result != null)
            enlistingThread = currentThread;
         return result;
      }
      finally
      {
         if (interrupted)
            currentThread.interrupt();
      }
   }
  
   /**
    * The synchronization is now enlisted
    *
    * @param synch the synchronization
    */
   synchronized void addEnlisted(Synchronization synch)
   {
      if (enlisted == null)
         enlisted = new ArrayList();
      enlisted.add(synch);
   }
  
   /**
    * Remove an enlisted synchronization
    *
    * @param synch the synchronization
    * @return true when the synchronization was enlisted
    */
   synchronized boolean removeEnlisted(Synchronization synch)
   {
      return enlisted.remove(synch);
   }
  
   /**
    * This thread has finished enlisting
    */
   synchronized void enlisted()
   {
      Thread currentThread = Thread.currentThread();
      if (enlistingThread == null || enlistingThread != currentThread)
      {
         log.warn("Thread " + currentThread + " not the enlisting thread " + enlistingThread, new Exception("STACKTRACE"));
         return;
      }
      enlistingThread = null;
      notifyAll();
   }
  
   /**
    * Get a registered transaction synchronizer.
    *
    * @param tx the transaction
    * @return the registered transaction synchronizer for this transaction
    * @throws RolledbackException if the transaction is already rolled back
    * @throws SystemException for an error in the tranaction manager
    */
   static TransactionSynchronizer getRegisteredSynchronizer(Transaction tx) throws RollbackException, SystemException
   {
      TransactionSynchronizer result = (TransactionSynchronizer) txSynchs.get(tx);
      if (result == null)
      {
         result = new TransactionSynchronizer(tx);
         tx.registerSynchronization(result);
         txSynchs.set(tx, result);
      }
      return result;
   }
  
   /**
    * Check whether we have a CCM synchronization
    *
    * @param tx the transaction
    */
   static Synchronization getCCMSynchronization(Transaction tx)
   {
      TransactionSynchronizer ts = (TransactionSynchronizer) txSynchs.get(tx);
      if (ts != null)
         return ts.ccmSynch;
      else
         return null;
   }
  
   /**
    * Register a new CCM synchronization
    *
    * @param tx the transaction
    * @param synch the synchronization
    * @throws RolledbackException if the transaction is already rolled back
    * @throws SystemException for an error in the tranaction manager
    */
   static void registerCCMSynchronization(Transaction tx, Synchronization synch) throws RollbackException, SystemException
   {
      TransactionSynchronizer ts = getRegisteredSynchronizer(tx);
      ts.ccmSynch = synch;
   }
  
   /**
    * Lock for the given transaction
    *
    * @param tx the transaction
    */
   static void lock(Transaction tx)
   {
      try
      {
         txSynchs.lock(tx);
      }
      catch (InterruptedException e)
      {
         Thread.currentThread().interrupt();
         throw new NestedRuntimeException("Unable to get synchronization", e);
      }
   }
  
   /**
    * Unlock for the given transaction
    *
    * @param tx the transaction
    */
   static void unlock(Transaction tx)
   {
      txSynchs.unlock(tx);
   }

   public void beforeCompletion()
   {
      if (enlisted != null)
      {
         int i = 0;
         while (i < enlisted.size())
         {
            Synchronization synch = (Synchronization) enlisted.get(i);
            invokeBefore(synch);
            ++i;
         }
      }
     
      if (ccmSynch != null)
         invokeBefore(ccmSynch);
   }

   public void afterCompletion(int status)
   {
      if (enlisted != null)
      {
         int i = 0;
         while (i < enlisted.size())
         {
            Synchronization synch = (Synchronization) enlisted.get(i);
            invokeAfter(synch, status);
            ++i;
         }
      }
     
      if (ccmSynch != null)
         invokeAfter(ccmSynch, status);
   }

   /**
    * Invoke a beforeCompletion
    *
    * @param synch the synchronization
    */
   protected void invokeBefore(Synchronization synch)
   {
      try
      {
         synch.beforeCompletion();
      }
      catch (Throwable t)
      {
         log.warn("Transaction " + tx + " error in before completion " + synch, t);
      }
   }

   /**
    * Invoke an afterCompletion
    *
    * @param synch the synchronization
    * @param status the status of the transaction
    */
   protected void invokeAfter(Synchronization synch, int status)
   {
      try
      {
         synch.afterCompletion(status);
      }
      catch (Throwable t)
      {
         log.warn("Transaction " + tx + " error in after completion " + synch, t);
      }
   }
}
TOP

Related Classes of org.jboss.resource.connectionmanager.TransactionSynchronizer

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.