1   /**
    2    *  Licensed to the Apache Software Foundation (ASF) under one or more
    3    *  contributor license agreements.  See the NOTICE file distributed with
    4    *  this work for additional information regarding copyright ownership.
    5    *  The ASF licenses this file to You under the Apache License, Version 2.0
    6    *  (the "License"); you may not use this file except in compliance with
    7    *  the License.  You may obtain a copy of the License at
    8    *
    9    *     http://www.apache.org/licenses/LICENSE-2.0
   10    *
   11    *  Unless required by applicable law or agreed to in writing, software
   12    *  distributed under the License is distributed on an "AS IS" BASIS,
   13    *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   14    *  See the License for the specific language governing permissions and
   15    *  limitations under the License.
   16    */
   17   package org.apache.geronimo.concurrent.impl.handlers;
   18   
   19   import java.util.Map;
   20   import java.util.Set;
   21   
   22   import javax.transaction.TransactionManager;
   23   
   24   import org.apache.geronimo.gbean.AbstractName;
   25   import org.apache.geronimo.gbean.AbstractNameQuery;
   26   import org.apache.geronimo.kernel.Kernel;
   27   import org.apache.geronimo.kernel.KernelRegistry;
   28   
   29   /*
   30    * Geronimo-specific managed context handler that for handling transactions. 
   31    */
   32   public class TransactionContextHandler extends org.apache.geronimo.concurrent.handlers.TransactionContextHandler {
   33   
   34       private static TransactionManager transactionManager;
   35           
   36       public static TransactionManager getTransactionManager() {
   37           if (transactionManager == null) {
   38               try {
   39                   Kernel kernel = KernelRegistry.getSingleKernel();
   40                   AbstractNameQuery query = new AbstractNameQuery(TransactionManager.class.getName());
   41                   Set gbeans = kernel.listGBeans(query);
   42                   if (gbeans != null && !gbeans.isEmpty()) {
   43                       AbstractName name = (AbstractName)gbeans.iterator().next();
   44                       transactionManager = (TransactionManager)kernel.getGBean(name); 
   45                       LOG.debug("TransactionManager: " + transactionManager);
   46                   } else {
   47                       LOG.debug("TransactionManager not found");
   48                   }
   49               } catch (Exception e) {
   50                   LOG.warn("Failed to find TransactionManager", e);
   51               }
   52           }
   53           return transactionManager;
   54       }
   55      
   56       public void saveContext(Map<String, Object> context) {
   57           context.put(TRANSACTION_MANAGER, getTransactionManager());
   58       }
   59       
   60   }