tk/FreeSql
1
0
Fork 0
mirror of https://github.com/nsnail/FreeSql.git synced 2025-04-22 10:42:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

- 优化 DbContext/Repository 局部调整；

Browse Source
This commit is contained in:
28810 2019-08-13 19:13:48 +08:00
parent 79ab3ae217
commit a79ee52b4f
14 changed files with 211 additions and 197 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Examples/base_entity/Entities/User.cs
View File

@ -1,6 +1,6 @@
using System; using FreeSql;
using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Text;
public class UserGroup : BaseEntity<UserGroup, int> public class UserGroup : BaseEntity<UserGroup, int>
{ {

3
Examples/base_entity/Program.cs
View File

@ -1,4 +1,5 @@
using System; using FreeSql;
using System;
using System.Threading.Tasks; using System.Threading.Tasks;
namespace base_entity namespace base_entity

64
FreeSql.DbContext/DbContext/DbContext.cs
View File

@ -9,41 +9,51 @@ namespace FreeSql
{ {
public abstract partial class DbContext : IDisposable public abstract partial class DbContext : IDisposable
{ {
internal IFreeSql _ormPriv;
public IFreeSql Orm => _ormPriv ?? throw new ArgumentNullException("请在 OnConfiguring 或 AddFreeDbContext 中配置 UseFreeSql");
internal IFreeSql _orm; #region Property UnitOfWork
public IFreeSql Orm => _orm ?? throw new ArgumentNullException("请在 OnConfiguring 或 AddFreeDbContext 中配置 UseFreeSql"); internal bool _isUseUnitOfWork = true; //是否使用工作单元事务
IUnitOfWork _uowPriv;
protected IUnitOfWork _uowPriv; public IUnitOfWork UnitOfWork
internal IUnitOfWork _uow => _isUseUnitOfWork ? (_uowPriv ?? (_uowPriv = new UnitOfWork(Orm))) : null;
internal bool _isUseUnitOfWork = true; //不使用工作单元事务
public IUnitOfWork UnitOfWork => _uow;
DbContextOptions _options;
public DbContextOptions Options
{ {
set => _uowPriv = value;
get get
{ {
if (_options != null) return _options; if (_uowPriv != null) return _uowPriv;
if (FreeSqlDbContextExtenssions._dicSetDbContextOptions.TryGetValue(Orm, out _options)) return _options; if (_isUseUnitOfWork == false) return null;
_options = new DbContextOptions(); return _uowPriv = new UnitOfWork(Orm);
return _options;
} }
set => _options = value;
} }
#endregion
#region Property Options
internal DbContextOptions _optionsPriv;
public DbContextOptions Options
{
set => _optionsPriv = value;
get
{
if (_optionsPriv != null) return _optionsPriv;
if (FreeSqlDbContextExtenssions._dicSetDbContextOptions.TryGetValue(Orm, out _optionsPriv)) return _optionsPriv;
_optionsPriv = new DbContextOptions();
return _optionsPriv;
}
}
#endregion
protected DbContext() protected DbContext()
{ {
var builder = new DbContextOptionsBuilder(); var builder = new DbContextOptionsBuilder();
OnConfiguring(builder); OnConfiguring(builder);
_orm = builder._fsql; _ormPriv = builder._fsql;
_options = builder._options; _optionsPriv = builder._options;
if (_orm != null) InitPropSets(); if (_ormPriv != null) InitPropSets();
} }
protected virtual void OnConfiguring(DbContextOptionsBuilder builder) { } protected virtual void OnConfiguring(DbContextOptionsBuilder builder) { }
#region Set
static ConcurrentDictionary<Type, PropertyInfo[]> _dicGetDbSetProps = new ConcurrentDictionary<Type, PropertyInfo[]>(); static ConcurrentDictionary<Type, PropertyInfo[]> _dicGetDbSetProps = new ConcurrentDictionary<Type, PropertyInfo[]>();
internal void InitPropSets() internal void InitPropSets()
{ {
@ -61,16 +71,19 @@ namespace FreeSql
} }
} }
protected List<IDbSet> _listSet = new List<IDbSet>();
protected Dictionary<Type, IDbSet> _dicSet = new Dictionary<Type, IDbSet>(); protected Dictionary<Type, IDbSet> _dicSet = new Dictionary<Type, IDbSet>();
public DbSet<TEntity> Set<TEntity>() where TEntity : class => this.Set(typeof(TEntity)) as DbSet<TEntity>; public DbSet<TEntity> Set<TEntity>() where TEntity : class => this.Set(typeof(TEntity)) as DbSet<TEntity>;
public virtual IDbSet Set(Type entityType) public virtual IDbSet Set(Type entityType)
{ {
if (_dicSet.ContainsKey(entityType)) return _dicSet[entityType]; if (_dicSet.ContainsKey(entityType)) return _dicSet[entityType];
var sd = Activator.CreateInstance(typeof(DbContextDbSet<>).MakeGenericType(entityType), this) as IDbSet; var sd = Activator.CreateInstance(typeof(DbContextDbSet<>).MakeGenericType(entityType), this) as IDbSet;
_listSet.Add(sd);
if (entityType != typeof(object)) _dicSet.Add(entityType, sd); if (entityType != typeof(object)) _dicSet.Add(entityType, sd);
return sd; return sd;
} }
protected Dictionary<string, IDbSet> AllSets { get; } = new Dictionary<string, IDbSet>(); protected Dictionary<string, IDbSet> AllSets { get; } = new Dictionary<string, IDbSet>();
#endregion
#region DbSet 快速代理 #region DbSet 快速代理
/// <summary> /// <summary>
@ -118,6 +131,7 @@ namespace FreeSql
public void AttachRange<TEntity>(IEnumerable<TEntity> data) where TEntity : class => this.Set<TEntity>().AttachRange(data); public void AttachRange<TEntity>(IEnumerable<TEntity> data) where TEntity : class => this.Set<TEntity>().AttachRange(data);
#endregion #endregion
#region Queue Action
internal class ExecCommandInfo internal class ExecCommandInfo
{ {
public ExecCommandInfoType actionType { get; set; } public ExecCommandInfoType actionType { get; set; }
@ -131,31 +145,33 @@ namespace FreeSql
internal void EnqueueAction(ExecCommandInfoType actionType, IDbSet dbSet, Type stateType, object state) => internal void EnqueueAction(ExecCommandInfoType actionType, IDbSet dbSet, Type stateType, object state) =>
_actions.Enqueue(new ExecCommandInfo { actionType = actionType, dbSet = dbSet, stateType = stateType, state = state }); _actions.Enqueue(new ExecCommandInfo { actionType = actionType, dbSet = dbSet, stateType = stateType, state = state });
#endregion
~DbContext() => this.Dispose(); ~DbContext() => this.Dispose();
bool _isdisposed = false; bool _isdisposed = false;
public void Dispose() public void Dispose()
{ {
if (_isdisposed) return; if (_isdisposed) return;
_isdisposed = true;
try try
{ {
_actions.Clear(); _actions.Clear();
foreach (var set in _dicSet) foreach (var set in _listSet)
try try
{ {
set.Value.Dispose(); set.Dispose();
} }
catch { } catch { }
_listSet.Clear();
_dicSet.Clear(); _dicSet.Clear();
AllSets.Clear(); AllSets.Clear();
_uow?.Rollback(); UnitOfWork?.Rollback();
} }
finally finally
{ {
_isdisposed = true;
GC.SuppressFinalize(this); GC.SuppressFinalize(this);
} }
} }

2
FreeSql.DbContext/DbContext/DbContextAsync.cs
View File

@ -13,7 +13,7 @@ namespace FreeSql
async public virtual Task<int> SaveChangesAsync() async public virtual Task<int> SaveChangesAsync()
{ {
await ExecCommandAsync(); await ExecCommandAsync();
_uow?.Commit(); UnitOfWork?.Commit();
var ret = _affrows; var ret = _affrows;
_affrows = 0; _affrows = 0;
return ret; return ret;

2
FreeSql.DbContext/DbContext/DbContextSync.cs
View File

@ -12,7 +12,7 @@ namespace FreeSql
public virtual int SaveChanges() public virtual int SaveChanges()
{ {
ExecCommand(); ExecCommand();
_uow?.Commit(); UnitOfWork?.Commit();
var ret = _affrows; var ret = _affrows;
_affrows = 0; _affrows = 0;
return ret; return ret;

2
FreeSql.DbContext/DbContext/FreeContext.cs
View File

@ -7,7 +7,7 @@ namespace FreeSql
public FreeContext(IFreeSql orm) public FreeContext(IFreeSql orm)
{ {
_orm = orm; _ormPriv = orm;
} }
} }
} }

98
FreeSql.DbContext/DbSet/DbSet.cs
View File

@ -11,14 +11,12 @@ using System.Reflection;
namespace FreeSql namespace FreeSql
{ {
internal class DbContextDbSet<TEntity> : DbSet<TEntity> where TEntity : class class DbContextDbSet<TEntity> : DbSet<TEntity> where TEntity : class
{ {
public DbContextDbSet(DbContext ctx) public DbContextDbSet(DbContext ctx)
{ {
_ctx = ctx; _db = ctx;
_uow = ctx._uow; _uow = ctx.UnitOfWork;
_fsql = ctx.Orm;
} }
} }
@ -28,25 +26,21 @@ namespace FreeSql
} }
public abstract partial class DbSet<TEntity> : IDbSet where TEntity : class public abstract partial class DbSet<TEntity> : IDbSet where TEntity : class
{ {
internal DbContext _db { get; set; }
internal DbContext _ctx; internal IUnitOfWork _uow { get; set; }
internal IUnitOfWork _uow;
internal IFreeSql _fsql;
protected virtual ISelect<TEntity> OrmSelect(object dywhere) protected virtual ISelect<TEntity> OrmSelect(object dywhere)
{ {
DbContextExecCommand(); //查询前先提交，否则会出脏读 DbContextExecCommand(); //查询前先提交，否则会出脏读
return _fsql.Select<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction(false)).TrackToList(TrackToList).WhereDynamic(dywhere); return _db.Orm.Select<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction(false)).TrackToList(TrackToList).WhereDynamic(dywhere);
} }
~DbSet() ~DbSet() => this.Dispose();
{
this.Dispose();
}
bool _isdisposed = false; bool _isdisposed = false;
public void Dispose() public void Dispose()
{ {
if (_isdisposed) return; if (_isdisposed) return;
_isdisposed = true;
try try
{ {
this._dicUpdateTimes.Clear(); this._dicUpdateTimes.Clear();
@ -54,26 +48,22 @@ namespace FreeSql
} }
finally finally
{ {
_isdisposed = true;
GC.SuppressFinalize(this); GC.SuppressFinalize(this);
} }
} }
protected virtual IInsert<TEntity> OrmInsert() => _fsql.Insert<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()); protected virtual IInsert<TEntity> OrmInsert() => _db.Orm.Insert<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction());
protected virtual IInsert<TEntity> OrmInsert(TEntity data) => _fsql.Insert<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).AppendData(data); protected virtual IInsert<TEntity> OrmInsert(TEntity data) => _db.Orm.Insert<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).AppendData(data);
protected virtual IInsert<TEntity> OrmInsert(IEnumerable<TEntity> data) => _fsql.Insert<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).AppendData(data); protected virtual IInsert<TEntity> OrmInsert(IEnumerable<TEntity> data) => _db.Orm.Insert<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).AppendData(data);
protected virtual IUpdate<TEntity> OrmUpdate(IEnumerable<TEntity> entitys) => _fsql.Update<TEntity>().AsType(_entityType).SetSource(entitys).WithTransaction(_uow?.GetOrBeginTransaction()); protected virtual IUpdate<TEntity> OrmUpdate(IEnumerable<TEntity> entitys) => _db.Orm.Update<TEntity>().AsType(_entityType).SetSource(entitys).WithTransaction(_uow?.GetOrBeginTransaction());
protected virtual IDelete<TEntity> OrmDelete(object dywhere) => _fsql.Delete<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).WhereDynamic(dywhere); protected virtual IDelete<TEntity> OrmDelete(object dywhere) => _db.Orm.Delete<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).WhereDynamic(dywhere);
internal void EnqueueToDbContext(DbContext.ExecCommandInfoType actionType, EntityState state) internal void EnqueueToDbContext(DbContext.ExecCommandInfoType actionType, EntityState state) =>
{ _db.EnqueueAction(actionType, this, typeof(EntityState), state);
_ctx.EnqueueAction(actionType, this, typeof(EntityState), state);
} internal void IncrAffrows(int affrows) =>
internal void IncrAffrows(int affrows) _db._affrows += affrows;
{
_ctx._affrows += affrows;
}
internal void TrackToList(object list) internal void TrackToList(object list)
{ {
@ -89,7 +79,7 @@ namespace FreeSql
var itemType = item.GetType(); var itemType = item.GetType();
if (itemType == typeof(object)) return; if (itemType == typeof(object)) return;
if (itemType.FullName.StartsWith("Submission#")) itemType = itemType.BaseType; if (itemType.FullName.StartsWith("Submission#")) itemType = itemType.BaseType;
var dbset = _ctx.Set(itemType); var dbset = _db.Set(itemType);
dbset?.GetType().GetMethod("TrackToList", BindingFlags.Instance | BindingFlags.NonPublic).Invoke(dbset, new object[] { list }); dbset?.GetType().GetMethod("TrackToList", BindingFlags.Instance | BindingFlags.NonPublic).Invoke(dbset, new object[] { list });
return; return;
} }
@ -98,11 +88,11 @@ namespace FreeSql
foreach (var item in ls) foreach (var item in ls)
{ {
var key = _fsql.GetEntityKeyString(_entityType, item, false); var key = _db.Orm.GetEntityKeyString(_entityType, item, false);
if (key == null) continue; if (key == null) continue;
_states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) => _states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) =>
{ {
_fsql.MapEntityValue(_entityType, item, ov.Value); _db.Orm.MapEntityValue(_entityType, item, ov.Value);
ov.Time = DateTime.Now; ov.Time = DateTime.Now;
return ov; return ov;
}); });
@ -116,7 +106,7 @@ namespace FreeSql
protected ConcurrentDictionary<string, EntityState> _states = new ConcurrentDictionary<string, EntityState>(); protected ConcurrentDictionary<string, EntityState> _states = new ConcurrentDictionary<string, EntityState>();
internal ConcurrentDictionary<string, EntityState> _statesInternal => _states; internal ConcurrentDictionary<string, EntityState> _statesInternal => _states;
TableInfo _tablePriv; TableInfo _tablePriv;
protected TableInfo _table => _tablePriv ?? (_tablePriv = _fsql.CodeFirst.GetTableByEntity(_entityType)); protected TableInfo _table => _tablePriv ?? (_tablePriv = _db.Orm.CodeFirst.GetTableByEntity(_entityType));
ColumnInfo[] _tableIdentitysPriv; ColumnInfo[] _tableIdentitysPriv;
protected ColumnInfo[] _tableIdentitys => _tableIdentitysPriv ?? (_tableIdentitysPriv = _table.Primarys.Where(a => a.Attribute.IsIdentity).ToArray()); protected ColumnInfo[] _tableIdentitys => _tableIdentitysPriv ?? (_tableIdentitysPriv = _table.Primarys.Where(a => a.Attribute.IsIdentity).ToArray());
protected Type _entityType = typeof(TEntity); protected Type _entityType = typeof(TEntity);
@ -131,7 +121,7 @@ namespace FreeSql
{ {
if (entityType == typeof(object)) throw new Exception("ISelect.AsType 参数不支持指定为 object"); if (entityType == typeof(object)) throw new Exception("ISelect.AsType 参数不支持指定为 object");
if (entityType == _entityType) return; if (entityType == _entityType) return;
var newtb = _fsql.CodeFirst.GetTableByEntity(entityType); var newtb = _db.Orm.CodeFirst.GetTableByEntity(entityType);
_entityType = entityType; _entityType = entityType;
_tablePriv = newtb ?? throw new Exception("DbSet.AsType 参数错误，请传入正确的实体类型"); _tablePriv = newtb ?? throw new Exception("DbSet.AsType 参数错误，请传入正确的实体类型");
_tableIdentitysPriv = null; _tableIdentitysPriv = null;
@ -158,15 +148,15 @@ namespace FreeSql
public void AttachRange(IEnumerable<TEntity> data) public void AttachRange(IEnumerable<TEntity> data)
{ {
if (data == null || data.Any() == false) return; if (data == null || data.Any() == false) return;
if (_table.Primarys.Any() == false) throw new Exception($"不可附加，实体没有主键：{_fsql.GetEntityString(_entityType, data.First())}"); if (_table.Primarys.Any() == false) throw new Exception($"不可附加，实体没有主键：{_db.Orm.GetEntityString(_entityType, data.First())}");
foreach (var item in data) foreach (var item in data)
{ {
var key = _fsql.GetEntityKeyString(_entityType, item, false); var key = _db.Orm.GetEntityKeyString(_entityType, item, false);
if (string.IsNullOrEmpty(key)) throw new Exception($"不可附加，未设置主键的值：{_fsql.GetEntityString(_entityType, item)}"); if (string.IsNullOrEmpty(key)) throw new Exception($"不可附加，未设置主键的值：{_db.Orm.GetEntityString(_entityType, item)}");
_states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) => _states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) =>
{ {
_fsql.MapEntityValue(_entityType, item, ov.Value); _db.Orm.MapEntityValue(_entityType, item, ov.Value);
ov.Time = DateTime.Now; ov.Time = DateTime.Now;
return ov; return ov;
}); });
@ -184,15 +174,15 @@ namespace FreeSql
EntityState CreateEntityState(TEntity data) EntityState CreateEntityState(TEntity data)
{ {
if (data == null) throw new ArgumentNullException(nameof(data)); if (data == null) throw new ArgumentNullException(nameof(data));
var key = _fsql.GetEntityKeyString(_entityType, data, false); var key = _db.Orm.GetEntityKeyString(_entityType, data, false);
var state = new EntityState((TEntity)Activator.CreateInstance(_entityType), key); var state = new EntityState((TEntity)Activator.CreateInstance(_entityType), key);
_fsql.MapEntityValue(_entityType, data, state.Value); _db.Orm.MapEntityValue(_entityType, data, state.Value);
return state; return state;
} }
bool? ExistsInStates(TEntity data) bool? ExistsInStates(TEntity data)
{ {
if (data == null) throw new ArgumentNullException(nameof(data)); if (data == null) throw new ArgumentNullException(nameof(data));
var key = _fsql.GetEntityKeyString(_entityType, data, false); var key = _db.Orm.GetEntityKeyString(_entityType, data, false);
if (string.IsNullOrEmpty(key)) return null; if (string.IsNullOrEmpty(key)) return null;
return _states.ContainsKey(key); return _states.ContainsKey(key);
} }
@ -217,13 +207,13 @@ namespace FreeSql
} }
if (_table.Primarys.Any() == false) if (_table.Primarys.Any() == false)
{ {
if (isThrow) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可添加，实体没有主键：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
var key = _fsql.GetEntityKeyString(_entityType, data, true); var key = _db.Orm.GetEntityKeyString(_entityType, data, true);
if (string.IsNullOrEmpty(key)) if (string.IsNullOrEmpty(key))
{ {
switch (_fsql.Ado.DataType) switch (_db.Orm.Ado.DataType)
{ {
case DataType.SqlServer: case DataType.SqlServer:
case DataType.PostgreSQL: case DataType.PostgreSQL:
@ -235,7 +225,7 @@ namespace FreeSql
{ {
return true; return true;
} }
if (isThrow) throw new Exception($"不可添加，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可添加，未设置主键的值：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
} }
@ -243,13 +233,13 @@ namespace FreeSql
{ {
if (_states.ContainsKey(key)) if (_states.ContainsKey(key))
{ {
if (isThrow) throw new Exception($"不可添加，已存在于状态管理：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可添加，已存在于状态管理：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
var idval = _fsql.GetEntityIdentityValueWithPrimary(_entityType, data); var idval = _db.Orm.GetEntityIdentityValueWithPrimary(_entityType, data);
if (idval > 0) if (idval > 0)
{ {
if (isThrow) throw new Exception($"不可添加，自增属性有值：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可添加，自增属性有值：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
} }
@ -276,18 +266,18 @@ namespace FreeSql
} }
if (_table.Primarys.Any() == false) if (_table.Primarys.Any() == false)
{ {
if (isThrow) throw new Exception($"不可更新，实体没有主键：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可更新，实体没有主键：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
var key = _fsql.GetEntityKeyString(_entityType, data, false); var key = _db.Orm.GetEntityKeyString(_entityType, data, false);
if (string.IsNullOrEmpty(key)) if (string.IsNullOrEmpty(key))
{ {
if (isThrow) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可更新，未设置主键的值：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
if (_states.TryGetValue(key, out var tryval) == false) if (_states.TryGetValue(key, out var tryval) == false)
{ {
if (isThrow) throw new Exception($"不可更新，数据未被跟踪，应该先查询 或者 Attach：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可更新，数据未被跟踪，应该先查询 或者 Attach：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
return true; return true;
@ -313,13 +303,13 @@ namespace FreeSql
} }
if (_table.Primarys.Any() == false) if (_table.Primarys.Any() == false)
{ {
if (isThrow) throw new Exception($"不可删除，实体没有主键：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可删除，实体没有主键：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
var key = _fsql.GetEntityKeyString(_entityType, data, false); var key = _db.Orm.GetEntityKeyString(_entityType, data, false);
if (string.IsNullOrEmpty(key)) if (string.IsNullOrEmpty(key))
{ {
if (isThrow) throw new Exception($"不可删除，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}"); if (isThrow) throw new Exception($"不可删除，未设置主键的值：{_db.Orm.GetEntityString(_entityType, data)}");
return false; return false;
} }
//if (_states.TryGetValue(key, out var tryval) == false) { //if (_states.TryGetValue(key, out var tryval) == false) {

54
FreeSql.DbContext/DbSet/DbSetAsync.cs
View File

@ -14,7 +14,7 @@ namespace FreeSql
Task DbContextExecCommandAsync() Task DbContextExecCommandAsync()
{ {
_dicUpdateTimes.Clear(); _dicUpdateTimes.Clear();
return _ctx.ExecCommandAsync(); return _db.ExecCommandAsync();
} }
async Task<int> DbContextBetchAddAsync(EntityState[] adds) async Task<int> DbContextBetchAddAsync(EntityState[] adds)
@ -31,7 +31,7 @@ namespace FreeSql
if (_tableIdentitys.Length > 0) if (_tableIdentitys.Length > 0)
{ {
//有自增，马上执行 //有自增，马上执行
switch (_fsql.Ado.DataType) switch (_db.Orm.Ado.DataType)
{ {
case DataType.SqlServer: case DataType.SqlServer:
case DataType.PostgreSQL: case DataType.PostgreSQL:
@ -40,9 +40,9 @@ namespace FreeSql
await DbContextExecCommandAsync(); await DbContextExecCommandAsync();
var idtval = await this.OrmInsert(data).ExecuteIdentityAsync(); var idtval = await this.OrmInsert(data).ExecuteIdentityAsync();
IncrAffrows(1); IncrAffrows(1);
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval); _db.Orm.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
Attach(data); Attach(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
await AddOrUpdateNavigateListAsync(data); await AddOrUpdateNavigateListAsync(data);
} }
else else
@ -50,9 +50,9 @@ namespace FreeSql
await DbContextExecCommandAsync(); await DbContextExecCommandAsync();
var newval = (await this.OrmInsert(data).ExecuteInsertedAsync()).First(); var newval = (await this.OrmInsert(data).ExecuteInsertedAsync()).First();
IncrAffrows(1); IncrAffrows(1);
_fsql.MapEntityValue(_entityType, newval, data); _db.Orm.MapEntityValue(_entityType, newval, data);
Attach(newval); Attach(newval);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
await AddOrUpdateNavigateListAsync(data); await AddOrUpdateNavigateListAsync(data);
} }
return; return;
@ -64,9 +64,9 @@ namespace FreeSql
await DbContextExecCommandAsync(); await DbContextExecCommandAsync();
var idtval = await this.OrmInsert(data).ExecuteIdentityAsync(); var idtval = await this.OrmInsert(data).ExecuteIdentityAsync();
IncrAffrows(1); IncrAffrows(1);
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval); _db.Orm.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
Attach(data); Attach(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
await AddOrUpdateNavigateListAsync(data); await AddOrUpdateNavigateListAsync(data);
} }
return; return;
@ -74,7 +74,7 @@ namespace FreeSql
} }
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data)); EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data));
Attach(data); Attach(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
await AddOrUpdateNavigateListAsync(data); await AddOrUpdateNavigateListAsync(data);
} }
public Task AddAsync(TEntity data) => AddPrivAsync(data, true); public Task AddAsync(TEntity data) => AddPrivAsync(data, true);
@ -89,19 +89,19 @@ namespace FreeSql
if (_tableIdentitys.Length > 0) if (_tableIdentitys.Length > 0)
{ {
//有自增，马上执行 //有自增，马上执行
switch (_fsql.Ado.DataType) switch (_db.Orm.Ado.DataType)
{ {
case DataType.SqlServer: case DataType.SqlServer:
case DataType.PostgreSQL: case DataType.PostgreSQL:
await DbContextExecCommandAsync(); await DbContextExecCommandAsync();
var rets = await this.OrmInsert(data).ExecuteInsertedAsync(); var rets = await this.OrmInsert(data).ExecuteInsertedAsync();
if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_fsql.Ado.DataType} 的返回数据，与添加的数目不匹配"); if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_db.Orm.Ado.DataType} 的返回数据，与添加的数目不匹配");
var idx = 0; var idx = 0;
foreach (var s in data) foreach (var s in data)
_fsql.MapEntityValue(_entityType, rets[idx++], s); _db.Orm.MapEntityValue(_entityType, rets[idx++], s);
IncrAffrows(rets.Count); IncrAffrows(rets.Count);
AttachRange(rets); AttachRange(rets);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
foreach (var item in data) foreach (var item in data)
await AddOrUpdateNavigateListAsync(item); await AddOrUpdateNavigateListAsync(item);
return; return;
@ -119,7 +119,7 @@ namespace FreeSql
foreach (var item in data) foreach (var item in data)
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item)); EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item));
AttachRange(data); AttachRange(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
foreach (var item in data) foreach (var item in data)
await AddOrUpdateNavigateListAsync(item); await AddOrUpdateNavigateListAsync(item);
} }
@ -161,7 +161,7 @@ namespace FreeSql
{ {
if (dbset == null) if (dbset == null)
{ {
dbset = _ctx.Set(tref.RefEntityType); dbset = _db.Set(tref.RefEntityType);
dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdateAsync", new Type[] { tref.RefEntityType }); dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdateAsync", new Type[] { tref.RefEntityType });
} }
for (var colidx = 0; colidx < tref.Columns.Count; colidx++) for (var colidx = 0; colidx < tref.Columns.Count; colidx++)
@ -189,10 +189,10 @@ namespace FreeSql
if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999; if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999;
var lstval2 = default(EntityState); var lstval2 = default(EntityState);
if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_fsql.GetEntityString(_entityType, uplst2.Value)}"); if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_db.Orm.GetEntityString(_entityType, uplst2.Value)}");
var cuig1 = _fsql.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true); var cuig1 = _db.Orm.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true);
var cuig2 = uplst2 != null ? _fsql.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null; var cuig2 = uplst2 != null ? _db.Orm.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null;
List<EntityState> data = null; List<EntityState> data = null;
string[] cuig = null; string[] cuig = null;
@ -224,9 +224,9 @@ namespace FreeSql
foreach (var newval in data) foreach (var newval in data)
{ {
if (_states.TryGetValue(newval.Key, out var tryold)) if (_states.TryGetValue(newval.Key, out var tryold))
_fsql.MapEntityValue(_entityType, newval.Value, tryold.Value); _db.Orm.MapEntityValue(_entityType, newval.Value, tryold.Value);
if (newval.OldValue != null) if (newval.OldValue != null)
_fsql.MapEntityValue(_entityType, newval.Value, newval.OldValue); _db.Orm.MapEntityValue(_entityType, newval.Value, newval.OldValue);
} }
return affrows; return affrows;
} }
@ -237,11 +237,11 @@ namespace FreeSql
async public Task UpdateAsync(TEntity data) async public Task UpdateAsync(TEntity data)
{ {
var exists = ExistsInStates(data); var exists = ExistsInStates(data);
if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}"); if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_db.Orm.GetEntityString(_entityType, data)}");
if (exists == false) if (exists == false)
{ {
var olddata = await OrmSelect(data).FirstAsync(); var olddata = await OrmSelect(data).FirstAsync();
if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_fsql.GetEntityString(_entityType, data)}"); if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_db.Orm.GetEntityString(_entityType, data)}");
} }
await UpdateRangePrivAsync(new[] { data }, true); await UpdateRangePrivAsync(new[] { data }, true);
@ -260,7 +260,7 @@ namespace FreeSql
state.OldValue = item; state.OldValue = item;
EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state); EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state);
} }
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
foreach (var item in data) foreach (var item in data)
await AddOrUpdateNavigateListAsync(item); await AddOrUpdateNavigateListAsync(item);
} }
@ -279,7 +279,7 @@ namespace FreeSql
async public Task AddOrUpdateAsync(TEntity data) async public Task AddOrUpdateAsync(TEntity data)
{ {
if (data == null) throw new ArgumentNullException(nameof(data)); if (data == null) throw new ArgumentNullException(nameof(data));
if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}"); if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_db.Orm.GetEntityString(_entityType, data)}");
var flagExists = ExistsInStates(data); var flagExists = ExistsInStates(data);
if (flagExists == false) if (flagExists == false)
@ -291,15 +291,15 @@ namespace FreeSql
if (flagExists == true && CanUpdate(data, false)) if (flagExists == true && CanUpdate(data, false))
{ {
await DbContextExecCommandAsync(); await DbContextExecCommandAsync();
var affrows = _ctx._affrows; var affrows = _db._affrows;
await UpdateRangePrivAsync(new[] { data }, false); await UpdateRangePrivAsync(new[] { data }, false);
await DbContextExecCommandAsync(); await DbContextExecCommandAsync();
affrows = _ctx._affrows - affrows; affrows = _db._affrows - affrows;
if (affrows > 0) return; if (affrows > 0) return;
} }
if (CanAdd(data, false)) if (CanAdd(data, false))
{ {
_fsql.ClearEntityPrimaryValueWithIdentity(_entityType, data); _db.Orm.ClearEntityPrimaryValueWithIdentity(_entityType, data);
await AddPrivAsync(data, false); await AddPrivAsync(data, false);
} }
} }

