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 org.granite.tide.data.DataContext;
024import org.granite.tide.data.DataEnabled.PublishMode;
025import org.granite.tide.data.TideSynchronizationManager;
026import org.hibernate.Session;
027import org.hibernate.action.spi.BeforeTransactionCompletionProcess;
028import org.hibernate.engine.spi.SessionImplementor;
029import org.hibernate.internal.SessionImpl;
030import org.springframework.orm.hibernate4.SessionHolder;
031
032/**
033 * Responsible for registering a publishing queued process on the Hibernate 4 session
034 * @author william
035 */
036public 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}