001    /** 
002     * 
003     * Copyright 2004 Protique Ltd
004     * 
005     * Licensed under the Apache License, Version 2.0 (the "License"); 
006     * you may not use this file except in compliance with the License. 
007     * You may obtain a copy of the License at 
008     * 
009     * http://www.apache.org/licenses/LICENSE-2.0
010     * 
011     * Unless required by applicable law or agreed to in writing, software
012     * distributed under the License is distributed on an "AS IS" BASIS, 
013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
014     * See the License for the specific language governing permissions and 
015     * limitations under the License. 
016     * 
017     **/
018    package org.activemq.util;
019    
020    import org.activemq.store.PersistenceAdapter;
021    
022    import javax.jms.JMSException;
023    
024    /**
025     * A helper class for running code with a PersistenceAdapter
026     * in a transaction.
027     *
028     * @version $Revision: 1.1.1.1 $
029     */
030    public class TransactionTemplate {
031        private PersistenceAdapter persistenceAdapter;
032    
033        public TransactionTemplate(PersistenceAdapter persistenceAdapter) {
034            this.persistenceAdapter = persistenceAdapter;
035        }
036    
037        public void run(Callback task) throws JMSException {
038            persistenceAdapter.beginTransaction();
039            Throwable throwable = null;
040            try {
041                task.execute();
042            }
043            catch (JMSException t) {
044                throwable = t;
045                throw t;
046            }
047            catch (RuntimeException t) {
048                throwable = t;
049                throw t;
050            }
051            catch (Throwable t) {
052                throwable = t;
053                throw JMSExceptionHelper.newJMSException("Persistence task failed: "+t, t);
054            } finally {
055                if (throwable == null) {
056                    persistenceAdapter.commitTransaction();
057                }
058                else {
059                    persistenceAdapter.rollbackTransaction();
060                }            
061            }
062        }
063    }