56
FreeSql.DbContext/DbSet/DbSetSync.cs
View File

@ -14,7 +14,7 @@ namespace FreeSql
void DbContextExecCommand() void DbContextExecCommand()
{ {
_dicUpdateTimes.Clear(); _dicUpdateTimes.Clear();
_ctx.ExecCommand(); _db.ExecCommand();
} }
int DbContextBetchAdd(EntityState[] adds) int DbContextBetchAdd(EntityState[] adds)
@ -31,7 +31,7 @@ namespace FreeSql
if (_tableIdentitys.Length > 0) if (_tableIdentitys.Length > 0)
{ {
//有自增，马上执行 //有自增，马上执行
switch (_fsql.Ado.DataType) switch (_db.Orm.Ado.DataType)
{ {
case DataType.SqlServer: case DataType.SqlServer:
case DataType.PostgreSQL: case DataType.PostgreSQL:
@ -40,9 +40,9 @@ namespace FreeSql
DbContextExecCommand(); DbContextExecCommand();
var idtval = this.OrmInsert(data).ExecuteIdentity(); var idtval = this.OrmInsert(data).ExecuteIdentity();
IncrAffrows(1); IncrAffrows(1);
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval); _db.Orm.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
Attach(data); Attach(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
AddOrUpdateNavigateList(data); AddOrUpdateNavigateList(data);
} }
else else
@ -50,9 +50,9 @@ namespace FreeSql
DbContextExecCommand(); DbContextExecCommand();
var newval = this.OrmInsert(data).ExecuteInserted().First(); var newval = this.OrmInsert(data).ExecuteInserted().First();
IncrAffrows(1); IncrAffrows(1);
_fsql.MapEntityValue(_entityType, newval, data); _db.Orm.MapEntityValue(_entityType, newval, data);
Attach(newval); Attach(newval);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
AddOrUpdateNavigateList(data); AddOrUpdateNavigateList(data);
} }
return; return;
@ -64,9 +64,9 @@ namespace FreeSql
DbContextExecCommand(); DbContextExecCommand();
var idtval = this.OrmInsert(data).ExecuteIdentity(); var idtval = this.OrmInsert(data).ExecuteIdentity();
IncrAffrows(1); IncrAffrows(1);
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval); _db.Orm.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
Attach(data); Attach(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
AddOrUpdateNavigateList(data); AddOrUpdateNavigateList(data);
} }
return; return;
@ -74,7 +74,7 @@ namespace FreeSql
} }
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data)); EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data));
Attach(data); Attach(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
AddOrUpdateNavigateList(data); AddOrUpdateNavigateList(data);
} }
/// <summary> /// <summary>
@ -93,19 +93,19 @@ namespace FreeSql
if (_tableIdentitys.Length > 0) if (_tableIdentitys.Length > 0)
{ {
//有自增，马上执行 //有自增，马上执行
switch (_fsql.Ado.DataType) switch (_db.Orm.Ado.DataType)
{ {
case DataType.SqlServer: case DataType.SqlServer:
case DataType.PostgreSQL: case DataType.PostgreSQL:
DbContextExecCommand(); DbContextExecCommand();
var rets = this.OrmInsert(data).ExecuteInserted(); var rets = this.OrmInsert(data).ExecuteInserted();
if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_fsql.Ado.DataType} 的返回数据，与添加的数目不匹配"); if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_db.Orm.Ado.DataType} 的返回数据，与添加的数目不匹配");
var idx = 0; var idx = 0;
foreach (var s in data) foreach (var s in data)
_fsql.MapEntityValue(_entityType, rets[idx++], s); _db.Orm.MapEntityValue(_entityType, rets[idx++], s);
IncrAffrows(rets.Count); IncrAffrows(rets.Count);
AttachRange(rets); AttachRange(rets);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
foreach (var item in data) foreach (var item in data)
AddOrUpdateNavigateList(item); AddOrUpdateNavigateList(item);
return; return;
@ -123,7 +123,7 @@ namespace FreeSql
foreach (var item in data) foreach (var item in data)
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item)); EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item));
AttachRange(data); AttachRange(data);
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
foreach (var item in data) foreach (var item in data)
AddOrUpdateNavigateList(item); AddOrUpdateNavigateList(item);
} }
@ -175,7 +175,7 @@ namespace FreeSql
{ {
if (dbset == null) if (dbset == null)
{ {
dbset = _ctx.Set(tref.RefEntityType); dbset = _db.Set(tref.RefEntityType);
dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdate", new Type[] { tref.RefEntityType }); dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdate", new Type[] { tref.RefEntityType });
} }
for (var colidx = 0; colidx < tref.Columns.Count; colidx++) for (var colidx = 0; colidx < tref.Columns.Count; colidx++)
@ -202,10 +202,10 @@ namespace FreeSql
if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999; if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999;
var lstval2 = default(EntityState); var lstval2 = default(EntityState);
if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_fsql.GetEntityString(_entityType, uplst2.Value)}"); if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_db.Orm.GetEntityString(_entityType, uplst2.Value)}");
var cuig1 = _fsql.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true); var cuig1 = _db.Orm.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true);
var cuig2 = uplst2 != null ? _fsql.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null; var cuig2 = uplst2 != null ? _db.Orm.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null;
List<EntityState> data = null; List<EntityState> data = null;
string[] cuig = null; string[] cuig = null;
@ -237,9 +237,9 @@ namespace FreeSql
foreach (var newval in data) foreach (var newval in data)
{ {
if (_states.TryGetValue(newval.Key, out var tryold)) if (_states.TryGetValue(newval.Key, out var tryold))
_fsql.MapEntityValue(_entityType, newval.Value, tryold.Value); _db.Orm.MapEntityValue(_entityType, newval.Value, tryold.Value);
if (newval.OldValue != null) if (newval.OldValue != null)
_fsql.MapEntityValue(_entityType, newval.Value, newval.OldValue); _db.Orm.MapEntityValue(_entityType, newval.Value, newval.OldValue);
} }
return affrows; return affrows;
} }
@ -256,11 +256,11 @@ namespace FreeSql
public void Update(TEntity data) public void Update(TEntity data)
{ {
var exists = ExistsInStates(data); var exists = ExistsInStates(data);
if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}"); if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_db.Orm.GetEntityString(_entityType, data)}");
if (exists == false) if (exists == false)
{ {
var olddata = OrmSelect(data).First(); var olddata = OrmSelect(data).First();
if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_fsql.GetEntityString(_entityType, data)}"); if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_db.Orm.GetEntityString(_entityType, data)}");
} }
UpdateRangePriv(new[] { data }, true); UpdateRangePriv(new[] { data }, true);
@ -279,7 +279,7 @@ namespace FreeSql
state.OldValue = item; state.OldValue = item;
EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state); EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state);
} }
if (_ctx.Options.EnableAddOrUpdateNavigateList) if (_db.Options.EnableAddOrUpdateNavigateList)
foreach (var item in data) foreach (var item in data)
AddOrUpdateNavigateList(item); AddOrUpdateNavigateList(item);
} }
@ -305,7 +305,7 @@ namespace FreeSql
{ {
var state = CreateEntityState(item); var state = CreateEntityState(item);
_states.TryRemove(state.Key, out var trystate); _states.TryRemove(state.Key, out var trystate);
_fsql.ClearEntityPrimaryValueWithIdentityAndGuid(_entityType, item); _db.Orm.ClearEntityPrimaryValueWithIdentityAndGuid(_entityType, item);
EnqueueToDbContext(DbContext.ExecCommandInfoType.Delete, state); EnqueueToDbContext(DbContext.ExecCommandInfoType.Delete, state);
} }
@ -320,7 +320,7 @@ namespace FreeSql
public void AddOrUpdate(TEntity data) public void AddOrUpdate(TEntity data)
{ {
if (data == null) throw new ArgumentNullException(nameof(data)); if (data == null) throw new ArgumentNullException(nameof(data));
if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}"); if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_db.Orm.GetEntityString(_entityType, data)}");
var flagExists = ExistsInStates(data); var flagExists = ExistsInStates(data);
if (flagExists == false) if (flagExists == false)
@ -332,15 +332,15 @@ namespace FreeSql
if (flagExists == true && CanUpdate(data, false)) if (flagExists == true && CanUpdate(data, false))
{ {
DbContextExecCommand(); DbContextExecCommand();
var affrows = _ctx._affrows; var affrows = _db._affrows;
UpdateRangePriv(new[] { data }, false); UpdateRangePriv(new[] { data }, false);
DbContextExecCommand(); DbContextExecCommand();
affrows = _ctx._affrows - affrows; affrows = _db._affrows - affrows;
if (affrows > 0) return; if (affrows > 0) return;
} }
if (CanAdd(data, false)) if (CanAdd(data, false))
{ {
_fsql.ClearEntityPrimaryValueWithIdentity(_entityType, data); _db.Orm.ClearEntityPrimaryValueWithIdentity(_entityType, data);
AddPriv(data, false); AddPriv(data, false);
} }
} }

