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
015package org.apache.tapestry5.internal.hibernate;
016
017import org.apache.tapestry5.hibernate.HibernateSessionManager;
018import org.apache.tapestry5.hibernate.HibernateSessionSource;
019import org.apache.tapestry5.ioc.services.ThreadCleanupListener;
020import org.hibernate.Session;
021import org.hibernate.Transaction;
022
023public 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}