Example of UnitOfWork

UnitOfWork.cs

/* UnitOfWork using example

using( var context = new UnitOfWork( "db connection string here" ) )
{
    var customer = await context.Use<ICustomerRepository>()
                                .GetBySerial( serial );

} */


using Myapp.Core.Util;
using System;
using System.Collections.Generic;
using System.Data;
using System.Data.SqlClient;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Unity;
using Unity.Resolution;

namespace Myapp.Core.Data
{
    public class UnitOfWork : IDisposable
    {
        private bool _disposed = false;
        private IDictionary<string, object> _repositories;

        public IDbConnection Connection { get; private set; }
        public IDbTransaction Transaction { get; private set; }

        public UnitOfWork( string connString )
        {
            _repositories = new Dictionary<string, object>();

            Connection = new SqlConnection( connString );
            Connection.Open();

            Transaction = Connection.BeginTransaction();
        }

        public T Use<T>() where T : class
        {
            var typeName = typeof( T ).Name;

            if( !_repositories.Keys.Contains( typeName ) )
            {
                 var repository = UnityConfig.Container
                                            .Resolve<T>( new ResolverOverride[]
                                            {
                                                new ParameterOverride( "context", this )
                                            } );

                _repositories.Add( typeName, repository );
            }

            return (T)_repositories[typeName];
        }
           
        public void SaveChanges()
        {
            if( Transaction == null )
            {
                throw new InvalidOperationException( "Transaction have already been commited. Check your transaction handling." );
            }

            Transaction.Commit();
            Transaction = null;
        }

        public void Dispose()
        {
            Dispose( true );
            GC.SuppressFinalize( this );
        }

        protected virtual void Dispose( bool disposing )
        {
            if( _disposed )
                return;

            if( disposing )
            {
                if( Transaction != null )
                {
                    Transaction.Rollback();
                    Transaction = null;
                }

                if( Connection != null )
                {
                    Connection.Close();
                    Connection = null;
                }

                // Free any other managed objects here.
                //
            }

            // Free any unmanaged objects here.
            //
            _disposed = true;
        }

        ~UnitOfWork()
        {
            Dispose( false );
        }
    }
}