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
021package org.granite.tide.spring;
022
023import java.util.HashMap;
024import java.util.Map;
025
026import org.granite.tide.data.TideSynchronizationManager;
027import org.granite.util.TypeUtil;
028import 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 */
035public 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}