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.spring;
022    
023    import java.util.HashMap;
024    import java.util.Map;
025    
026    import org.granite.tide.data.TideSynchronizationManager;
027    import org.granite.util.TypeUtil;
028    import org.springframework.orm.jpa.EntityManagerHolder;
029    
030    /**
031     * Responsible for registering a publishing synchronization on the JPA entity manager
032     * Delegated to Hibernate-specific workarounds when necessary
033     * @author william
034     */
035    public class JPASynchronizationManager implements TideSynchronizationManager {
036            
037            private Map<String, TideSynchronizationManager> syncsMap = new HashMap<String, TideSynchronizationManager>();
038            
039            public JPASynchronizationManager() {
040                    try {
041                            syncsMap.put("org.hibernate.impl.SessionImpl", TypeUtil.newInstance("org.granite.tide.spring.Hibernate3SynchronizationManager", TideSynchronizationManager.class));
042                    }
043                    catch (Throwable t) {
044                            // Hibernate 3 not found
045                    }
046                    try {
047                            syncsMap.put("org.hibernate.internal.SessionImpl", TypeUtil.newInstance("org.granite.tide.spring.Hibernate4SynchronizationManager", TideSynchronizationManager.class));
048                    }
049                    catch (Throwable t) {
050                            // Hibernate 4 not found
051                    }
052            }
053            
054            public boolean registerSynchronization(Object resource, boolean shouldRemoveContextAtEnd) {
055                    EntityManagerHolder entityManagerHolder = (EntityManagerHolder)resource;
056                    Object delegate = entityManagerHolder.getEntityManager().getDelegate();
057                    if (syncsMap.containsKey(delegate.getClass().getName()))
058                            return syncsMap.get(delegate.getClass().getName()).registerSynchronization(delegate, shouldRemoveContextAtEnd);
059                    
060                    return false;
061            }
062    }