dapper_unitofwork/UnitOfWork.cs at master · matthewblott/dapper_unitofwork · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
namespace dapper_unitofwork
{
  using System;
  using System.Data;
  using Repositories.Implementations;
  using Repositories.Interfaces;

  public class UnitOfWork : IUnitOfWork
  {
    private IDbConnection _connection;
    private IDbTransaction _transaction;
    private ICategoryRepository _categoryRepository;
    private ICustomerRepository _customerRepository;
    private IEmployeeRepository _employeeRepository;
    private IOrderRepository _orderRepository;
    private IProductRepository _productRepository;
    private IRegionRepository _regionRepository;
    private IShipperRepository _shipperRepository;
    private ISupplierRepository _supplierRepository;
    private ITerritoryRepository _territoryRepository;

    private bool _disposed;

    public UnitOfWork(IDbConnection connection)
    {
      _connection = connection;
      _connection.Open();
      _transaction = _connection.BeginTransaction();
    }

    public ICategoryRepository CategoryRepository => _categoryRepository ??= new CategoryRepository(_transaction);
    public ICustomerRepository CustomerRepository => _customerRepository ??= new CustomerRepository(_transaction);
    public IEmployeeRepository EmployeeRepository  => _employeeRepository ??= new EmployeeRepository(_transaction);
    public IOrderRepository OrderRepository => _orderRepository ??= new OrderRepository(_transaction);
    public IProductRepository ProductRepository => _productRepository ??= new ProductRepository(_transaction);
    public IRegionRepository RegionRepository => _regionRepository ??= new RegionRepository(_transaction);
    public IShipperRepository ShipperRepository => _shipperRepository ??= new ShipperRepository(_transaction);
    public ISupplierRepository SupplierRepository => _supplierRepository ??= new SupplierRepository(_transaction);
    public ITerritoryRepository TerritoryRepository => _territoryRepository ??= new TerritoryRepository(_transaction);

    public void Commit()
    {
      try
      {
        _transaction.Commit();
      }
      catch
      {
        _transaction.Rollback();
        throw;
      }
      finally
      {
        _transaction.Dispose();
        _transaction = _connection.BeginTransaction();
        ResetRepositories();
      }

    }

    private void ResetRepositories()
    {
      _categoryRepository = null;
      _customerRepository = null;
      _employeeRepository = null;
      _orderRepository = null;
      _productRepository = null;
      _regionRepository = null;
      _shipperRepository = null;
      _supplierRepository = null;
      _territoryRepository = null;
    }

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

    private void Dispose(bool disposing)
    {
      if (_disposed)
      {
        return;
      }

      if (disposing)
      {
        if (_transaction != null)
        {
          _transaction.Dispose();
          _transaction = null;
        }
        if (_connection != null)
        {
          _connection.Dispose();
          _connection = null;
        }
      }
      _disposed = true;

    }

    ~UnitOfWork()
    {
      Dispose(false);
    }

  }

}