FreeSql/FreeSql.DbContext/Repository/ContextSet/RepositoryDbContext.cs

using System;
using System.Collections.Concurrent;
using System.Reflection;
using System.Threading.Tasks;

namespace FreeSql {
	internal class RepositoryDbContext : DbContext {

		protected IBaseRepository _repos;
		public RepositoryDbContext(IFreeSql orm, IBaseRepository repos) : base() {
			_orm = orm;
			_repos = repos;
			_isUseUnitOfWork = false;
			_uowPriv = _repos.UnitOfWork;
		}


		static ConcurrentDictionary<Type, FieldInfo> _dicGetRepositoryDbField = new ConcurrentDictionary<Type, FieldInfo>();
		static FieldInfo GetRepositoryDbField(Type type) => _dicGetRepositoryDbField.GetOrAdd(type, tp => typeof(BaseRepository<,>).MakeGenericType(tp, typeof(int)).GetField("_dbPriv", BindingFlags.Instance | BindingFlags.NonPublic));
		public override IDbSet Set(Type entityType) {
			if (_dicSet.ContainsKey(entityType)) return _dicSet[entityType];

			var tb = _orm.CodeFirst.GetTableByEntity(entityType);
			if (tb == null) return null;

			object repos = _repos;
			if (entityType != _repos.EntityType) {
				repos = Activator.CreateInstance(typeof(DefaultRepository<,>).MakeGenericType(entityType, typeof(int)), _repos.Orm);
				(repos as IBaseRepository).UnitOfWork = _repos.UnitOfWork;
				GetRepositoryDbField(entityType).SetValue(repos, this);

				typeof(RepositoryDbContext).GetMethod("SetRepositoryDataFilter").MakeGenericMethod(_repos.EntityType)
					.Invoke(null, new object[] { repos, _repos });
			}

			var sd = Activator.CreateInstance(typeof(RepositoryDbSet<>).MakeGenericType(entityType), repos) as IDbSet;
			if (entityType != typeof(object)) _dicSet.Add(entityType, sd);
			return sd;
		}

		public static void SetRepositoryDataFilter<TEntity>(object repos, BaseRepository<TEntity> baseRepo) where TEntity : class {
			var filter = baseRepo.DataFilter as DataFilter<TEntity>;
			DataFilterUtil.SetRepositoryDataFilter(repos, fl => {
				foreach (var f in filter._filters)
					fl.Apply<TEntity>(f.Key, f.Value.Expression);
			});
		}

		public override int SaveChanges() {
			ExecCommand();
			var ret = _affrows;
			_affrows = 0;
			return ret;
		}
		async public override Task<int> SaveChangesAsync() {
			await ExecCommandAsync();
			var ret = _affrows;
			_affrows = 0;
			return ret;
		}
	}
}