001    // Copyright 2007, 2008 The Apache Software Foundation
002    //
003    // Licensed under the Apache License, Version 2.0 (the "License");
004    // you may not use this file except in compliance with the License.
005    // You may obtain a copy of the License at
006    //
007    //     http://www.apache.org/licenses/LICENSE-2.0
008    //
009    // Unless required by applicable law or agreed to in writing, software
010    // distributed under the License is distributed on an "AS IS" BASIS,
011    // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
012    // See the License for the specific language governing permissions and
013    // limitations under the License.
014    
015    package org.apache.tapestry5.internal.hibernate;
016    
017    import org.apache.tapestry5.hibernate.HibernateSessionManager;
018    import org.apache.tapestry5.hibernate.HibernateSessionSource;
019    import org.apache.tapestry5.ioc.services.ThreadCleanupListener;
020    import org.hibernate.Session;
021    import org.hibernate.Transaction;
022    
023    public class HibernateSessionManagerImpl implements HibernateSessionManager, ThreadCleanupListener
024    {
025        private final Session session;
026    
027        private Transaction transaction;
028    
029        public HibernateSessionManagerImpl(HibernateSessionSource source)
030        {
031            session = source.create();
032    
033            startNewTransaction();
034        }
035    
036        private void startNewTransaction()
037        {
038            transaction = session.beginTransaction();
039        }
040    
041        public void abort()
042        {
043            transaction.rollback();
044            startNewTransaction();
045        }
046    
047        public void commit()
048        {
049            transaction.commit();
050            startNewTransaction();
051        }
052    
053        public Session getSession()
054        {
055            return session;
056        }
057    
058        /**
059         * Rollsback the transaction at the end of the request, then closes the session. This means that any uncommitted
060         * changes are lost; code should inject the HSM and invoke {@link #commit()} after making any changes, if they
061         * should persist.
062         */
063        public void threadDidCleanup()
064        {
065            transaction.rollback();
066    
067            session.close();
068        }
069    }