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 org.granite.tide.data.DataContext;
024    import org.granite.tide.data.DataEnabled.PublishMode;
025    import org.granite.tide.data.TideSynchronizationManager;
026    import org.hibernate.Session;
027    import org.hibernate.action.spi.BeforeTransactionCompletionProcess;
028    import org.hibernate.engine.spi.SessionImplementor;
029    import org.hibernate.internal.SessionImpl;
030    import org.springframework.orm.hibernate4.SessionHolder;
031    
032    /**
033     * Responsible for registering a publishing queued process on the Hibernate 4 session
034     * @author william
035     */
036    public class Hibernate4SynchronizationManager implements TideSynchronizationManager {
037            
038            public boolean registerSynchronization(Object resource, boolean shouldRemoveContextAtEnd) {
039                    Session session = null;
040                    if (resource instanceof Session)
041                            session = (Session)resource;
042                    else if (resource instanceof SessionHolder)
043                            session = ((SessionHolder)resource).getSession();
044                    else
045                            throw new IllegalArgumentException("Unknow resource for registering synchronization " + resource);
046                    
047                    ((SessionImpl)session).getActionQueue().registerProcess(new HibernateDataPublishingProcess(shouldRemoveContextAtEnd));
048                    return true;
049            }
050            
051            private static class HibernateDataPublishingProcess implements BeforeTransactionCompletionProcess {
052                    
053                    private boolean removeContext;
054                    
055                    public HibernateDataPublishingProcess(boolean removeContext) {
056                            this.removeContext = removeContext;
057                    }
058                    
059                    public void doBeforeTransactionCompletion(SessionImplementor session) {
060                            DataContext.publish(PublishMode.ON_COMMIT);
061                            if (removeContext)
062                                    DataContext.remove();
063                    }
064            }
065    
066    }