qliberace/Common/qdr.fnd.core.data/Domain/DataDomain.cs

using System;
using System.Linq;
using System.Reflection;
using Microsoft.EntityFrameworkCore;
using Quadarax.Foundation.Core.Data.Interface.Domain;

namespace Quadarax.Foundation.Core.Data.Domain
{
    public abstract class DataDomain : DbContext, IDataDomain
    {
        private MethodInfo _mtSet;
        private MethodInfo _mtgSet;
        protected DataDomain(DbContextOptions options) : base(options)
        {

        }
        public TEntitySet GetDbSet<TEntitySet>()
        {
            //TODO: comment because db constext is in this scope static
            //if (_mtSet == null)
                _mtSet = GetType().GetMethod(nameof(DbContext.Set), Type.EmptyTypes);//GetType().GetMethod("Set");
            //if (_mtgSet==null)
                _mtgSet = _mtSet.MakeGenericMethod(typeof(TEntitySet).GetGenericArguments().First());

            var returnValue = _mtgSet.Invoke(this, new object[] {});
            return (TEntitySet)returnValue;
        }

        public IUnitOfWork<TDomainContext> CreateUnitOfWork<TDomainContext>() where TDomainContext : IDomain
        {
            return new UnitOfWork<TDomainContext>((IDomainContextResolver<TDomainContext>)new DomainContextResolver<DataDomain>(this));
        }


        public void Commit()
        {
            this.SaveChanges();
        }

        public void Rollback()
        {
            var changedEntries = ChangeTracker.Entries()
                .Where(x => x.State != EntityState.Unchanged).ToList();

            foreach (var entry in changedEntries)
            {
                switch(entry.State)
                {
                    case EntityState.Modified:
                        entry.CurrentValues.SetValues(entry.OriginalValues);
                        entry.State = EntityState.Unchanged;
                        break;
                    case EntityState.Added:
                        entry.State = EntityState.Detached;
                        break;
                    case EntityState.Deleted:
                        entry.State = EntityState.Unchanged;
                        break;
                }
            }
        }
    }     
}