PixelDragonsMVC.NET - An open source MVC framework

/***********************************************************
 * 
 * Copyright 2007 Pixel Dragons Ltd. All rights reserved.
 * 
 * http://www.codeplex.com/PixelDragonsMVC for licence, more
 * information and latest source code. 
 * 
 * Check out our other stuff at http://www.pixeldragons.com
 * 
 ***********************************************************/

using System;
using System.Web;
using System.Configuration;

using PixelDragons.MVC.Configuration;

using NHibernate;
using log4net;

namespace PixelDragons.MVC.Persistence
{
	public class SessionManager
	{
		#region Fields
        private static readonly ILog _logger = LogManager.GetLogger(typeof(SessionManager));
		private ISessionFactory _sessionFactory = null;
		private ISession _session = null;
		#endregion

        #region Constructors
        public SessionManager(MVCConfigurationHandler settings)
        {
            NHibernate.Cfg.Configuration config = new NHibernate.Cfg.Configuration();

            foreach (EntityAssembly assembly in settings.EntityAssemblies)
            {
                config.AddAssembly(assembly.Name);
            }

            _sessionFactory = config.BuildSessionFactory();
        }
        #endregion

		#region Properties
		public ISessionFactory SessionFactory
		{
			get { return _sessionFactory; }
			set { _sessionFactory = value; }
		}

        public ISession ActiveSession
        {
            get
            {
                if (HttpContext.Current != null)
                {
                    if (HttpContext.Current.Items["ActiveSession"] == null)
                    {
                        OpenActiveSession();
                    }
                    return (ISession)HttpContext.Current.Items["ActiveSession"];
                }
                else
                {
                    if (_session == null)
                    {
                        OpenActiveSession();
                    }
                    return _session;
                }
            }
        }

        public ISession TemporarySession
        {
            get
            {
                ISession tempSession = SessionFactory.OpenSession();
                return tempSession;
            }
        }
		#endregion

        #region Methods
        public ISession OpenSession()
		{
            return _sessionFactory.OpenSession();
		}

		public ISession OpenSession(IInterceptor interceptor)
		{
			if (interceptor != null)
			{
                return _sessionFactory.OpenSession(interceptor);
			}
			else
			{
                return _sessionFactory.OpenSession();
			}
		}	

		public void OpenActiveSession()
		{
			if (HttpContext.Current != null)
			{
				HttpContext.Current.Items["ActiveSession"] = OpenSession();
			}
			else
			{
				if (_session == null)
				{
					_session = OpenSession();
				}
			}
		}

		public void CloseActiveSession(TransactionManager transactionManager)
		{
			if (HttpContext.Current != null)
			{
				if (HttpContext.Current.Items["ActiveSession"] != null)
				{
					ISession activeSession = (ISession)HttpContext.Current.Items["ActiveSession"];
					if (activeSession != null && activeSession.IsOpen)
					{
						try
						{
                            foreach (ITransaction transaction in transactionManager.Transactions)
                            {
                                if (transaction != null && (!transaction.WasCommitted) && (!transaction.WasRolledBack))
                                {
                                    _logger.Debug("Committing transaction");
                                    transaction.Commit();
                                }
                            }

                            activeSession.Flush();
						}
						finally
						{
							activeSession.Close();
							HttpContext.Current.Items["ActiveSession"] = null;
						}
					}
					HttpContext.Current.Items["ActiveSession"] = null;
				}
			}
			else
			{
				if (_session != null)
				{
					if (_session != null && _session.IsOpen)
					{
						try
						{
                            foreach (ITransaction transaction in transactionManager.Transactions)
                            {
                                if (transaction != null && (!transaction.WasCommitted) && (!transaction.WasRolledBack))
                                {
                                    _logger.Debug("Committing transaction");
                                    transaction.Commit();
                                }
                            }

                            _session.Flush();                            
						}
						finally
						{
							_session.Close();
							_session = null;
						}
					}
					_session = null;
				}
			}
        }
        #endregion
    }
}