001    /*
002      GRANITE DATA SERVICES
003      Copyright (C) 2011 GRANITE DATA SERVICES S.A.S.
004    
005      This file is part of Granite Data Services.
006    
007      Granite Data Services is free software; you can redistribute it and/or modify
008      it under the terms of the GNU Library General Public License as published by
009      the Free Software Foundation; either version 2 of the License, or (at your
010      option) any later version.
011    
012      Granite Data Services is distributed in the hope that it will be useful, but
013      WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
014      FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public License
015      for more details.
016    
017      You should have received a copy of the GNU Library General Public License
018      along with this library; if not, see <http://www.gnu.org/licenses/>.
019    */
020    
021    package org.granite.tide.data;
022    
023    import javax.persistence.EntityTransaction;
024    
025    
026    /**
027     * Responsible for attaching a entity with the entity mangager
028     * @author cingram
029     *
030     */
031    public class JPATransactionManager extends JPALocalTransactionManager {
032            
033            private JTATransactionManager jtaTm = new JTATransactionManager();
034            
035            
036            @Override
037            public Object begin(TideTransactionPersistenceManager pm) {
038                    Object tx = null;
039                    try {
040                            tx = super.begin(pm);
041                    }
042                    catch (IllegalStateException e) {
043                            tx = jtaTm.begin(pm);                   
044                    }
045                    return tx;
046            }
047            
048            @Override
049            public void commit(Object tx) throws Exception {
050                    if (tx instanceof EntityTransaction)
051                            super.commit(tx);
052                    else
053                            jtaTm.commit(tx);
054            }
055            
056            @Override
057            public void rollback(Object tx) {
058                    if (tx instanceof EntityTransaction)
059                            super.rollback(tx);
060                    else
061                            jtaTm.rollback(tx);
062            }
063    }