mirror of
https://github.com/nsnail/FreeSql.git
synced 2025-06-18 20:08:15 +08:00
源代码改用vs默认格式化
This commit is contained in:
28810
@ -8,274 +8,326 @@ using System.Linq;
|
||||
using System.Linq.Expressions;
|
||||
using System.Reflection;
|
||||
|
||||
namespace FreeSql {
|
||||
namespace FreeSql
|
||||
{
|
||||
|
||||
internal class DbContextDbSet<TEntity> : DbSet<TEntity> where TEntity : class {
|
||||
internal class DbContextDbSet<TEntity> : DbSet<TEntity> where TEntity : class
|
||||
{
|
||||
|
||||
public DbContextDbSet(DbContext ctx) {
|
||||
_ctx = ctx;
|
||||
_uow = ctx._uow;
|
||||
_fsql = ctx._fsql;
|
||||
}
|
||||
}
|
||||
public DbContextDbSet(DbContext ctx)
|
||||
{
|
||||
_ctx = ctx;
|
||||
_uow = ctx._uow;
|
||||
_fsql = ctx._fsql;
|
||||
}
|
||||
}
|
||||
|
||||
public interface IDbSet : IDisposable {
|
||||
Type EntityType { get; }
|
||||
}
|
||||
public abstract partial class DbSet<TEntity> : IDbSet where TEntity : class {
|
||||
public interface IDbSet : IDisposable
|
||||
{
|
||||
Type EntityType { get; }
|
||||
}
|
||||
public abstract partial class DbSet<TEntity> : IDbSet where TEntity : class
|
||||
{
|
||||
|
||||
internal DbContext _ctx;
|
||||
internal IUnitOfWork _uow;
|
||||
internal IFreeSql _fsql;
|
||||
internal DbContext _ctx;
|
||||
internal IUnitOfWork _uow;
|
||||
internal IFreeSql _fsql;
|
||||
|
||||
protected virtual ISelect<TEntity> OrmSelect(object dywhere) {
|
||||
DbContextExecCommand(); //查询前先提交，否则会出脏读
|
||||
return _fsql.Select<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction(false)).TrackToList(TrackToList).WhereDynamic(dywhere);
|
||||
}
|
||||
protected virtual ISelect<TEntity> OrmSelect(object dywhere)
|
||||
{
|
||||
DbContextExecCommand(); //查询前先提交，否则会出脏读
|
||||
return _fsql.Select<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction(false)).TrackToList(TrackToList).WhereDynamic(dywhere);
|
||||
}
|
||||
|
||||
~DbSet() {
|
||||
this.Dispose();
|
||||
}
|
||||
bool _isdisposed = false;
|
||||
public void Dispose() {
|
||||
if (_isdisposed) return;
|
||||
try {
|
||||
this._dicUpdateTimes.Clear();
|
||||
this._states.Clear();
|
||||
} finally {
|
||||
_isdisposed = true;
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
}
|
||||
~DbSet()
|
||||
{
|
||||
this.Dispose();
|
||||
}
|
||||
bool _isdisposed = false;
|
||||
public void Dispose()
|
||||
{
|
||||
if (_isdisposed) return;
|
||||
try
|
||||
{
|
||||
this._dicUpdateTimes.Clear();
|
||||
this._states.Clear();
|
||||
}
|
||||
finally
|
||||
{
|
||||
_isdisposed = true;
|
||||
GC.SuppressFinalize(this);
|
||||
}
|
||||
}
|
||||
|
||||
protected virtual IInsert<TEntity> OrmInsert() => _fsql.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(IEnumerable<TEntity> data) => _fsql.Insert<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).AppendData(data);
|
||||
protected virtual IInsert<TEntity> OrmInsert() => _fsql.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(IEnumerable<TEntity> data) => _fsql.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 IDelete<TEntity> OrmDelete(object dywhere) => _fsql.Delete<TEntity>().AsType(_entityType).WithTransaction(_uow?.GetOrBeginTransaction()).WhereDynamic(dywhere);
|
||||
protected virtual IUpdate<TEntity> OrmUpdate(IEnumerable<TEntity> entitys) => _fsql.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);
|
||||
|
||||
internal void EnqueueToDbContext(DbContext.ExecCommandInfoType actionType, EntityState state) {
|
||||
_ctx.EnqueueAction(actionType, this, typeof(EntityState), state);
|
||||
}
|
||||
internal void IncrAffrows(int affrows) {
|
||||
_ctx._affrows += affrows;
|
||||
}
|
||||
internal void EnqueueToDbContext(DbContext.ExecCommandInfoType actionType, EntityState state)
|
||||
{
|
||||
_ctx.EnqueueAction(actionType, this, typeof(EntityState), state);
|
||||
}
|
||||
internal void IncrAffrows(int affrows)
|
||||
{
|
||||
_ctx._affrows += affrows;
|
||||
}
|
||||
|
||||
internal void TrackToList(object list) {
|
||||
if (list == null) return;
|
||||
var ls = list as IList<TEntity>;
|
||||
if (ls == null) {
|
||||
var ie = list as IEnumerable;
|
||||
if (ie == null) return;
|
||||
foreach (var item in ie) {
|
||||
if (item == null) return;
|
||||
var itemType = item.GetType();
|
||||
if (itemType == typeof(object)) return;
|
||||
if (itemType.FullName.StartsWith("Submission#")) itemType = itemType.BaseType;
|
||||
var dbset = _ctx.Set(itemType);
|
||||
dbset?.GetType().GetMethod("TrackToList", BindingFlags.Instance | BindingFlags.NonPublic).Invoke(dbset, new object[] { list });
|
||||
return;
|
||||
}
|
||||
return;
|
||||
}
|
||||
internal void TrackToList(object list)
|
||||
{
|
||||
if (list == null) return;
|
||||
var ls = list as IList<TEntity>;
|
||||
if (ls == null)
|
||||
{
|
||||
var ie = list as IEnumerable;
|
||||
if (ie == null) return;
|
||||
foreach (var item in ie)
|
||||
{
|
||||
if (item == null) return;
|
||||
var itemType = item.GetType();
|
||||
if (itemType == typeof(object)) return;
|
||||
if (itemType.FullName.StartsWith("Submission#")) itemType = itemType.BaseType;
|
||||
var dbset = _ctx.Set(itemType);
|
||||
dbset?.GetType().GetMethod("TrackToList", BindingFlags.Instance | BindingFlags.NonPublic).Invoke(dbset, new object[] { list });
|
||||
return;
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
foreach (var item in ls) {
|
||||
var key = _fsql.GetEntityKeyString(_entityType, item, false);
|
||||
if (key == null) continue;
|
||||
_states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) => {
|
||||
_fsql.MapEntityValue(_entityType, item, ov.Value);
|
||||
ov.Time = DateTime.Now;
|
||||
return ov;
|
||||
});
|
||||
}
|
||||
}
|
||||
foreach (var item in ls)
|
||||
{
|
||||
var key = _fsql.GetEntityKeyString(_entityType, item, false);
|
||||
if (key == null) continue;
|
||||
_states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) =>
|
||||
{
|
||||
_fsql.MapEntityValue(_entityType, item, ov.Value);
|
||||
ov.Time = DateTime.Now;
|
||||
return ov;
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
public ISelect<TEntity> Select => this.OrmSelect(null);
|
||||
public ISelect<TEntity> Where(Expression<Func<TEntity, bool>> exp) => this.OrmSelect(null).Where(exp);
|
||||
public ISelect<TEntity> WhereIf(bool condition, Expression<Func<TEntity, bool>> exp) => this.OrmSelect(null).WhereIf(condition, exp);
|
||||
public ISelect<TEntity> Select => this.OrmSelect(null);
|
||||
public ISelect<TEntity> Where(Expression<Func<TEntity, bool>> exp) => this.OrmSelect(null).Where(exp);
|
||||
public ISelect<TEntity> WhereIf(bool condition, Expression<Func<TEntity, bool>> exp) => this.OrmSelect(null).WhereIf(condition, exp);
|
||||
|
||||
protected ConcurrentDictionary<string, EntityState> _states = new ConcurrentDictionary<string, EntityState>();
|
||||
internal ConcurrentDictionary<string, EntityState> _statesInternal => _states;
|
||||
TableInfo _tablePriv;
|
||||
protected TableInfo _table => _tablePriv ?? (_tablePriv = _fsql.CodeFirst.GetTableByEntity(_entityType));
|
||||
ColumnInfo[] _tableIdentitysPriv;
|
||||
protected ColumnInfo[] _tableIdentitys => _tableIdentitysPriv ?? (_tableIdentitysPriv = _table.Primarys.Where(a => a.Attribute.IsIdentity).ToArray());
|
||||
protected Type _entityType = typeof(TEntity);
|
||||
public Type EntityType => _entityType;
|
||||
protected ConcurrentDictionary<string, EntityState> _states = new ConcurrentDictionary<string, EntityState>();
|
||||
internal ConcurrentDictionary<string, EntityState> _statesInternal => _states;
|
||||
TableInfo _tablePriv;
|
||||
protected TableInfo _table => _tablePriv ?? (_tablePriv = _fsql.CodeFirst.GetTableByEntity(_entityType));
|
||||
ColumnInfo[] _tableIdentitysPriv;
|
||||
protected ColumnInfo[] _tableIdentitys => _tableIdentitysPriv ?? (_tableIdentitysPriv = _table.Primarys.Where(a => a.Attribute.IsIdentity).ToArray());
|
||||
protected Type _entityType = typeof(TEntity);
|
||||
public Type EntityType => _entityType;
|
||||
|
||||
/// <summary>
|
||||
/// 动态Type，在使用 DbSet<object> 后使用本方法，指定实体类型
|
||||
/// </summary>
|
||||
/// <param name="entityType"></param>
|
||||
/// <returns></returns>
|
||||
public void AsType(Type entityType) {
|
||||
if (entityType == typeof(object)) throw new Exception("ISelect.AsType 参数不支持指定为 object");
|
||||
if (entityType == _entityType) return;
|
||||
var newtb = _fsql.CodeFirst.GetTableByEntity(entityType);
|
||||
_entityType = entityType;
|
||||
_tablePriv = newtb ?? throw new Exception("DbSet.AsType 参数错误，请传入正确的实体类型");
|
||||
_tableIdentitysPriv = null;
|
||||
}
|
||||
/// <summary>
|
||||
/// 动态Type，在使用 DbSet<object> 后使用本方法，指定实体类型
|
||||
/// </summary>
|
||||
/// <param name="entityType"></param>
|
||||
/// <returns></returns>
|
||||
public void AsType(Type entityType)
|
||||
{
|
||||
if (entityType == typeof(object)) throw new Exception("ISelect.AsType 参数不支持指定为 object");
|
||||
if (entityType == _entityType) return;
|
||||
var newtb = _fsql.CodeFirst.GetTableByEntity(entityType);
|
||||
_entityType = entityType;
|
||||
_tablePriv = newtb ?? throw new Exception("DbSet.AsType 参数错误，请传入正确的实体类型");
|
||||
_tableIdentitysPriv = null;
|
||||
}
|
||||
|
||||
public class EntityState {
|
||||
public EntityState(TEntity value, string key) {
|
||||
this.Value = value;
|
||||
this.Key = key;
|
||||
this.Time = DateTime.Now;
|
||||
}
|
||||
public TEntity OldValue { get; set; }
|
||||
public TEntity Value { get; set; }
|
||||
public string Key { get; set; }
|
||||
public DateTime Time { get; set; }
|
||||
}
|
||||
/// <summary>
|
||||
/// 附加实体，可用于不查询就更新或删除
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Attach(TEntity data) => AttachRange(new[] { data });
|
||||
public void AttachRange(IEnumerable<TEntity> data) {
|
||||
if (data == null || data.Any() == false) return;
|
||||
if (_table.Primarys.Any() == false) throw new Exception($"不可附加，实体没有主键：{_fsql.GetEntityString(_entityType, data.First())}");
|
||||
foreach (var item in data) {
|
||||
var key = _fsql.GetEntityKeyString(_entityType, item, false);
|
||||
if (string.IsNullOrEmpty(key)) throw new Exception($"不可附加，未设置主键的值：{_fsql.GetEntityString(_entityType, item)}");
|
||||
public class EntityState
|
||||
{
|
||||
public EntityState(TEntity value, string key)
|
||||
{
|
||||
this.Value = value;
|
||||
this.Key = key;
|
||||
this.Time = DateTime.Now;
|
||||
}
|
||||
public TEntity OldValue { get; set; }
|
||||
public TEntity Value { get; set; }
|
||||
public string Key { get; set; }
|
||||
public DateTime Time { get; set; }
|
||||
}
|
||||
/// <summary>
|
||||
/// 附加实体，可用于不查询就更新或删除
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Attach(TEntity data) => AttachRange(new[] { data });
|
||||
public void AttachRange(IEnumerable<TEntity> data)
|
||||
{
|
||||
if (data == null || data.Any() == false) return;
|
||||
if (_table.Primarys.Any() == false) throw new Exception($"不可附加，实体没有主键：{_fsql.GetEntityString(_entityType, data.First())}");
|
||||
foreach (var item in data)
|
||||
{
|
||||
var key = _fsql.GetEntityKeyString(_entityType, item, false);
|
||||
if (string.IsNullOrEmpty(key)) throw new Exception($"不可附加，未设置主键的值：{_fsql.GetEntityString(_entityType, item)}");
|
||||
|
||||
_states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) => {
|
||||
_fsql.MapEntityValue(_entityType, item, ov.Value);
|
||||
ov.Time = DateTime.Now;
|
||||
return ov;
|
||||
});
|
||||
}
|
||||
}
|
||||
/// <summary>
|
||||
/// 清空状态数据
|
||||
/// </summary>
|
||||
public void FlushState() {
|
||||
_states.Clear();
|
||||
}
|
||||
_states.AddOrUpdate(key, k => CreateEntityState(item), (k, ov) =>
|
||||
{
|
||||
_fsql.MapEntityValue(_entityType, item, ov.Value);
|
||||
ov.Time = DateTime.Now;
|
||||
return ov;
|
||||
});
|
||||
}
|
||||
}
|
||||
/// <summary>
|
||||
/// 清空状态数据
|
||||
/// </summary>
|
||||
public void FlushState()
|
||||
{
|
||||
_states.Clear();
|
||||
}
|
||||
|
||||
#region Utils
|
||||
EntityState CreateEntityState(TEntity data) {
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
var state = new EntityState((TEntity)Activator.CreateInstance(_entityType), key);
|
||||
_fsql.MapEntityValue(_entityType, data, state.Value);
|
||||
return state;
|
||||
}
|
||||
bool? ExistsInStates(TEntity data) {
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
if (string.IsNullOrEmpty(key)) return null;
|
||||
return _states.ContainsKey(key);
|
||||
}
|
||||
#region Utils
|
||||
EntityState CreateEntityState(TEntity data)
|
||||
{
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
var state = new EntityState((TEntity)Activator.CreateInstance(_entityType), key);
|
||||
_fsql.MapEntityValue(_entityType, data, state.Value);
|
||||
return state;
|
||||
}
|
||||
bool? ExistsInStates(TEntity data)
|
||||
{
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
if (string.IsNullOrEmpty(key)) return null;
|
||||
return _states.ContainsKey(key);
|
||||
}
|
||||
|
||||
bool CanAdd(IEnumerable<TEntity> data, bool isThrow) {
|
||||
if (data == null) {
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (data.Any() == false) return false;
|
||||
foreach (var s in data) if (CanAdd(s, isThrow) == false) return false;
|
||||
return true;
|
||||
}
|
||||
bool CanAdd(TEntity data, bool isThrow) {
|
||||
if (data == null) {
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (_table.Primarys.Any() == false) {
|
||||
if (isThrow) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, true);
|
||||
if (string.IsNullOrEmpty(key)) {
|
||||
switch (_fsql.Ado.DataType) {
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
return true;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
if (_tableIdentitys.Length == 1 && _table.Primarys.Length == 1) {
|
||||
return true;
|
||||
}
|
||||
if (isThrow) throw new Exception($"不可添加，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
} else {
|
||||
if (_states.ContainsKey(key)) {
|
||||
if (isThrow) throw new Exception($"不可添加，已存在于状态管理：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var idval = _fsql.GetEntityIdentityValueWithPrimary(_entityType, data);
|
||||
if (idval > 0) {
|
||||
if (isThrow) throw new Exception($"不可添加，自增属性有值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
bool CanAdd(IEnumerable<TEntity> data, bool isThrow)
|
||||
{
|
||||
if (data == null)
|
||||
{
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (data.Any() == false) return false;
|
||||
foreach (var s in data) if (CanAdd(s, isThrow) == false) return false;
|
||||
return true;
|
||||
}
|
||||
bool CanAdd(TEntity data, bool isThrow)
|
||||
{
|
||||
if (data == null)
|
||||
{
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (_table.Primarys.Any() == false)
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, true);
|
||||
if (string.IsNullOrEmpty(key))
|
||||
{
|
||||
switch (_fsql.Ado.DataType)
|
||||
{
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
return true;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
if (_tableIdentitys.Length == 1 && _table.Primarys.Length == 1)
|
||||
{
|
||||
return true;
|
||||
}
|
||||
if (isThrow) throw new Exception($"不可添加，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (_states.ContainsKey(key))
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可添加，已存在于状态管理：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var idval = _fsql.GetEntityIdentityValueWithPrimary(_entityType, data);
|
||||
if (idval > 0)
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可添加，自增属性有值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CanUpdate(IEnumerable<TEntity> data, bool isThrow) {
|
||||
if (data == null) {
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (data.Any() == false) return false;
|
||||
foreach (var s in data) if (CanUpdate(s, isThrow) == false) return false;
|
||||
return true;
|
||||
}
|
||||
bool CanUpdate(TEntity data, bool isThrow) {
|
||||
if (data == null) {
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (_table.Primarys.Any() == false) {
|
||||
if (isThrow) throw new Exception($"不可更新，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
if (string.IsNullOrEmpty(key)) {
|
||||
if (isThrow) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
if (_states.TryGetValue(key, out var tryval) == false) {
|
||||
if (isThrow) throw new Exception($"不可更新，数据未被跟踪，应该先查询 或者 Attach：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
bool CanUpdate(IEnumerable<TEntity> data, bool isThrow)
|
||||
{
|
||||
if (data == null)
|
||||
{
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (data.Any() == false) return false;
|
||||
foreach (var s in data) if (CanUpdate(s, isThrow) == false) return false;
|
||||
return true;
|
||||
}
|
||||
bool CanUpdate(TEntity data, bool isThrow)
|
||||
{
|
||||
if (data == null)
|
||||
{
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (_table.Primarys.Any() == false)
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可更新，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
if (string.IsNullOrEmpty(key))
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
if (_states.TryGetValue(key, out var tryval) == false)
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可更新，数据未被跟踪，应该先查询 或者 Attach：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
bool CanRemove(IEnumerable<TEntity> data, bool isThrow) {
|
||||
if (data == null) {
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (data.Any() == false) return false;
|
||||
foreach (var s in data) if (CanRemove(s, isThrow) == false) return false;
|
||||
return true;
|
||||
}
|
||||
bool CanRemove(TEntity data, bool isThrow) {
|
||||
if (data == null) {
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (_table.Primarys.Any() == false) {
|
||||
if (isThrow) throw new Exception($"不可删除，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
if (string.IsNullOrEmpty(key)) {
|
||||
if (isThrow) throw new Exception($"不可删除，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
//if (_states.TryGetValue(key, out var tryval) == false) {
|
||||
// if (isThrow) throw new Exception($"不可删除，数据未被跟踪，应该先查询：{_fsql.GetEntityString(_entityType, data)}");
|
||||
// return false;
|
||||
//}
|
||||
return true;
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
bool CanRemove(IEnumerable<TEntity> data, bool isThrow)
|
||||
{
|
||||
if (data == null)
|
||||
{
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (data.Any() == false) return false;
|
||||
foreach (var s in data) if (CanRemove(s, isThrow) == false) return false;
|
||||
return true;
|
||||
}
|
||||
bool CanRemove(TEntity data, bool isThrow)
|
||||
{
|
||||
if (data == null)
|
||||
{
|
||||
if (isThrow) throw new ArgumentNullException(nameof(data));
|
||||
return false;
|
||||
}
|
||||
if (_table.Primarys.Any() == false)
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可删除，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
var key = _fsql.GetEntityKeyString(_entityType, data, false);
|
||||
if (string.IsNullOrEmpty(key))
|
||||
{
|
||||
if (isThrow) throw new Exception($"不可删除，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
return false;
|
||||
}
|
||||
//if (_states.TryGetValue(key, out var tryval) == false) {
|
||||
// if (isThrow) throw new Exception($"不可删除，数据未被跟踪，应该先查询：{_fsql.GetEntityString(_entityType, data)}");
|
||||
// return false;
|
||||
//}
|
||||
return true;
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
|
||||
@ -6,263 +6,303 @@ using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace FreeSql {
|
||||
partial class DbSet<TEntity> {
|
||||
namespace FreeSql
|
||||
{
|
||||
partial class DbSet<TEntity>
|
||||
{
|
||||
|
||||
Task DbContextExecCommandAsync() {
|
||||
_dicUpdateTimes.Clear();
|
||||
return _ctx.ExecCommandAsync();
|
||||
}
|
||||
Task DbContextExecCommandAsync()
|
||||
{
|
||||
_dicUpdateTimes.Clear();
|
||||
return _ctx.ExecCommandAsync();
|
||||
}
|
||||
|
||||
async Task<int> DbContextBetchAddAsync(EntityState[] adds) {
|
||||
if (adds.Any() == false) return 0;
|
||||
var affrows = await this.OrmInsert(adds.Select(a => a.Value)).ExecuteAffrowsAsync();
|
||||
return affrows;
|
||||
}
|
||||
async Task<int> DbContextBetchAddAsync(EntityState[] adds)
|
||||
{
|
||||
if (adds.Any() == false) return 0;
|
||||
var affrows = await this.OrmInsert(adds.Select(a => a.Value)).ExecuteAffrowsAsync();
|
||||
return affrows;
|
||||
}
|
||||
|
||||
#region Add
|
||||
async Task AddPrivAsync(TEntity data, bool isCheck) {
|
||||
if (isCheck && CanAdd(data, true) == false) return;
|
||||
if (_tableIdentitys.Length > 0) {
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType) {
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
if (_tableIdentitys.Length == 1 && _table.Primarys.Length == 1) {
|
||||
await DbContextExecCommandAsync();
|
||||
var idtval = await this.OrmInsert(data).ExecuteIdentityAsync();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
} else {
|
||||
await DbContextExecCommandAsync();
|
||||
var newval = (await this.OrmInsert(data).ExecuteInsertedAsync()).First();
|
||||
IncrAffrows(1);
|
||||
_fsql.MapEntityValue(_entityType, newval, data);
|
||||
Attach(newval);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
}
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
if (_tableIdentitys.Length == 1 && _table.Primarys.Length == 1) {
|
||||
await DbContextExecCommandAsync();
|
||||
var idtval = await this.OrmInsert(data).ExecuteIdentityAsync();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
}
|
||||
return;
|
||||
}
|
||||
}
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data));
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
}
|
||||
public Task AddAsync(TEntity data) => AddPrivAsync(data, true);
|
||||
async public Task AddRangeAsync(IEnumerable<TEntity> data) {
|
||||
if (CanAdd(data, true) == false) return;
|
||||
if (data.ElementAtOrDefault(1) == default(TEntity)) {
|
||||
await AddAsync(data.First());
|
||||
return;
|
||||
}
|
||||
if (_tableIdentitys.Length > 0) {
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType) {
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
await DbContextExecCommandAsync();
|
||||
var rets = await this.OrmInsert(data).ExecuteInsertedAsync();
|
||||
if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_fsql.Ado.DataType} 的返回数据，与添加的数目不匹配");
|
||||
var idx = 0;
|
||||
foreach (var s in data)
|
||||
_fsql.MapEntityValue(_entityType, rets[idx++], s);
|
||||
IncrAffrows(rets.Count);
|
||||
AttachRange(rets);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
await AddOrUpdateNavigateListAsync(item);
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
foreach (var s in data)
|
||||
await AddPrivAsync(s, false);
|
||||
return;
|
||||
}
|
||||
} else {
|
||||
//进入队列，等待 SaveChanges 时执行
|
||||
foreach (var item in data)
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item));
|
||||
AttachRange(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
await AddOrUpdateNavigateListAsync(item);
|
||||
}
|
||||
}
|
||||
async Task AddOrUpdateNavigateListAsync(TEntity item) {
|
||||
Type itemType = null;
|
||||
foreach (var prop in _table.Properties) {
|
||||
if (_table.ColumnsByCsIgnore.ContainsKey(prop.Key)) continue;
|
||||
if (_table.ColumnsByCs.ContainsKey(prop.Key)) continue;
|
||||
var tref = _table.GetTableRef(prop.Key, true);
|
||||
if (tref == null) continue;
|
||||
#region Add
|
||||
async Task AddPrivAsync(TEntity data, bool isCheck)
|
||||
{
|
||||
if (isCheck && CanAdd(data, true) == false) return;
|
||||
if (_tableIdentitys.Length > 0)
|
||||
{
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType)
|
||||
{
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
if (_tableIdentitys.Length == 1 && _table.Primarys.Length == 1)
|
||||
{
|
||||
await DbContextExecCommandAsync();
|
||||
var idtval = await this.OrmInsert(data).ExecuteIdentityAsync();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
}
|
||||
else
|
||||
{
|
||||
await DbContextExecCommandAsync();
|
||||
var newval = (await this.OrmInsert(data).ExecuteInsertedAsync()).First();
|
||||
IncrAffrows(1);
|
||||
_fsql.MapEntityValue(_entityType, newval, data);
|
||||
Attach(newval);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
}
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
if (_tableIdentitys.Length == 1 && _table.Primarys.Length == 1)
|
||||
{
|
||||
await DbContextExecCommandAsync();
|
||||
var idtval = await this.OrmInsert(data).ExecuteIdentityAsync();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
}
|
||||
return;
|
||||
}
|
||||
}
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data));
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
await AddOrUpdateNavigateListAsync(data);
|
||||
}
|
||||
public Task AddAsync(TEntity data) => AddPrivAsync(data, true);
|
||||
async public Task AddRangeAsync(IEnumerable<TEntity> data)
|
||||
{
|
||||
if (CanAdd(data, true) == false) return;
|
||||
if (data.ElementAtOrDefault(1) == default(TEntity))
|
||||
{
|
||||
await AddAsync(data.First());
|
||||
return;
|
||||
}
|
||||
if (_tableIdentitys.Length > 0)
|
||||
{
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType)
|
||||
{
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
await DbContextExecCommandAsync();
|
||||
var rets = await this.OrmInsert(data).ExecuteInsertedAsync();
|
||||
if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_fsql.Ado.DataType} 的返回数据，与添加的数目不匹配");
|
||||
var idx = 0;
|
||||
foreach (var s in data)
|
||||
_fsql.MapEntityValue(_entityType, rets[idx++], s);
|
||||
IncrAffrows(rets.Count);
|
||||
AttachRange(rets);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
await AddOrUpdateNavigateListAsync(item);
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
foreach (var s in data)
|
||||
await AddPrivAsync(s, false);
|
||||
return;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
//进入队列，等待 SaveChanges 时执行
|
||||
foreach (var item in data)
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item));
|
||||
AttachRange(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
await AddOrUpdateNavigateListAsync(item);
|
||||
}
|
||||
}
|
||||
async Task AddOrUpdateNavigateListAsync(TEntity item)
|
||||
{
|
||||
Type itemType = null;
|
||||
foreach (var prop in _table.Properties)
|
||||
{
|
||||
if (_table.ColumnsByCsIgnore.ContainsKey(prop.Key)) continue;
|
||||
if (_table.ColumnsByCs.ContainsKey(prop.Key)) continue;
|
||||
var tref = _table.GetTableRef(prop.Key, true);
|
||||
if (tref == null) continue;
|
||||
|
||||
switch (tref.RefType) {
|
||||
case Internal.Model.TableRefType.OneToOne:
|
||||
case Internal.Model.TableRefType.ManyToOne:
|
||||
case Internal.Model.TableRefType.ManyToMany:
|
||||
continue;
|
||||
case Internal.Model.TableRefType.OneToMany:
|
||||
if (itemType == null) itemType = item.GetType();
|
||||
if (_table.TypeLazy != null && itemType == _table.TypeLazy) {
|
||||
var lazyField = _dicLazyIsSetField.GetOrAdd(_table.TypeLazy, tl => new ConcurrentDictionary<string, System.Reflection.FieldInfo>()).GetOrAdd(prop.Key, propName =>
|
||||
_table.TypeLazy.GetField($"__lazy__{propName}", System.Reflection.BindingFlags.GetField | System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance));
|
||||
if (lazyField != null) {
|
||||
var lazyFieldValue = (bool)lazyField.GetValue(item);
|
||||
if (lazyFieldValue == false) continue;
|
||||
}
|
||||
}
|
||||
var propVal = prop.Value.GetValue(item);
|
||||
var propValEach = propVal as IEnumerable;
|
||||
if (propValEach == null) continue;
|
||||
object dbset = null;
|
||||
System.Reflection.MethodInfo dbsetAddOrUpdate = null;
|
||||
foreach (var propValItem in propValEach) {
|
||||
if (dbset == null) {
|
||||
dbset = _ctx.Set(tref.RefEntityType);
|
||||
dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdateAsync", new Type[] { tref.RefEntityType });
|
||||
}
|
||||
for (var colidx = 0; colidx < tref.Columns.Count; colidx++) {
|
||||
tref.RefColumns[colidx].Table.Properties[tref.RefColumns[colidx].CsName]
|
||||
.SetValue(propValItem, tref.Columns[colidx].Table.Properties[tref.Columns[colidx].CsName].GetValue(item));
|
||||
}
|
||||
Task task = dbsetAddOrUpdate.Invoke(dbset, new object[] { propValItem }) as Task;
|
||||
await task;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
switch (tref.RefType)
|
||||
{
|
||||
case Internal.Model.TableRefType.OneToOne:
|
||||
case Internal.Model.TableRefType.ManyToOne:
|
||||
case Internal.Model.TableRefType.ManyToMany:
|
||||
continue;
|
||||
case Internal.Model.TableRefType.OneToMany:
|
||||
if (itemType == null) itemType = item.GetType();
|
||||
if (_table.TypeLazy != null && itemType == _table.TypeLazy)
|
||||
{
|
||||
var lazyField = _dicLazyIsSetField.GetOrAdd(_table.TypeLazy, tl => new ConcurrentDictionary<string, System.Reflection.FieldInfo>()).GetOrAdd(prop.Key, propName =>
|
||||
_table.TypeLazy.GetField($"__lazy__{propName}", System.Reflection.BindingFlags.GetField | System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance));
|
||||
if (lazyField != null)
|
||||
{
|
||||
var lazyFieldValue = (bool)lazyField.GetValue(item);
|
||||
if (lazyFieldValue == false) continue;
|
||||
}
|
||||
}
|
||||
var propVal = prop.Value.GetValue(item);
|
||||
var propValEach = propVal as IEnumerable;
|
||||
if (propValEach == null) continue;
|
||||
object dbset = null;
|
||||
System.Reflection.MethodInfo dbsetAddOrUpdate = null;
|
||||
foreach (var propValItem in propValEach)
|
||||
{
|
||||
if (dbset == null)
|
||||
{
|
||||
dbset = _ctx.Set(tref.RefEntityType);
|
||||
dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdateAsync", new Type[] { tref.RefEntityType });
|
||||
}
|
||||
for (var colidx = 0; colidx < tref.Columns.Count; colidx++)
|
||||
{
|
||||
tref.RefColumns[colidx].Table.Properties[tref.RefColumns[colidx].CsName]
|
||||
.SetValue(propValItem, tref.Columns[colidx].Table.Properties[tref.Columns[colidx].CsName].GetValue(item));
|
||||
}
|
||||
Task task = dbsetAddOrUpdate.Invoke(dbset, new object[] { propValItem }) as Task;
|
||||
await task;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region UpdateAsync
|
||||
Task<int> DbContextBetchUpdateAsync(EntityState[] ups) => DbContextBetchUpdatePrivAsync(ups, false);
|
||||
Task<int> DbContextBetchUpdateNowAsync(EntityState[] ups) => DbContextBetchUpdatePrivAsync(ups, true);
|
||||
async Task<int> DbContextBetchUpdatePrivAsync(EntityState[] ups, bool isLiveUpdate) {
|
||||
if (ups.Any() == false) return 0;
|
||||
var uplst1 = ups[ups.Length - 1];
|
||||
var uplst2 = ups.Length > 1 ? ups[ups.Length - 2] : null;
|
||||
#region UpdateAsync
|
||||
Task<int> DbContextBetchUpdateAsync(EntityState[] ups) => DbContextBetchUpdatePrivAsync(ups, false);
|
||||
Task<int> DbContextBetchUpdateNowAsync(EntityState[] ups) => DbContextBetchUpdatePrivAsync(ups, true);
|
||||
async Task<int> DbContextBetchUpdatePrivAsync(EntityState[] ups, bool isLiveUpdate)
|
||||
{
|
||||
if (ups.Any() == false) return 0;
|
||||
var uplst1 = ups[ups.Length - 1];
|
||||
var uplst2 = ups.Length > 1 ? ups[ups.Length - 2] : null;
|
||||
|
||||
if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999;
|
||||
var lstval2 = default(EntityState);
|
||||
if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_fsql.GetEntityString(_entityType, uplst2.Value)}");
|
||||
if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999;
|
||||
var lstval2 = default(EntityState);
|
||||
if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_fsql.GetEntityString(_entityType, uplst2.Value)}");
|
||||
|
||||
var cuig1 = _fsql.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true);
|
||||
var cuig2 = uplst2 != null ? _fsql.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null;
|
||||
var cuig1 = _fsql.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true);
|
||||
var cuig2 = uplst2 != null ? _fsql.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null;
|
||||
|
||||
List<EntityState> data = null;
|
||||
string[] cuig = null;
|
||||
if (uplst2 != null && string.Compare(string.Join(",", cuig1), string.Join(",", cuig2)) != 0) {
|
||||
//最后一个不保存
|
||||
data = ups.ToList();
|
||||
data.RemoveAt(ups.Length - 1);
|
||||
cuig = cuig2;
|
||||
} else if (isLiveUpdate) {
|
||||
//立即保存
|
||||
data = ups.ToList();
|
||||
cuig = cuig1;
|
||||
}
|
||||
List<EntityState> data = null;
|
||||
string[] cuig = null;
|
||||
if (uplst2 != null && string.Compare(string.Join(",", cuig1), string.Join(",", cuig2)) != 0)
|
||||
{
|
||||
//最后一个不保存
|
||||
data = ups.ToList();
|
||||
data.RemoveAt(ups.Length - 1);
|
||||
cuig = cuig2;
|
||||
}
|
||||
else if (isLiveUpdate)
|
||||
{
|
||||
//立即保存
|
||||
data = ups.ToList();
|
||||
cuig = cuig1;
|
||||
}
|
||||
|
||||
if (data?.Count > 0) {
|
||||
if (data?.Count > 0)
|
||||
{
|
||||
|
||||
if (cuig.Length == _table.Columns.Count)
|
||||
return ups.Length == data.Count ? -998 : -997;
|
||||
if (cuig.Length == _table.Columns.Count)
|
||||
return ups.Length == data.Count ? -998 : -997;
|
||||
|
||||
var updateSource = data.Select(a => a.Value).ToArray();
|
||||
var update = this.OrmUpdate(null).SetSource(updateSource).IgnoreColumns(cuig);
|
||||
var updateSource = data.Select(a => a.Value).ToArray();
|
||||
var update = this.OrmUpdate(null).SetSource(updateSource).IgnoreColumns(cuig);
|
||||
|
||||
var affrows = await update.ExecuteAffrowsAsync();
|
||||
var affrows = await update.ExecuteAffrowsAsync();
|
||||
|
||||
foreach (var newval in data) {
|
||||
if (_states.TryGetValue(newval.Key, out var tryold))
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, tryold.Value);
|
||||
if (newval.OldValue != null)
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, newval.OldValue);
|
||||
}
|
||||
return affrows;
|
||||
}
|
||||
foreach (var newval in data)
|
||||
{
|
||||
if (_states.TryGetValue(newval.Key, out var tryold))
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, tryold.Value);
|
||||
if (newval.OldValue != null)
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, newval.OldValue);
|
||||
}
|
||||
return affrows;
|
||||
}
|
||||
|
||||
//等待下次对比再保存
|
||||
return 0;
|
||||
}
|
||||
async public Task UpdateAsync(TEntity data) {
|
||||
var exists = ExistsInStates(data);
|
||||
if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
if (exists == false) {
|
||||
var olddata = await OrmSelect(data).FirstAsync();
|
||||
if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_fsql.GetEntityString(_entityType, data)}");
|
||||
}
|
||||
//等待下次对比再保存
|
||||
return 0;
|
||||
}
|
||||
async public Task UpdateAsync(TEntity data)
|
||||
{
|
||||
var exists = ExistsInStates(data);
|
||||
if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
if (exists == false)
|
||||
{
|
||||
var olddata = await OrmSelect(data).FirstAsync();
|
||||
if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_fsql.GetEntityString(_entityType, data)}");
|
||||
}
|
||||
|
||||
await UpdateRangePrivAsync(new[] { data }, true);
|
||||
}
|
||||
public Task UpdateRangeAsync(IEnumerable<TEntity> data) => UpdateRangePrivAsync(data, true);
|
||||
async Task UpdateRangePrivAsync(IEnumerable<TEntity> data, bool isCheck) {
|
||||
if (CanUpdate(data, true) == false) return;
|
||||
foreach (var item in data) {
|
||||
if (_dicUpdateTimes.ContainsKey(item))
|
||||
await DbContextExecCommandAsync();
|
||||
_dicUpdateTimes.Add(item, 1);
|
||||
await UpdateRangePrivAsync(new[] { data }, true);
|
||||
}
|
||||
public Task UpdateRangeAsync(IEnumerable<TEntity> data) => UpdateRangePrivAsync(data, true);
|
||||
async Task UpdateRangePrivAsync(IEnumerable<TEntity> data, bool isCheck)
|
||||
{
|
||||
if (CanUpdate(data, true) == false) return;
|
||||
foreach (var item in data)
|
||||
{
|
||||
if (_dicUpdateTimes.ContainsKey(item))
|
||||
await DbContextExecCommandAsync();
|
||||
_dicUpdateTimes.Add(item, 1);
|
||||
|
||||
var state = CreateEntityState(item);
|
||||
state.OldValue = item;
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state);
|
||||
}
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
await AddOrUpdateNavigateListAsync(item);
|
||||
}
|
||||
#endregion
|
||||
var state = CreateEntityState(item);
|
||||
state.OldValue = item;
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state);
|
||||
}
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
await AddOrUpdateNavigateListAsync(item);
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region RemoveAsync
|
||||
async Task<int> DbContextBetchRemoveAsync(EntityState[] dels) {
|
||||
if (dels.Any() == false) return 0;
|
||||
var affrows = await this.OrmDelete(dels.Select(a => a.Value)).ExecuteAffrowsAsync();
|
||||
return Math.Max(dels.Length, affrows);
|
||||
}
|
||||
#endregion
|
||||
#region RemoveAsync
|
||||
async Task<int> DbContextBetchRemoveAsync(EntityState[] dels)
|
||||
{
|
||||
if (dels.Any() == false) return 0;
|
||||
var affrows = await this.OrmDelete(dels.Select(a => a.Value)).ExecuteAffrowsAsync();
|
||||
return Math.Max(dels.Length, affrows);
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region AddOrUpdateAsync
|
||||
async public Task AddOrUpdateAsync(TEntity data) {
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
#region AddOrUpdateAsync
|
||||
async public Task AddOrUpdateAsync(TEntity data)
|
||||
{
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
|
||||
var flagExists = ExistsInStates(data);
|
||||
if (flagExists == false) {
|
||||
var olddata = await OrmSelect(data).FirstAsync();
|
||||
if (olddata == null) flagExists = false;
|
||||
}
|
||||
var flagExists = ExistsInStates(data);
|
||||
if (flagExists == false)
|
||||
{
|
||||
var olddata = await OrmSelect(data).FirstAsync();
|
||||
if (olddata == null) flagExists = false;
|
||||
}
|
||||
|
||||
if (flagExists == true && CanUpdate(data, false)) {
|
||||
await DbContextExecCommandAsync();
|
||||
var affrows = _ctx._affrows;
|
||||
await UpdateRangePrivAsync(new[] { data }, false);
|
||||
await DbContextExecCommandAsync();
|
||||
affrows = _ctx._affrows - affrows;
|
||||
if (affrows > 0) return;
|
||||
}
|
||||
if (CanAdd(data, false)) {
|
||||
_fsql.ClearEntityPrimaryValueWithIdentity(_entityType, data);
|
||||
await AddPrivAsync(data, false);
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
if (flagExists == true && CanUpdate(data, false))
|
||||
{
|
||||
await DbContextExecCommandAsync();
|
||||
var affrows = _ctx._affrows;
|
||||
await UpdateRangePrivAsync(new[] { data }, false);
|
||||
await DbContextExecCommandAsync();
|
||||
affrows = _ctx._affrows - affrows;
|
||||
if (affrows > 0) return;
|
||||
}
|
||||
if (CanAdd(data, false))
|
||||
{
|
||||
_fsql.ClearEntityPrimaryValueWithIdentity(_entityType, data);
|
||||
await AddPrivAsync(data, false);
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
|
||||
@ -6,300 +6,344 @@ using System.Collections.Concurrent;
|
||||
using System.Linq;
|
||||
using System.Reflection;
|
||||
|
||||
namespace FreeSql {
|
||||
partial class DbSet<TEntity> {
|
||||
namespace FreeSql
|
||||
{
|
||||
partial class DbSet<TEntity>
|
||||
{
|
||||
|
||||
void DbContextExecCommand() {
|
||||
_dicUpdateTimes.Clear();
|
||||
_ctx.ExecCommand();
|
||||
}
|
||||
void DbContextExecCommand()
|
||||
{
|
||||
_dicUpdateTimes.Clear();
|
||||
_ctx.ExecCommand();
|
||||
}
|
||||
|
||||
int DbContextBetchAdd(EntityState[] adds) {
|
||||
if (adds.Any() == false) return 0;
|
||||
var affrows = this.OrmInsert(adds.Select(a => a.Value)).ExecuteAffrows();
|
||||
return affrows;
|
||||
}
|
||||
int DbContextBetchAdd(EntityState[] adds)
|
||||
{
|
||||
if (adds.Any() == false) return 0;
|
||||
var affrows = this.OrmInsert(adds.Select(a => a.Value)).ExecuteAffrows();
|
||||
return affrows;
|
||||
}
|
||||
|
||||
#region Add
|
||||
void AddPriv(TEntity data, bool isCheck) {
|
||||
if (isCheck && CanAdd(data, true) == false) return;
|
||||
if (_tableIdentitys.Length > 0) {
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType) {
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
if (_tableIdentitys.Length == 1) {
|
||||
DbContextExecCommand();
|
||||
var idtval = this.OrmInsert(data).ExecuteIdentity();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
} else {
|
||||
DbContextExecCommand();
|
||||
var newval = this.OrmInsert(data).ExecuteInserted().First();
|
||||
IncrAffrows(1);
|
||||
_fsql.MapEntityValue(_entityType, newval, data);
|
||||
Attach(newval);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
}
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
if (_tableIdentitys.Length == 1) {
|
||||
DbContextExecCommand();
|
||||
var idtval = this.OrmInsert(data).ExecuteIdentity();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
}
|
||||
return;
|
||||
}
|
||||
}
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data));
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
}
|
||||
/// <summary>
|
||||
/// 添加
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Add(TEntity data) => AddPriv(data, true);
|
||||
public void AddRange(IEnumerable<TEntity> data) {
|
||||
if (CanAdd(data, true) == false) return;
|
||||
if (data.ElementAtOrDefault(1) == default(TEntity)) {
|
||||
Add(data.First());
|
||||
return;
|
||||
}
|
||||
if (_tableIdentitys.Length > 0) {
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType) {
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
DbContextExecCommand();
|
||||
var rets = this.OrmInsert(data).ExecuteInserted();
|
||||
if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_fsql.Ado.DataType} 的返回数据，与添加的数目不匹配");
|
||||
var idx = 0;
|
||||
foreach (var s in data)
|
||||
_fsql.MapEntityValue(_entityType, rets[idx++], s);
|
||||
IncrAffrows(rets.Count);
|
||||
AttachRange(rets);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
AddOrUpdateNavigateList(item);
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
foreach (var s in data)
|
||||
AddPriv(s, false);
|
||||
return;
|
||||
}
|
||||
} else {
|
||||
//进入队列，等待 SaveChanges 时执行
|
||||
foreach (var item in data)
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item));
|
||||
AttachRange(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
AddOrUpdateNavigateList(item);
|
||||
}
|
||||
}
|
||||
static ConcurrentDictionary<Type, ConcurrentDictionary<string, FieldInfo>> _dicLazyIsSetField = new ConcurrentDictionary<Type, ConcurrentDictionary<string, FieldInfo>>();
|
||||
void AddOrUpdateNavigateList(TEntity item) {
|
||||
Type itemType = null;
|
||||
foreach (var prop in _table.Properties) {
|
||||
if (_table.ColumnsByCsIgnore.ContainsKey(prop.Key)) continue;
|
||||
if (_table.ColumnsByCs.ContainsKey(prop.Key)) continue;
|
||||
#region Add
|
||||
void AddPriv(TEntity data, bool isCheck)
|
||||
{
|
||||
if (isCheck && CanAdd(data, true) == false) return;
|
||||
if (_tableIdentitys.Length > 0)
|
||||
{
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType)
|
||||
{
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
if (_tableIdentitys.Length == 1)
|
||||
{
|
||||
DbContextExecCommand();
|
||||
var idtval = this.OrmInsert(data).ExecuteIdentity();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
}
|
||||
else
|
||||
{
|
||||
DbContextExecCommand();
|
||||
var newval = this.OrmInsert(data).ExecuteInserted().First();
|
||||
IncrAffrows(1);
|
||||
_fsql.MapEntityValue(_entityType, newval, data);
|
||||
Attach(newval);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
}
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
if (_tableIdentitys.Length == 1)
|
||||
{
|
||||
DbContextExecCommand();
|
||||
var idtval = this.OrmInsert(data).ExecuteIdentity();
|
||||
IncrAffrows(1);
|
||||
_fsql.SetEntityIdentityValueWithPrimary(_entityType, data, idtval);
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
}
|
||||
return;
|
||||
}
|
||||
}
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(data));
|
||||
Attach(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
AddOrUpdateNavigateList(data);
|
||||
}
|
||||
/// <summary>
|
||||
/// 添加
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Add(TEntity data) => AddPriv(data, true);
|
||||
public void AddRange(IEnumerable<TEntity> data)
|
||||
{
|
||||
if (CanAdd(data, true) == false) return;
|
||||
if (data.ElementAtOrDefault(1) == default(TEntity))
|
||||
{
|
||||
Add(data.First());
|
||||
return;
|
||||
}
|
||||
if (_tableIdentitys.Length > 0)
|
||||
{
|
||||
//有自增，马上执行
|
||||
switch (_fsql.Ado.DataType)
|
||||
{
|
||||
case DataType.SqlServer:
|
||||
case DataType.PostgreSQL:
|
||||
DbContextExecCommand();
|
||||
var rets = this.OrmInsert(data).ExecuteInserted();
|
||||
if (rets.Count != data.Count()) throw new Exception($"特别错误：批量添加失败，{_fsql.Ado.DataType} 的返回数据，与添加的数目不匹配");
|
||||
var idx = 0;
|
||||
foreach (var s in data)
|
||||
_fsql.MapEntityValue(_entityType, rets[idx++], s);
|
||||
IncrAffrows(rets.Count);
|
||||
AttachRange(rets);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
AddOrUpdateNavigateList(item);
|
||||
return;
|
||||
case DataType.MySql:
|
||||
case DataType.Oracle:
|
||||
case DataType.Sqlite:
|
||||
foreach (var s in data)
|
||||
AddPriv(s, false);
|
||||
return;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
//进入队列，等待 SaveChanges 时执行
|
||||
foreach (var item in data)
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Insert, CreateEntityState(item));
|
||||
AttachRange(data);
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
AddOrUpdateNavigateList(item);
|
||||
}
|
||||
}
|
||||
static ConcurrentDictionary<Type, ConcurrentDictionary<string, FieldInfo>> _dicLazyIsSetField = new ConcurrentDictionary<Type, ConcurrentDictionary<string, FieldInfo>>();
|
||||
void AddOrUpdateNavigateList(TEntity item)
|
||||
{
|
||||
Type itemType = null;
|
||||
foreach (var prop in _table.Properties)
|
||||
{
|
||||
if (_table.ColumnsByCsIgnore.ContainsKey(prop.Key)) continue;
|
||||
if (_table.ColumnsByCs.ContainsKey(prop.Key)) continue;
|
||||
|
||||
object propVal = null;
|
||||
object propVal = null;
|
||||
|
||||
if (itemType == null) itemType = item.GetType();
|
||||
if (_table.TypeLazy != null && itemType == _table.TypeLazy) {
|
||||
var lazyField = _dicLazyIsSetField.GetOrAdd(_table.TypeLazy, tl => new ConcurrentDictionary<string, FieldInfo>()).GetOrAdd(prop.Key, propName =>
|
||||
_table.TypeLazy.GetField($"__lazy__{propName}", BindingFlags.GetField | BindingFlags.NonPublic | BindingFlags.Instance));
|
||||
if (lazyField != null) {
|
||||
var lazyFieldValue = (bool)lazyField.GetValue(item);
|
||||
if (lazyFieldValue == false) continue;
|
||||
}
|
||||
propVal = prop.Value.GetValue(item);
|
||||
} else {
|
||||
propVal = prop.Value.GetValue(item);
|
||||
if (propVal == null) continue;
|
||||
}
|
||||
if (itemType == null) itemType = item.GetType();
|
||||
if (_table.TypeLazy != null && itemType == _table.TypeLazy)
|
||||
{
|
||||
var lazyField = _dicLazyIsSetField.GetOrAdd(_table.TypeLazy, tl => new ConcurrentDictionary<string, FieldInfo>()).GetOrAdd(prop.Key, propName =>
|
||||
_table.TypeLazy.GetField($"__lazy__{propName}", BindingFlags.GetField | BindingFlags.NonPublic | BindingFlags.Instance));
|
||||
if (lazyField != null)
|
||||
{
|
||||
var lazyFieldValue = (bool)lazyField.GetValue(item);
|
||||
if (lazyFieldValue == false) continue;
|
||||
}
|
||||
propVal = prop.Value.GetValue(item);
|
||||
}
|
||||
else
|
||||
{
|
||||
propVal = prop.Value.GetValue(item);
|
||||
if (propVal == null) continue;
|
||||
}
|
||||
|
||||
var tref = _table.GetTableRef(prop.Key, true);
|
||||
if (tref == null) continue;
|
||||
var tref = _table.GetTableRef(prop.Key, true);
|
||||
if (tref == null) continue;
|
||||
|
||||
switch(tref.RefType) {
|
||||
case Internal.Model.TableRefType.OneToOne:
|
||||
case Internal.Model.TableRefType.ManyToOne:
|
||||
case Internal.Model.TableRefType.ManyToMany:
|
||||
continue;
|
||||
case Internal.Model.TableRefType.OneToMany:
|
||||
var propValEach = propVal as IEnumerable;
|
||||
if (propValEach == null) continue;
|
||||
object dbset = null;
|
||||
MethodInfo dbsetAddOrUpdate = null;
|
||||
foreach (var propValItem in propValEach) {
|
||||
if (dbset == null) {
|
||||
dbset = _ctx.Set(tref.RefEntityType);
|
||||
dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdate", new Type[] { tref.RefEntityType });
|
||||
}
|
||||
for (var colidx = 0; colidx < tref.Columns.Count; colidx++) {
|
||||
tref.RefColumns[colidx].Table.Properties[tref.RefColumns[colidx].CsName]
|
||||
.SetValue(propValItem, tref.Columns[colidx].Table.Properties[tref.Columns[colidx].CsName].GetValue(item));
|
||||
}
|
||||
dbsetAddOrUpdate.Invoke(dbset, new object[] { propValItem });
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
switch (tref.RefType)
|
||||
{
|
||||
case Internal.Model.TableRefType.OneToOne:
|
||||
case Internal.Model.TableRefType.ManyToOne:
|
||||
case Internal.Model.TableRefType.ManyToMany:
|
||||
continue;
|
||||
case Internal.Model.TableRefType.OneToMany:
|
||||
var propValEach = propVal as IEnumerable;
|
||||
if (propValEach == null) continue;
|
||||
object dbset = null;
|
||||
MethodInfo dbsetAddOrUpdate = null;
|
||||
foreach (var propValItem in propValEach)
|
||||
{
|
||||
if (dbset == null)
|
||||
{
|
||||
dbset = _ctx.Set(tref.RefEntityType);
|
||||
dbsetAddOrUpdate = dbset.GetType().GetMethod("AddOrUpdate", new Type[] { tref.RefEntityType });
|
||||
}
|
||||
for (var colidx = 0; colidx < tref.Columns.Count; colidx++)
|
||||
{
|
||||
tref.RefColumns[colidx].Table.Properties[tref.RefColumns[colidx].CsName]
|
||||
.SetValue(propValItem, tref.Columns[colidx].Table.Properties[tref.Columns[colidx].CsName].GetValue(item));
|
||||
}
|
||||
dbsetAddOrUpdate.Invoke(dbset, new object[] { propValItem });
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region Update
|
||||
int DbContextBetchUpdate(EntityState[] ups) => DbContextBetchUpdatePriv(ups, false);
|
||||
int DbContextBetchUpdateNow(EntityState[] ups) => DbContextBetchUpdatePriv(ups, true);
|
||||
int DbContextBetchUpdatePriv(EntityState[] ups, bool isLiveUpdate) {
|
||||
if (ups.Any() == false) return 0;
|
||||
var uplst1 = ups[ups.Length - 1];
|
||||
var uplst2 = ups.Length > 1 ? ups[ups.Length - 2] : null;
|
||||
#region Update
|
||||
int DbContextBetchUpdate(EntityState[] ups) => DbContextBetchUpdatePriv(ups, false);
|
||||
int DbContextBetchUpdateNow(EntityState[] ups) => DbContextBetchUpdatePriv(ups, true);
|
||||
int DbContextBetchUpdatePriv(EntityState[] ups, bool isLiveUpdate)
|
||||
{
|
||||
if (ups.Any() == false) return 0;
|
||||
var uplst1 = ups[ups.Length - 1];
|
||||
var uplst2 = ups.Length > 1 ? ups[ups.Length - 2] : null;
|
||||
|
||||
if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999;
|
||||
var lstval2 = default(EntityState);
|
||||
if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_fsql.GetEntityString(_entityType, uplst2.Value)}");
|
||||
if (_states.TryGetValue(uplst1.Key, out var lstval1) == false) return -999;
|
||||
var lstval2 = default(EntityState);
|
||||
if (uplst2 != null && _states.TryGetValue(uplst2.Key, out lstval2) == false) throw new Exception($"特别错误：更新失败，数据未被跟踪：{_fsql.GetEntityString(_entityType, uplst2.Value)}");
|
||||
|
||||
var cuig1 = _fsql.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true);
|
||||
var cuig2 = uplst2 != null ? _fsql.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null;
|
||||
var cuig1 = _fsql.CompareEntityValueReturnColumns(_entityType, uplst1.Value, lstval1.Value, true);
|
||||
var cuig2 = uplst2 != null ? _fsql.CompareEntityValueReturnColumns(_entityType, uplst2.Value, lstval2.Value, true) : null;
|
||||
|
||||
List<EntityState> data = null;
|
||||
string[] cuig = null;
|
||||
if (uplst2 != null && string.Compare(string.Join(",", cuig1), string.Join(",", cuig2)) != 0) {
|
||||
//最后一个不保存
|
||||
data = ups.ToList();
|
||||
data.RemoveAt(ups.Length - 1);
|
||||
cuig = cuig2;
|
||||
} else if (isLiveUpdate) {
|
||||
//立即保存
|
||||
data = ups.ToList();
|
||||
cuig = cuig1;
|
||||
}
|
||||
List<EntityState> data = null;
|
||||
string[] cuig = null;
|
||||
if (uplst2 != null && string.Compare(string.Join(",", cuig1), string.Join(",", cuig2)) != 0)
|
||||
{
|
||||
//最后一个不保存
|
||||
data = ups.ToList();
|
||||
data.RemoveAt(ups.Length - 1);
|
||||
cuig = cuig2;
|
||||
}
|
||||
else if (isLiveUpdate)
|
||||
{
|
||||
//立即保存
|
||||
data = ups.ToList();
|
||||
cuig = cuig1;
|
||||
}
|
||||
|
||||
if (data?.Count > 0) {
|
||||
if (data?.Count > 0)
|
||||
{
|
||||
|
||||
if (cuig.Length == _table.Columns.Count)
|
||||
return ups.Length == data.Count ? -998 : -997;
|
||||
if (cuig.Length == _table.Columns.Count)
|
||||
return ups.Length == data.Count ? -998 : -997;
|
||||
|
||||
var updateSource = data.Select(a => a.Value).ToArray();
|
||||
var update = this.OrmUpdate(null).SetSource(updateSource).IgnoreColumns(cuig);
|
||||
var updateSource = data.Select(a => a.Value).ToArray();
|
||||
var update = this.OrmUpdate(null).SetSource(updateSource).IgnoreColumns(cuig);
|
||||
|
||||
var affrows = update.ExecuteAffrows();
|
||||
var affrows = update.ExecuteAffrows();
|
||||
|
||||
foreach (var newval in data) {
|
||||
if (_states.TryGetValue(newval.Key, out var tryold))
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, tryold.Value);
|
||||
if (newval.OldValue != null)
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, newval.OldValue);
|
||||
}
|
||||
return affrows;
|
||||
}
|
||||
foreach (var newval in data)
|
||||
{
|
||||
if (_states.TryGetValue(newval.Key, out var tryold))
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, tryold.Value);
|
||||
if (newval.OldValue != null)
|
||||
_fsql.MapEntityValue(_entityType, newval.Value, newval.OldValue);
|
||||
}
|
||||
return affrows;
|
||||
}
|
||||
|
||||
//等待下次对比再保存
|
||||
return 0;
|
||||
}
|
||||
//等待下次对比再保存
|
||||
return 0;
|
||||
}
|
||||
|
||||
Dictionary<TEntity, byte> _dicUpdateTimes = new Dictionary<TEntity, byte>();
|
||||
/// <summary>
|
||||
/// 更新
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Update(TEntity data) {
|
||||
var exists = ExistsInStates(data);
|
||||
if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
if (exists == false) {
|
||||
var olddata = OrmSelect(data).First();
|
||||
if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_fsql.GetEntityString(_entityType, data)}");
|
||||
}
|
||||
Dictionary<TEntity, byte> _dicUpdateTimes = new Dictionary<TEntity, byte>();
|
||||
/// <summary>
|
||||
/// 更新
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Update(TEntity data)
|
||||
{
|
||||
var exists = ExistsInStates(data);
|
||||
if (exists == null) throw new Exception($"不可更新，未设置主键的值：{_fsql.GetEntityString(_entityType, data)}");
|
||||
if (exists == false)
|
||||
{
|
||||
var olddata = OrmSelect(data).First();
|
||||
if (olddata == null) throw new Exception($"不可更新，数据库不存在该记录：{_fsql.GetEntityString(_entityType, data)}");
|
||||
}
|
||||
|
||||
UpdateRangePriv(new[] { data }, true);
|
||||
}
|
||||
public void UpdateRange(IEnumerable<TEntity> data) => UpdateRangePriv(data, true);
|
||||
void UpdateRangePriv(IEnumerable<TEntity> data, bool isCheck) {
|
||||
if (CanUpdate(data, true) == false) return;
|
||||
foreach (var item in data) {
|
||||
if (_dicUpdateTimes.ContainsKey(item))
|
||||
DbContextExecCommand();
|
||||
_dicUpdateTimes.Add(item, 1);
|
||||
UpdateRangePriv(new[] { data }, true);
|
||||
}
|
||||
public void UpdateRange(IEnumerable<TEntity> data) => UpdateRangePriv(data, true);
|
||||
void UpdateRangePriv(IEnumerable<TEntity> data, bool isCheck)
|
||||
{
|
||||
if (CanUpdate(data, true) == false) return;
|
||||
foreach (var item in data)
|
||||
{
|
||||
if (_dicUpdateTimes.ContainsKey(item))
|
||||
DbContextExecCommand();
|
||||
_dicUpdateTimes.Add(item, 1);
|
||||
|
||||
var state = CreateEntityState(item);
|
||||
state.OldValue = item;
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state);
|
||||
}
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
AddOrUpdateNavigateList(item);
|
||||
}
|
||||
#endregion
|
||||
var state = CreateEntityState(item);
|
||||
state.OldValue = item;
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Update, state);
|
||||
}
|
||||
if (_ctx.Options.EnableAddOrUpdateNavigateList)
|
||||
foreach (var item in data)
|
||||
AddOrUpdateNavigateList(item);
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region Remove
|
||||
int DbContextBetchRemove(EntityState[] dels) {
|
||||
if (dels.Any() == false) return 0;
|
||||
var affrows = this.OrmDelete(dels.Select(a => a.Value)).ExecuteAffrows();
|
||||
return Math.Max(dels.Length, affrows);
|
||||
}
|
||||
#region Remove
|
||||
int DbContextBetchRemove(EntityState[] dels)
|
||||
{
|
||||
if (dels.Any() == false) return 0;
|
||||
var affrows = this.OrmDelete(dels.Select(a => a.Value)).ExecuteAffrows();
|
||||
return Math.Max(dels.Length, affrows);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 删除
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Remove(TEntity data) => RemoveRange(new[] { data });
|
||||
public void RemoveRange(IEnumerable<TEntity> data) {
|
||||
if (CanRemove(data, true) == false) return;
|
||||
foreach (var item in data) {
|
||||
var state = CreateEntityState(item);
|
||||
_states.TryRemove(state.Key, out var trystate);
|
||||
_fsql.ClearEntityPrimaryValueWithIdentityAndGuid(_entityType, item);
|
||||
/// <summary>
|
||||
/// 删除
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void Remove(TEntity data) => RemoveRange(new[] { data });
|
||||
public void RemoveRange(IEnumerable<TEntity> data)
|
||||
{
|
||||
if (CanRemove(data, true) == false) return;
|
||||
foreach (var item in data)
|
||||
{
|
||||
var state = CreateEntityState(item);
|
||||
_states.TryRemove(state.Key, out var trystate);
|
||||
_fsql.ClearEntityPrimaryValueWithIdentityAndGuid(_entityType, item);
|
||||
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Delete, state);
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
EnqueueToDbContext(DbContext.ExecCommandInfoType.Delete, state);
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region AddOrUpdate
|
||||
/// <summary>
|
||||
/// 添加或更新
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void AddOrUpdate(TEntity data) {
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
#region AddOrUpdate
|
||||
/// <summary>
|
||||
/// 添加或更新
|
||||
/// </summary>
|
||||
/// <param name="data"></param>
|
||||
public void AddOrUpdate(TEntity data)
|
||||
{
|
||||
if (data == null) throw new ArgumentNullException(nameof(data));
|
||||
if (_table.Primarys.Any() == false) throw new Exception($"不可添加，实体没有主键：{_fsql.GetEntityString(_entityType, data)}");
|
||||
|
||||
var flagExists = ExistsInStates(data);
|
||||
if (flagExists == false) {
|
||||
var olddata = OrmSelect(data).First();
|
||||
if (olddata == null) flagExists = false;
|
||||
}
|
||||
var flagExists = ExistsInStates(data);
|
||||
if (flagExists == false)
|
||||
{
|
||||
var olddata = OrmSelect(data).First();
|
||||
if (olddata == null) flagExists = false;
|
||||
}
|
||||
|
||||
if (flagExists == true && CanUpdate(data, false)) {
|
||||
DbContextExecCommand();
|
||||
var affrows = _ctx._affrows;
|
||||
UpdateRangePriv(new[] { data }, false);
|
||||
DbContextExecCommand();
|
||||
affrows = _ctx._affrows - affrows;
|
||||
if (affrows > 0) return;
|
||||
}
|
||||
if (CanAdd(data, false)) {
|
||||
_fsql.ClearEntityPrimaryValueWithIdentity(_entityType, data);
|
||||
AddPriv(data, false);
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
if (flagExists == true && CanUpdate(data, false))
|
||||
{
|
||||
DbContextExecCommand();
|
||||
var affrows = _ctx._affrows;
|
||||
UpdateRangePriv(new[] { data }, false);
|
||||
DbContextExecCommand();
|
||||
affrows = _ctx._affrows - affrows;
|
||||
if (affrows > 0) return;
|
||||
}
|
||||
if (CanAdd(data, false))
|
||||
{
|
||||
_fsql.ClearEntityPrimaryValueWithIdentity(_entityType, data);
|
||||
AddPriv(data, false);
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user