    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    }