31
FreeSql.DbContext/Extenssions/DependencyInjection.cs
View File

@ -1,35 +1,44 @@
#if ns20 #if ns20
using Microsoft.Extensions.DependencyInjection; using Microsoft.Extensions.DependencyInjection;
using System; using System;
using System.Linq;
using System.Reflection;
namespace FreeSql namespace FreeSql
{ {
public static class DbContextDependencyInjection public static class DbContextDependencyInjection
{ {
static IServiceCollection AddFreeDbContext(this IServiceCollection services, Type dbContextType, Action<DbContextOptionsBuilder> options)
public static IServiceCollection AddFreeDbContext<TDbContext>(this IServiceCollection services, Action<DbContextOptionsBuilder> options) where TDbContext : DbContext
{ {
services.AddScoped(dbContextType, sp =>
services.AddScoped<TDbContext>(sp =>
{ {
var ctx = Activator.CreateInstance<TDbContext>(); var ctx = Activator.CreateInstance(dbContextType) as DbContext;
if (ctx != null && ctx._ormPriv == null)
if (ctx._orm == null)
{ {
var builder = new DbContextOptionsBuilder(); var builder = new DbContextOptionsBuilder();
options(builder); options(builder);
ctx._orm = builder._fsql; ctx._ormPriv = builder._fsql;
ctx._optionsPriv = builder._options;
if (ctx._orm == null) if (ctx._ormPriv == null)
throw new Exception("请在 OnConfiguring 或 AddFreeDbContext 中配置 UseFreeSql"); throw new Exception("请在 OnConfiguring 或 AddFreeDbContext 中配置 UseFreeSql");
ctx.InitPropSets(); ctx.InitPropSets();
} }
return ctx; return ctx;
}); });
return services;
}
public static IServiceCollection AddFreeDbContext<TDbContext>(this IServiceCollection services, Action<DbContextOptionsBuilder> options) where TDbContext : DbContext =>
AddFreeDbContext(services, typeof(TDbContext), options);
public static IServiceCollection AddFreeDbContext(this IServiceCollection services, Action<DbContextOptionsBuilder> options, Assembly[] assemblies)
{
if (assemblies?.Any() == true)
foreach (var asse in assemblies)
foreach (var dbType in asse.GetTypes().Where(a => a.IsAbstract == false && typeof(DbContext).IsAssignableFrom(a)))
AddFreeDbContext(services, dbType, options);
return services; return services;
} }
} }

29
FreeSql.DbContext/Repository/ContextSet/RepositoryDbContext.cs
View File

@ -8,13 +8,13 @@ namespace FreeSql
internal class RepositoryDbContext : DbContext internal class RepositoryDbContext : DbContext
{ {
protected IBaseRepository _repos; protected IBaseRepository _repo;
public RepositoryDbContext(IFreeSql orm, IBaseRepository repos) : base() public RepositoryDbContext(IFreeSql orm, IBaseRepository repo) : base()
{ {
_orm = orm; _ormPriv = orm;
_repos = repos;
_isUseUnitOfWork = false; _isUseUnitOfWork = false;
_uowPriv = _repos.UnitOfWork; UnitOfWork = _repo.UnitOfWork;
_repo = repo;
} }
@ -24,21 +24,22 @@ namespace FreeSql
{ {
if (_dicSet.ContainsKey(entityType)) return _dicSet[entityType]; if (_dicSet.ContainsKey(entityType)) return _dicSet[entityType];
var tb = _orm.CodeFirst.GetTableByEntity(entityType); var tb = _ormPriv.CodeFirst.GetTableByEntity(entityType);
if (tb == null) return null; if (tb == null) return null;
object repos = _repos; object repo = _repo;
if (entityType != _repos.EntityType) if (entityType != _repo.EntityType)
{ {
repos = Activator.CreateInstance(typeof(DefaultRepository<,>).MakeGenericType(entityType, typeof(int)), _repos.Orm); repo = Activator.CreateInstance(typeof(DefaultRepository<,>).MakeGenericType(entityType, typeof(int)), _repo.Orm);
(repos as IBaseRepository).UnitOfWork = _repos.UnitOfWork; (repo as IBaseRepository).UnitOfWork = _repo.UnitOfWork;
GetRepositoryDbField(entityType).SetValue(repos, this); GetRepositoryDbField(entityType).SetValue(repo, this);
typeof(RepositoryDbContext).GetMethod("SetRepositoryDataFilter").MakeGenericMethod(_repos.EntityType) typeof(RepositoryDbContext).GetMethod("SetRepositoryDataFilter").MakeGenericMethod(_repo.EntityType)
.Invoke(null, new object[] { repos, _repos }); .Invoke(null, new object[] { repo, _repo });
} }
var sd = Activator.CreateInstance(typeof(RepositoryDbSet<>).MakeGenericType(entityType), repos) as IDbSet; var sd = Activator.CreateInstance(typeof(RepositoryDbSet<>).MakeGenericType(entityType), repo) as IDbSet;
_listSet.Add(sd);
if (entityType != typeof(object)) _dicSet.Add(entityType, sd); if (entityType != typeof(object)) _dicSet.Add(entityType, sd);
return sd; return sd;
} }

31
FreeSql.DbContext/Repository/ContextSet/RepositoryDbSet.cs
View File

@ -8,60 +8,59 @@ namespace FreeSql
internal class RepositoryDbSet<TEntity> : DbSet<TEntity> where TEntity : class internal class RepositoryDbSet<TEntity> : DbSet<TEntity> where TEntity : class
{ {
protected BaseRepository<TEntity> _repos; protected BaseRepository<TEntity> _repo;
public RepositoryDbSet(BaseRepository<TEntity> repos) public RepositoryDbSet(BaseRepository<TEntity> repo)
{ {
_ctx = repos._db; _db = repo._db;
_fsql = repos.Orm; _uow = repo.UnitOfWork;
_uow = repos.UnitOfWork; _repo = repo;
_repos = repos;
} }
protected override ISelect<TEntity> OrmSelect(object dywhere) protected override ISelect<TEntity> OrmSelect(object dywhere)
{ {
var select = base.OrmSelect(dywhere); var select = base.OrmSelect(dywhere);
var filters = (_repos.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true); var filters = (_repo.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true);
foreach (var filter in filters) select.Where(filter.Value.Expression); foreach (var filter in filters) select.Where(filter.Value.Expression);
return select.AsTable(_repos.AsTableSelectInternal); return select.AsTable(_repo.AsTableSelectInternal);
} }
internal ISelect<TEntity> OrmSelectInternal(object dywhere) => OrmSelect(dywhere); internal ISelect<TEntity> OrmSelectInternal(object dywhere) => OrmSelect(dywhere);
protected override IUpdate<TEntity> OrmUpdate(IEnumerable<TEntity> entitys) protected override IUpdate<TEntity> OrmUpdate(IEnumerable<TEntity> entitys)
{ {
var update = base.OrmUpdate(entitys); var update = base.OrmUpdate(entitys);
var filters = (_repos.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true); var filters = (_repo.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true);
foreach (var filter in filters) foreach (var filter in filters)
{ {
if (entitys != null) if (entitys != null)
foreach (var entity in entitys) foreach (var entity in entitys)
if (filter.Value.ExpressionDelegate?.Invoke(entity) == false) if (filter.Value.ExpressionDelegate?.Invoke(entity) == false)
throw new Exception($"FreeSql.Repository Update 失败，因为设置了过滤器 {filter.Key}: {filter.Value.Expression}，更新的数据不符合 {_fsql.GetEntityString(_entityType, entity)}"); throw new Exception($"FreeSql.Repository Update 失败，因为设置了过滤器 {filter.Key}: {filter.Value.Expression}，更新的数据不符合 {_db.Orm.GetEntityString(_entityType, entity)}");
update.Where(filter.Value.Expression); update.Where(filter.Value.Expression);
} }
return update.AsTable(_repos.AsTableInternal); return update.AsTable(_repo.AsTableInternal);
} }
internal IUpdate<TEntity> OrmUpdateInternal(IEnumerable<TEntity> entitys) => OrmUpdate(entitys); internal IUpdate<TEntity> OrmUpdateInternal(IEnumerable<TEntity> entitys) => OrmUpdate(entitys);
protected override IDelete<TEntity> OrmDelete(object dywhere) protected override IDelete<TEntity> OrmDelete(object dywhere)
{ {
var delete = base.OrmDelete(dywhere); var delete = base.OrmDelete(dywhere);
var filters = (_repos.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true); var filters = (_repo.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true);
foreach (var filter in filters) delete.Where(filter.Value.Expression); foreach (var filter in filters) delete.Where(filter.Value.Expression);
return delete.AsTable(_repos.AsTableInternal); return delete.AsTable(_repo.AsTableInternal);
} }
internal IDelete<TEntity> OrmDeleteInternal(object dywhere) => OrmDelete(dywhere); internal IDelete<TEntity> OrmDeleteInternal(object dywhere) => OrmDelete(dywhere);
protected override IInsert<TEntity> OrmInsert(TEntity entity) => OrmInsert(new[] { entity }); protected override IInsert<TEntity> OrmInsert(TEntity entity) => OrmInsert(new[] { entity });
protected override IInsert<TEntity> OrmInsert(IEnumerable<TEntity> entitys) protected override IInsert<TEntity> OrmInsert(IEnumerable<TEntity> entitys)
{ {
var insert = base.OrmInsert(entitys); var insert = base.OrmInsert(entitys);
var filters = (_repos.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true); var filters = (_repo.DataFilter as DataFilter<TEntity>)._filters.Where(a => a.Value.IsEnabled == true);
foreach (var filter in filters) foreach (var filter in filters)
{ {
if (entitys != null) if (entitys != null)
foreach (var entity in entitys) foreach (var entity in entitys)
if (filter.Value.ExpressionDelegate?.Invoke(entity) == false) if (filter.Value.ExpressionDelegate?.Invoke(entity) == false)
throw new Exception($"FreeSql.Repository Insert 失败，因为设置了过滤器 {filter.Key}: {filter.Value.Expression}，插入的数据不符合 {_fsql.GetEntityString(_entityType, entity)}"); throw new Exception($"FreeSql.Repository Insert 失败，因为设置了过滤器 {filter.Key}: {filter.Value.Expression}，插入的数据不符合 {_db.Orm.GetEntityString(_entityType, entity)}");
} }
return insert.AsTable(_repos.AsTableInternal); return insert.AsTable(_repo.AsTableInternal);
} }
internal IInsert<TEntity> OrmInsertInternal(TEntity entity) => OrmInsert(entity); internal IInsert<TEntity> OrmInsertInternal(TEntity entity) => OrmInsert(entity);
internal IInsert<TEntity> OrmInsertInternal(IEnumerable<TEntity> entitys) => OrmInsert(entitys); internal IInsert<TEntity> OrmInsertInternal(IEnumerable<TEntity> entitys) => OrmInsert(entitys);

18
FreeSql.DbContext/Repository/ContextSet/RepositoryUnitOfWork.cs
View File

@ -42,23 +42,23 @@ namespace FreeSql
public GuidRepository<TEntity> GetGuidRepository<TEntity>(Expression<Func<TEntity, bool>> filter = null, Func<string, string> asTable = null) where TEntity : class public GuidRepository<TEntity> GetGuidRepository<TEntity>(Expression<Func<TEntity, bool>> filter = null, Func<string, string> asTable = null) where TEntity : class
{ {
var repos = new GuidRepository<TEntity>(_fsql, filter, asTable); var repo = new GuidRepository<TEntity>(_fsql, filter, asTable);
repos.UnitOfWork = this; repo.UnitOfWork = this;
return repos; return repo;
} }
public DefaultRepository<TEntity, TKey> GetRepository<TEntity, TKey>(Expression<Func<TEntity, bool>> filter = null) where TEntity : class public DefaultRepository<TEntity, TKey> GetRepository<TEntity, TKey>(Expression<Func<TEntity, bool>> filter = null) where TEntity : class
{ {
var repos = new DefaultRepository<TEntity, TKey>(_fsql, filter); var repo = new DefaultRepository<TEntity, TKey>(_fsql, filter);
repos.UnitOfWork = this; repo.UnitOfWork = this;
return repos; return repo;
} }
public BaseRepository<TEntity> GetRepository<TEntity>(Expression<Func<TEntity, bool>> filter = null) where TEntity : class public BaseRepository<TEntity> GetRepository<TEntity>(Expression<Func<TEntity, bool>> filter = null) where TEntity : class
{ {
var repos = new DefaultRepository<TEntity, int>(_fsql, filter); var repo = new DefaultRepository<TEntity, int>(_fsql, filter);
repos.UnitOfWork = this; repo.UnitOfWork = this;
return repos; return repo;
} }
} }
} }

12
FreeSql.DbContext/Repository/Repository/BaseRepository.cs
View File

@ -10,10 +10,11 @@ namespace FreeSql
where TEntity : class where TEntity : class
{ {
internal RepositoryDbContext _dbPriv; RepositoryDbContext _dbPriv;
internal RepositoryDbContext _db => _dbPriv ?? (_dbPriv = new RepositoryDbContext(Orm, this)); internal RepositoryDbContext _db => _dbPriv ?? (_dbPriv = new RepositoryDbContext(Orm, this));
internal RepositoryDbSet<TEntity> _dbsetPriv; RepositoryDbSet<TEntity> _dbsetPriv;
internal RepositoryDbSet<TEntity> _dbset => _dbsetPriv ?? (_dbsetPriv = _db.Set<TEntity>() as RepositoryDbSet<TEntity>); internal RepositoryDbSet<TEntity> _dbset => _dbsetPriv ?? (_dbsetPriv = _db.Set<TEntity>() as RepositoryDbSet<TEntity>);
public IDataFilter<TEntity> DataFilter { get; } = new DataFilter<TEntity>(); public IDataFilter<TEntity> DataFilter { get; } = new DataFilter<TEntity>();
Func<string, string> _asTableVal; Func<string, string> _asTableVal;
protected Func<string, string> AsTable protected Func<string, string> AsTable
@ -39,14 +40,12 @@ namespace FreeSql
AsTable = asTable; AsTable = asTable;
} }
~BaseRepository() ~BaseRepository() => this.Dispose();
{
this.Dispose();
}
bool _isdisposed = false; bool _isdisposed = false;
public void Dispose() public void Dispose()
{ {
if (_isdisposed) return; if (_isdisposed) return;
_isdisposed = true;
try try
{ {
_dbsetPriv?.Dispose(); _dbsetPriv?.Dispose();
@ -55,7 +54,6 @@ namespace FreeSql
} }
finally finally
{ {
_isdisposed = true;
GC.SuppressFinalize(this); GC.SuppressFinalize(this);
} }
} }