mirror of
https://gitee.com/dotnetchina/SqlSugar.git
synced 2025-05-16 04:59:34 +08:00
Update .net core project
This commit is contained in:
sunkaixuan
parent
7d87ec1b54
commit
c1a49140c3
@ -0,0 +1,516 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Data;
|
||||
using System.Linq;
|
||||
using System.Linq.Expressions;
|
||||
using System.Reflection;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace SqlSugar
|
||||
{
|
||||
public partial class InsertableProvider<T> : IInsertable<T> where T : class, new()
|
||||
{
|
||||
#region Protected Methods
|
||||
private string _ExecuteReturnBigIdentity()
|
||||
{
|
||||
InsertBuilder.IsReturnIdentity = true;
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
string sql = InsertBuilder.ToSqlString();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
private string _ExecuteReturnIdentity()
|
||||
{
|
||||
InsertBuilder.IsReturnIdentity = true;
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
string sql = InsertBuilder.ToSqlString();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
|
||||
private string _ExecuteCommand()
|
||||
{
|
||||
if (InsertBuilder.DbColumnInfoList.HasValue())
|
||||
{
|
||||
var pks = GetPrimaryKeys();
|
||||
foreach (var item in InsertBuilder.DbColumnInfoList)
|
||||
{
|
||||
var isPk = pks.Any(y => y.Equals(item.DbColumnName, StringComparison.CurrentCultureIgnoreCase)) || item.IsPrimarykey;
|
||||
if (isPk && item.PropertyType == UtilConstants.GuidType && item.Value.ObjToString() == Guid.Empty.ToString())
|
||||
{
|
||||
item.Value = Guid.NewGuid();
|
||||
if (InsertObjs.First().GetType().GetProperties().Any(it => it.Name == item.PropertyName))
|
||||
InsertObjs.First().GetType().GetProperties().First(it => it.Name == item.PropertyName).SetValue(InsertObjs.First(), item.Value, null);
|
||||
}
|
||||
}
|
||||
}
|
||||
InsertBuilder.IsReturnIdentity = false;
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
string sql = InsertBuilder.ToSqlString();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
private void AutoRemoveDataCache()
|
||||
{
|
||||
var moreSetts = this.Context.CurrentConnectionConfig.MoreSettings;
|
||||
var extService = this.Context.CurrentConnectionConfig.ConfigureExternalServices;
|
||||
if (moreSetts != null && moreSetts.IsAutoRemoveDataCache && extService != null && extService.DataInfoCacheService != null)
|
||||
{
|
||||
this.RemoveDataCache();
|
||||
}
|
||||
}
|
||||
protected virtual void PreToSql()
|
||||
{
|
||||
#region Identities
|
||||
if (!IsOffIdentity)
|
||||
{
|
||||
List<string> identities = GetIdentityKeys();
|
||||
if (identities != null && identities.Any())
|
||||
{
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !identities.Any(i => it.DbColumnName.Equals(i, StringComparison.CurrentCultureIgnoreCase));
|
||||
}).ToList();
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region IgnoreColumns
|
||||
if (this.Context.IgnoreColumns != null && this.Context.IgnoreColumns.Any())
|
||||
{
|
||||
var currentIgnoreColumns = this.Context.IgnoreColumns.Where(it => it.EntityName == this.EntityInfo.EntityName).ToList();
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !currentIgnoreColumns.Any(i => it.PropertyName.Equals(i.PropertyName, StringComparison.CurrentCulture));
|
||||
}).ToList();
|
||||
}
|
||||
|
||||
if (this.Context.IgnoreInsertColumns != null && this.Context.IgnoreInsertColumns.Any())
|
||||
{
|
||||
var currentIgnoreColumns = this.Context.IgnoreInsertColumns.Where(it => it.EntityName == this.EntityInfo.EntityName).ToList();
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !currentIgnoreColumns.Any(i => it.PropertyName.Equals(i.PropertyName, StringComparison.CurrentCulture));
|
||||
}).ToList();
|
||||
}
|
||||
#endregion
|
||||
if (this.IsSingle)
|
||||
{
|
||||
foreach (var item in this.InsertBuilder.DbColumnInfoList)
|
||||
{
|
||||
if (this.InsertBuilder.Parameters == null) this.InsertBuilder.Parameters = new List<SugarParameter>();
|
||||
var paramters = new SugarParameter(this.SqlBuilder.SqlParameterKeyWord + item.DbColumnName, item.Value, item.PropertyType);
|
||||
if (InsertBuilder.IsNoInsertNull && paramters.Value == null)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
if (item.IsJson)
|
||||
{
|
||||
paramters.IsJson = true;
|
||||
}
|
||||
if (item.IsArray)
|
||||
{
|
||||
paramters.IsArray = true;
|
||||
}
|
||||
this.InsertBuilder.Parameters.Add(paramters);
|
||||
}
|
||||
}
|
||||
}
|
||||
internal void Init()
|
||||
{
|
||||
InsertBuilder.EntityInfo = this.EntityInfo;
|
||||
Check.Exception(InsertObjs == null || InsertObjs.Count() == 0, "InsertObjs is null");
|
||||
int i = 0;
|
||||
foreach (var item in InsertObjs)
|
||||
{
|
||||
List<DbColumnInfo> insertItem = new List<DbColumnInfo>();
|
||||
if (item is Dictionary<string, object>)
|
||||
{
|
||||
SetInsertItemByDic(i, item, insertItem);
|
||||
}
|
||||
else
|
||||
{
|
||||
DataAop(item);
|
||||
SetInsertItemByEntity(i, item, insertItem);
|
||||
}
|
||||
this.InsertBuilder.DbColumnInfoList.AddRange(insertItem);
|
||||
++i;
|
||||
}
|
||||
}
|
||||
|
||||
private void DataAop(T item)
|
||||
{
|
||||
var dataEvent = this.Context.CurrentConnectionConfig.AopEvents?.DataExecuting;
|
||||
if (dataEvent != null && item != null)
|
||||
{
|
||||
foreach (var columnInfo in this.EntityInfo.Columns)
|
||||
{
|
||||
dataEvent(columnInfo.PropertyInfo.GetValue(item, null), new DataFilterModel() { OperationType = DataFilterType.InsertByObject, EntityValue = item, EntityColumnInfo = columnInfo });
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void SetInsertItemByDic(int i, T item, List<DbColumnInfo> insertItem)
|
||||
{
|
||||
foreach (var column in item as Dictionary<string, object>)
|
||||
{
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = column.Value,
|
||||
DbColumnName = column.Key,
|
||||
PropertyName = column.Key,
|
||||
PropertyType = column.Value == null ? DBNull.Value.GetType() : UtilMethods.GetUnderType(column.Value.GetType()),
|
||||
TableId = i
|
||||
};
|
||||
if (columnInfo.PropertyType.IsEnum())
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
insertItem.Add(columnInfo);
|
||||
}
|
||||
}
|
||||
private void SetInsertItemByEntity(int i, T item, List<DbColumnInfo> insertItem)
|
||||
{
|
||||
if (item == null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
foreach (var column in EntityInfo.Columns)
|
||||
{
|
||||
if (column.IsIgnore || column.IsOnlyIgnoreInsert) continue;
|
||||
var isMapping = IsMappingColumns;
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = PropertyCallAdapterProvider<T>.GetInstance(column.PropertyName).InvokeGet(item),
|
||||
DbColumnName = column.DbColumnName,
|
||||
PropertyName = column.PropertyName,
|
||||
PropertyType = UtilMethods.GetUnderType(column.PropertyInfo),
|
||||
TableId = i
|
||||
};
|
||||
if (column.DbColumnName == null)
|
||||
{
|
||||
column.DbColumnName = column.PropertyName;
|
||||
}
|
||||
if (isMapping)
|
||||
{
|
||||
columnInfo.DbColumnName = GetDbColumnName(column.PropertyName);
|
||||
}
|
||||
if (column.IsJson)
|
||||
{
|
||||
columnInfo.IsJson = true;
|
||||
}
|
||||
if (column.IsArray)
|
||||
{
|
||||
columnInfo.IsArray = true;
|
||||
}
|
||||
if (columnInfo.PropertyType.IsEnum() && columnInfo.Value != null)
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
if (column.IsJson && columnInfo.Value != null)
|
||||
{
|
||||
if (columnInfo.Value != null)
|
||||
columnInfo.Value = this.Context.Utilities.SerializeObject(columnInfo.Value);
|
||||
}
|
||||
//var tranColumn=EntityInfo.Columns.FirstOrDefault(it => it.IsTranscoding && it.DbColumnName.Equals(column.DbColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (column.IsTranscoding && columnInfo.Value.HasValue())
|
||||
{
|
||||
columnInfo.Value = UtilMethods.EncodeBase64(columnInfo.Value.ToString());
|
||||
}
|
||||
insertItem.Add(columnInfo);
|
||||
}
|
||||
}
|
||||
|
||||
private string GetDbColumnName(string propertyName)
|
||||
{
|
||||
if (!IsMappingColumns)
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
if (this.Context.MappingColumns.Any(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)))
|
||||
{
|
||||
this.MappingColumnList = this.Context.MappingColumns.Where(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)).ToList();
|
||||
}
|
||||
if (MappingColumnList == null || !MappingColumnList.Any())
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
else
|
||||
{
|
||||
var mappInfo = this.MappingColumnList.FirstOrDefault(it => it.PropertyName.Equals(propertyName, StringComparison.CurrentCultureIgnoreCase));
|
||||
return mappInfo == null ? propertyName : mappInfo.DbColumnName;
|
||||
}
|
||||
}
|
||||
|
||||
protected virtual List<string> GetPrimaryKeys()
|
||||
{
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetPrimaries(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => it.IsPrimarykey).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
protected virtual List<string> GetIdentityKeys()
|
||||
{
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetIsIdentities(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => {
|
||||
|
||||
Check.Exception(it.IsIdentity && it.UnderType == typeof(string), "IsIdentity key can not be type of string");
|
||||
return it.IsIdentity;
|
||||
|
||||
}).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
//private void TaskStart<Type>(Task<Type> result)
|
||||
//{
|
||||
// if (this.Context.CurrentConnectionConfig.IsShardSameThread)
|
||||
// {
|
||||
// Check.Exception(true, "IsShardSameThread=true can't be used async method");
|
||||
// }
|
||||
// result.Start();
|
||||
//}
|
||||
protected void RestoreMapping()
|
||||
{
|
||||
if (IsAs)
|
||||
{
|
||||
this.Context.MappingTables = OldMappingTableList;
|
||||
}
|
||||
}
|
||||
//protected IInsertable<T> CopyInsertable()
|
||||
//{
|
||||
// var asyncContext = this.Context.Utilities.CopyContext(true);
|
||||
// asyncContext.CurrentConnectionConfig.IsAutoCloseConnection = true;
|
||||
// asyncContext.IsAsyncMethod = true;
|
||||
// var asyncInsertable = asyncContext.Insertable<T>(this.InsertObjs);
|
||||
// var asyncInsertableBuilder = asyncInsertable.InsertBuilder;
|
||||
// asyncInsertableBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList;
|
||||
// asyncInsertableBuilder.EntityInfo = this.InsertBuilder.EntityInfo;
|
||||
// asyncInsertableBuilder.Parameters = this.InsertBuilder.Parameters;
|
||||
// asyncInsertableBuilder.sql = this.InsertBuilder.sql;
|
||||
// asyncInsertableBuilder.IsNoInsertNull = this.InsertBuilder.IsNoInsertNull;
|
||||
// asyncInsertableBuilder.IsReturnIdentity = this.InsertBuilder.IsReturnIdentity;
|
||||
// asyncInsertableBuilder.EntityInfo = this.InsertBuilder.EntityInfo;
|
||||
// asyncInsertableBuilder.TableWithString = this.InsertBuilder.TableWithString;
|
||||
// if (this.RemoveCacheFunc != null)
|
||||
// {
|
||||
// asyncInsertable.RemoveDataCache();
|
||||
// }
|
||||
// return asyncInsertable;
|
||||
//}
|
||||
|
||||
protected void After(string sql, long? result)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = InsertBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.AfterData = GetDiffTable(sql, result);
|
||||
diffModel.Time = this.Context.Ado.SqlExecutionTime;
|
||||
if (this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent != null)
|
||||
this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent(diffModel);
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
if (this.RemoveCacheFunc != null)
|
||||
{
|
||||
this.RemoveCacheFunc();
|
||||
}
|
||||
}
|
||||
protected void Before(string sql)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = InsertBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.BeforeData = null;
|
||||
diffModel.Sql = sql;
|
||||
diffModel.Parameters = parameters.ToArray();
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
}
|
||||
private List<DiffLogTableInfo> GetDiffTable(string sql, long? identity)
|
||||
{
|
||||
|
||||
if (GetIdentityKeys().HasValue() && this.InsertObjs.Count() > 1)
|
||||
{
|
||||
return GetDiffTableByEntity();
|
||||
}
|
||||
else if (GetIdentityKeys().IsNullOrEmpty())
|
||||
{
|
||||
return GetDiffTableByEntity();
|
||||
}
|
||||
else
|
||||
{
|
||||
return GetDiffTableBySql(identity);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private List<DiffLogTableInfo> GetDiffTableByEntity()
|
||||
{
|
||||
List<SugarParameter> parameters = new List<SugarParameter>();
|
||||
List<DiffLogTableInfo> result = new List<DiffLogTableInfo>();
|
||||
var dt2 = this.Context.Utilities.ListToDataTable<T>(this.InsertObjs.ToList());
|
||||
foreach (DataRow row in dt2.Rows)
|
||||
{
|
||||
DiffLogTableInfo item = new DiffLogTableInfo();
|
||||
item.TableDescription = this.EntityInfo.TableDescription;
|
||||
item.TableName = this.EntityInfo.DbTableName;
|
||||
item.Columns = new List<DiffLogColumnInfo>();
|
||||
foreach (DataColumn col in dt2.Columns)
|
||||
{
|
||||
var sugarColumn = this.EntityInfo.Columns.Where(it => it.DbColumnName != null).FirstOrDefault(it =>
|
||||
it.DbColumnName.Equals(col.ColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
DiffLogColumnInfo addItem = new DiffLogColumnInfo();
|
||||
addItem.Value = row[col.ColumnName];
|
||||
addItem.ColumnName = col.ColumnName;
|
||||
addItem.IsPrimaryKey = sugarColumn?.IsPrimarykey ?? false;
|
||||
addItem.ColumnDescription = sugarColumn?.ColumnDescription;
|
||||
item.Columns.Add(addItem);
|
||||
}
|
||||
result.Add(item);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
private List<DiffLogTableInfo> GetDiffTableBySql(long? identity)
|
||||
{
|
||||
List<SugarParameter> parameters = new List<SugarParameter>();
|
||||
List<DiffLogTableInfo> result = new List<DiffLogTableInfo>();
|
||||
var whereSql = string.Empty;
|
||||
List<IConditionalModel> cons = new List<IConditionalModel>();
|
||||
if (identity != null && identity > 0 && GetIdentityKeys().HasValue())
|
||||
{
|
||||
var fieldName = GetIdentityKeys().Last();
|
||||
|
||||
if (this.Context.CurrentConnectionConfig.DbType == DbType.PostgreSQL)
|
||||
{
|
||||
var fieldObjectType = this.EntityInfo.Columns.FirstOrDefault(x => x.DbColumnName == fieldName)
|
||||
.PropertyInfo.PropertyType;
|
||||
cons.Add(new ConditionalModel()
|
||||
{
|
||||
ConditionalType = ConditionalType.Equal,
|
||||
FieldName = fieldName,
|
||||
FieldValue = identity.ToString(),
|
||||
FieldValueConvertFunc = it => UtilMethods.ChangeType2(it, fieldObjectType)
|
||||
});
|
||||
}
|
||||
else
|
||||
cons.Add(new ConditionalModel() { ConditionalType = ConditionalType.Equal, FieldName = fieldName, FieldValue = identity.ToString() });
|
||||
}
|
||||
else
|
||||
{
|
||||
foreach (var item in this.EntityInfo.Columns.Where(it => it.IsIgnore == false && GetPrimaryKeys().Any(pk => pk.Equals(it.DbColumnName, StringComparison.CurrentCultureIgnoreCase))))
|
||||
{
|
||||
var fielddName = item.DbColumnName;
|
||||
var filedObject = this.EntityInfo.Columns.FirstOrDefault(it => it.PropertyName == item.PropertyName).PropertyInfo.GetValue(this.InsertObjs.Last(), null);
|
||||
var fieldValue = filedObject.ObjToString();
|
||||
if (filedObject != null && filedObject.GetType() != typeof(string) && this.Context.CurrentConnectionConfig.DbType == DbType.PostgreSQL)
|
||||
{
|
||||
cons.Add(new ConditionalModel() { ConditionalType = ConditionalType.Equal, FieldName = fielddName, FieldValue = fieldValue, FieldValueConvertFunc = it => UtilMethods.ChangeType2(it, filedObject.GetType()) });
|
||||
}
|
||||
else
|
||||
{
|
||||
cons.Add(new ConditionalModel() { ConditionalType = ConditionalType.Equal, FieldName = fielddName, FieldValue = fieldValue });
|
||||
}
|
||||
}
|
||||
}
|
||||
Check.Exception(cons.IsNullOrEmpty(), "Insertable.EnableDiffLogEvent need primary key");
|
||||
var sqlable = this.SqlBuilder.ConditionalModelToSql(cons);
|
||||
whereSql = sqlable.Key;
|
||||
parameters.AddRange(sqlable.Value);
|
||||
var dt = this.Context.Queryable<T>().Where(whereSql).AddParameters(parameters).ToDataTable();
|
||||
if (dt.Rows != null && dt.Rows.Count > 0)
|
||||
{
|
||||
foreach (DataRow row in dt.Rows)
|
||||
{
|
||||
DiffLogTableInfo item = new DiffLogTableInfo();
|
||||
item.TableDescription = this.EntityInfo.TableDescription;
|
||||
item.TableName = this.EntityInfo.DbTableName;
|
||||
item.Columns = new List<DiffLogColumnInfo>();
|
||||
foreach (DataColumn col in dt.Columns)
|
||||
{
|
||||
var sugarColumn = this.EntityInfo.Columns.Where(it => it.DbColumnName != null).First(it =>
|
||||
it.DbColumnName.Equals(col.ColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
DiffLogColumnInfo addItem = new DiffLogColumnInfo();
|
||||
addItem.Value = row[col.ColumnName];
|
||||
addItem.ColumnName = col.ColumnName;
|
||||
addItem.IsPrimaryKey = sugarColumn.IsPrimarykey;
|
||||
addItem.ColumnDescription = sugarColumn.ColumnDescription;
|
||||
item.Columns.Add(addItem);
|
||||
}
|
||||
result.Add(item);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
else
|
||||
{
|
||||
DiffLogTableInfo diffTable = new DiffLogTableInfo();
|
||||
diffTable.TableName = this.EntityInfo.DbTableName;
|
||||
diffTable.TableDescription = this.EntityInfo.TableDescription;
|
||||
diffTable.Columns = this.EntityInfo.Columns.Where(it => it.IsIgnore == false).Select(it => new DiffLogColumnInfo()
|
||||
{
|
||||
ColumnDescription = it.ColumnDescription,
|
||||
ColumnName = it.DbColumnName,
|
||||
Value = it.PropertyInfo.GetValue(this.InsertObjs.Last(), null),
|
||||
IsPrimaryKey = it.IsPrimarykey
|
||||
}).ToList();
|
||||
return new List<DiffLogTableInfo>() { diffTable };
|
||||
}
|
||||
}
|
||||
|
||||
public IInsertable<T> CallEntityMethod(Expression<Action<T>> method)
|
||||
{
|
||||
if (this.InsertObjs.HasValue())
|
||||
{
|
||||
var oldColumns = this.InsertBuilder.DbColumnInfoList.Select(it => it.PropertyName).ToList();
|
||||
var expression = (LambdaExpression.Lambda(method).Body as LambdaExpression).Body;
|
||||
Check.Exception(!(expression is MethodCallExpression), method.ToString() + " is not method");
|
||||
var callExpresion = expression as MethodCallExpression;
|
||||
UtilMethods.DataInoveByExpresson(this.InsertObjs, callExpresion);
|
||||
this.InsertBuilder.DbColumnInfoList = new List<DbColumnInfo>();
|
||||
Init();
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it => oldColumns.Contains(it.PropertyName)).ToList();
|
||||
}
|
||||
return this;
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
@ -9,7 +9,7 @@ using System.Threading.Tasks;
|
||||
|
||||
namespace SqlSugar
|
||||
{
|
||||
public class InsertableProvider<T> : IInsertable<T> where T : class, new()
|
||||
public partial class InsertableProvider<T> : IInsertable<T> where T : class, new()
|
||||
{
|
||||
public SqlSugarProvider Context { get; set; }
|
||||
public IAdo Ado { get { return Context.Ado; } }
|
||||
@ -518,501 +518,5 @@ namespace SqlSugar
|
||||
|
||||
#endregion
|
||||
|
||||
#region Protected Methods
|
||||
private string _ExecuteReturnBigIdentity()
|
||||
{
|
||||
InsertBuilder.IsReturnIdentity = true;
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
string sql = InsertBuilder.ToSqlString();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
private string _ExecuteReturnIdentity()
|
||||
{
|
||||
InsertBuilder.IsReturnIdentity = true;
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
string sql = InsertBuilder.ToSqlString();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
|
||||
private string _ExecuteCommand()
|
||||
{
|
||||
if (InsertBuilder.DbColumnInfoList.HasValue())
|
||||
{
|
||||
var pks = GetPrimaryKeys();
|
||||
foreach (var item in InsertBuilder.DbColumnInfoList)
|
||||
{
|
||||
var isPk = pks.Any(y => y.Equals(item.DbColumnName, StringComparison.CurrentCultureIgnoreCase)) || item.IsPrimarykey;
|
||||
if (isPk && item.PropertyType == UtilConstants.GuidType && item.Value.ObjToString() == Guid.Empty.ToString())
|
||||
{
|
||||
item.Value = Guid.NewGuid();
|
||||
if (InsertObjs.First().GetType().GetProperties().Any(it => it.Name == item.PropertyName))
|
||||
InsertObjs.First().GetType().GetProperties().First(it => it.Name == item.PropertyName).SetValue(InsertObjs.First(), item.Value, null);
|
||||
}
|
||||
}
|
||||
}
|
||||
InsertBuilder.IsReturnIdentity = false;
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
string sql = InsertBuilder.ToSqlString();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
private void AutoRemoveDataCache()
|
||||
{
|
||||
var moreSetts = this.Context.CurrentConnectionConfig.MoreSettings;
|
||||
var extService = this.Context.CurrentConnectionConfig.ConfigureExternalServices;
|
||||
if (moreSetts != null && moreSetts.IsAutoRemoveDataCache && extService != null && extService.DataInfoCacheService != null)
|
||||
{
|
||||
this.RemoveDataCache();
|
||||
}
|
||||
}
|
||||
protected virtual void PreToSql()
|
||||
{
|
||||
#region Identities
|
||||
if (!IsOffIdentity)
|
||||
{
|
||||
List<string> identities = GetIdentityKeys();
|
||||
if (identities != null && identities.Any())
|
||||
{
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !identities.Any(i => it.DbColumnName.Equals(i, StringComparison.CurrentCultureIgnoreCase));
|
||||
}).ToList();
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region IgnoreColumns
|
||||
if (this.Context.IgnoreColumns != null && this.Context.IgnoreColumns.Any())
|
||||
{
|
||||
var currentIgnoreColumns = this.Context.IgnoreColumns.Where(it => it.EntityName == this.EntityInfo.EntityName).ToList();
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !currentIgnoreColumns.Any(i => it.PropertyName.Equals(i.PropertyName, StringComparison.CurrentCulture));
|
||||
}).ToList();
|
||||
}
|
||||
|
||||
if (this.Context.IgnoreInsertColumns != null && this.Context.IgnoreInsertColumns.Any())
|
||||
{
|
||||
var currentIgnoreColumns = this.Context.IgnoreInsertColumns.Where(it => it.EntityName == this.EntityInfo.EntityName).ToList();
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !currentIgnoreColumns.Any(i => it.PropertyName.Equals(i.PropertyName, StringComparison.CurrentCulture));
|
||||
}).ToList();
|
||||
}
|
||||
#endregion
|
||||
if (this.IsSingle)
|
||||
{
|
||||
foreach (var item in this.InsertBuilder.DbColumnInfoList)
|
||||
{
|
||||
if (this.InsertBuilder.Parameters == null) this.InsertBuilder.Parameters = new List<SugarParameter>();
|
||||
var paramters = new SugarParameter(this.SqlBuilder.SqlParameterKeyWord + item.DbColumnName, item.Value, item.PropertyType);
|
||||
if (InsertBuilder.IsNoInsertNull && paramters.Value == null)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
if (item.IsJson)
|
||||
{
|
||||
paramters.IsJson = true;
|
||||
}
|
||||
if (item.IsArray)
|
||||
{
|
||||
paramters.IsArray = true;
|
||||
}
|
||||
this.InsertBuilder.Parameters.Add(paramters);
|
||||
}
|
||||
}
|
||||
}
|
||||
internal void Init()
|
||||
{
|
||||
InsertBuilder.EntityInfo = this.EntityInfo;
|
||||
Check.Exception(InsertObjs == null || InsertObjs.Count() == 0, "InsertObjs is null");
|
||||
int i = 0;
|
||||
foreach (var item in InsertObjs)
|
||||
{
|
||||
List<DbColumnInfo> insertItem = new List<DbColumnInfo>();
|
||||
if (item is Dictionary<string, object>)
|
||||
{
|
||||
SetInsertItemByDic(i, item, insertItem);
|
||||
}
|
||||
else
|
||||
{
|
||||
DataAop(item);
|
||||
SetInsertItemByEntity(i, item, insertItem);
|
||||
}
|
||||
this.InsertBuilder.DbColumnInfoList.AddRange(insertItem);
|
||||
++i;
|
||||
}
|
||||
}
|
||||
|
||||
private void DataAop(T item)
|
||||
{
|
||||
var dataEvent=this.Context.CurrentConnectionConfig.AopEvents?.DataExecuting;
|
||||
if (dataEvent != null && item != null)
|
||||
{
|
||||
foreach (var columnInfo in this.EntityInfo.Columns)
|
||||
{
|
||||
dataEvent(columnInfo.PropertyInfo.GetValue(item, null), new DataFilterModel() { OperationType = DataFilterType.InsertByObject,EntityValue=item, EntityColumnInfo = columnInfo });
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void SetInsertItemByDic(int i, T item, List<DbColumnInfo> insertItem)
|
||||
{
|
||||
foreach (var column in item as Dictionary<string, object>)
|
||||
{
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = column.Value,
|
||||
DbColumnName = column.Key,
|
||||
PropertyName = column.Key,
|
||||
PropertyType = column.Value == null ? DBNull.Value.GetType() : UtilMethods.GetUnderType(column.Value.GetType()),
|
||||
TableId = i
|
||||
};
|
||||
if (columnInfo.PropertyType.IsEnum())
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
insertItem.Add(columnInfo);
|
||||
}
|
||||
}
|
||||
private void SetInsertItemByEntity(int i, T item, List<DbColumnInfo> insertItem)
|
||||
{
|
||||
if (item == null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
foreach (var column in EntityInfo.Columns)
|
||||
{
|
||||
if (column.IsIgnore || column.IsOnlyIgnoreInsert) continue;
|
||||
var isMapping = IsMappingColumns;
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = PropertyCallAdapterProvider<T>.GetInstance(column.PropertyName).InvokeGet(item),
|
||||
DbColumnName = column.DbColumnName,
|
||||
PropertyName = column.PropertyName,
|
||||
PropertyType = UtilMethods.GetUnderType(column.PropertyInfo),
|
||||
TableId = i
|
||||
};
|
||||
if (column.DbColumnName == null)
|
||||
{
|
||||
column.DbColumnName = column.PropertyName;
|
||||
}
|
||||
if (isMapping)
|
||||
{
|
||||
columnInfo.DbColumnName = GetDbColumnName(column.PropertyName);
|
||||
}
|
||||
if (column.IsJson)
|
||||
{
|
||||
columnInfo.IsJson = true;
|
||||
}
|
||||
if (column.IsArray)
|
||||
{
|
||||
columnInfo.IsArray = true;
|
||||
}
|
||||
if (columnInfo.PropertyType.IsEnum()&& columnInfo.Value!=null)
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
if (column.IsJson&& columnInfo.Value!=null)
|
||||
{
|
||||
if(columnInfo.Value!=null)
|
||||
columnInfo.Value = this.Context.Utilities.SerializeObject(columnInfo.Value);
|
||||
}
|
||||
//var tranColumn=EntityInfo.Columns.FirstOrDefault(it => it.IsTranscoding && it.DbColumnName.Equals(column.DbColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (column.IsTranscoding&&columnInfo.Value.HasValue()) {
|
||||
columnInfo.Value = UtilMethods.EncodeBase64(columnInfo.Value.ToString());
|
||||
}
|
||||
insertItem.Add(columnInfo);
|
||||
}
|
||||
}
|
||||
|
||||
private string GetDbColumnName(string propertyName)
|
||||
{
|
||||
if (!IsMappingColumns)
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
if (this.Context.MappingColumns.Any(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)))
|
||||
{
|
||||
this.MappingColumnList = this.Context.MappingColumns.Where(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)).ToList();
|
||||
}
|
||||
if (MappingColumnList == null || !MappingColumnList.Any())
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
else
|
||||
{
|
||||
var mappInfo = this.MappingColumnList.FirstOrDefault(it => it.PropertyName.Equals(propertyName, StringComparison.CurrentCultureIgnoreCase));
|
||||
return mappInfo == null ? propertyName : mappInfo.DbColumnName;
|
||||
}
|
||||
}
|
||||
|
||||
protected virtual List<string> GetPrimaryKeys()
|
||||
{
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetPrimaries(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => it.IsPrimarykey).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
protected virtual List<string> GetIdentityKeys()
|
||||
{
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetIsIdentities(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => {
|
||||
|
||||
Check.Exception(it.IsIdentity&&it.UnderType == typeof(string), "IsIdentity key can not be type of string");
|
||||
return it.IsIdentity;
|
||||
|
||||
}).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
//private void TaskStart<Type>(Task<Type> result)
|
||||
//{
|
||||
// if (this.Context.CurrentConnectionConfig.IsShardSameThread)
|
||||
// {
|
||||
// Check.Exception(true, "IsShardSameThread=true can't be used async method");
|
||||
// }
|
||||
// result.Start();
|
||||
//}
|
||||
protected void RestoreMapping()
|
||||
{
|
||||
if (IsAs)
|
||||
{
|
||||
this.Context.MappingTables = OldMappingTableList;
|
||||
}
|
||||
}
|
||||
//protected IInsertable<T> CopyInsertable()
|
||||
//{
|
||||
// var asyncContext = this.Context.Utilities.CopyContext(true);
|
||||
// asyncContext.CurrentConnectionConfig.IsAutoCloseConnection = true;
|
||||
// asyncContext.IsAsyncMethod = true;
|
||||
// var asyncInsertable = asyncContext.Insertable<T>(this.InsertObjs);
|
||||
// var asyncInsertableBuilder = asyncInsertable.InsertBuilder;
|
||||
// asyncInsertableBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList;
|
||||
// asyncInsertableBuilder.EntityInfo = this.InsertBuilder.EntityInfo;
|
||||
// asyncInsertableBuilder.Parameters = this.InsertBuilder.Parameters;
|
||||
// asyncInsertableBuilder.sql = this.InsertBuilder.sql;
|
||||
// asyncInsertableBuilder.IsNoInsertNull = this.InsertBuilder.IsNoInsertNull;
|
||||
// asyncInsertableBuilder.IsReturnIdentity = this.InsertBuilder.IsReturnIdentity;
|
||||
// asyncInsertableBuilder.EntityInfo = this.InsertBuilder.EntityInfo;
|
||||
// asyncInsertableBuilder.TableWithString = this.InsertBuilder.TableWithString;
|
||||
// if (this.RemoveCacheFunc != null)
|
||||
// {
|
||||
// asyncInsertable.RemoveDataCache();
|
||||
// }
|
||||
// return asyncInsertable;
|
||||
//}
|
||||
|
||||
protected void After(string sql, long? result)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = InsertBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.AfterData = GetDiffTable(sql, result);
|
||||
diffModel.Time = this.Context.Ado.SqlExecutionTime;
|
||||
if (this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent != null)
|
||||
this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent(diffModel);
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
if (this.RemoveCacheFunc != null)
|
||||
{
|
||||
this.RemoveCacheFunc();
|
||||
}
|
||||
}
|
||||
protected void Before(string sql)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = InsertBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.BeforeData = null;
|
||||
diffModel.Sql = sql;
|
||||
diffModel.Parameters = parameters.ToArray();
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
}
|
||||
private List<DiffLogTableInfo> GetDiffTable(string sql, long? identity)
|
||||
{
|
||||
|
||||
if (GetIdentityKeys().HasValue() && this.InsertObjs.Count() > 1)
|
||||
{
|
||||
return GetDiffTableByEntity();
|
||||
}
|
||||
else if (GetIdentityKeys().IsNullOrEmpty())
|
||||
{
|
||||
return GetDiffTableByEntity();
|
||||
}
|
||||
else
|
||||
{
|
||||
return GetDiffTableBySql(identity);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private List<DiffLogTableInfo> GetDiffTableByEntity()
|
||||
{
|
||||
List<SugarParameter> parameters = new List<SugarParameter>();
|
||||
List<DiffLogTableInfo> result = new List<DiffLogTableInfo>();
|
||||
var dt2 = this.Context.Utilities.ListToDataTable<T>(this.InsertObjs.ToList());
|
||||
foreach (DataRow row in dt2.Rows)
|
||||
{
|
||||
DiffLogTableInfo item = new DiffLogTableInfo();
|
||||
item.TableDescription = this.EntityInfo.TableDescription;
|
||||
item.TableName = this.EntityInfo.DbTableName;
|
||||
item.Columns = new List<DiffLogColumnInfo>();
|
||||
foreach (DataColumn col in dt2.Columns)
|
||||
{
|
||||
var sugarColumn = this.EntityInfo.Columns.Where(it => it.DbColumnName != null).FirstOrDefault(it =>
|
||||
it.DbColumnName.Equals(col.ColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
DiffLogColumnInfo addItem = new DiffLogColumnInfo();
|
||||
addItem.Value = row[col.ColumnName];
|
||||
addItem.ColumnName = col.ColumnName;
|
||||
addItem.IsPrimaryKey = sugarColumn?.IsPrimarykey ?? false;
|
||||
addItem.ColumnDescription = sugarColumn?.ColumnDescription;
|
||||
item.Columns.Add(addItem);
|
||||
}
|
||||
result.Add(item);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
private List<DiffLogTableInfo> GetDiffTableBySql(long? identity)
|
||||
{
|
||||
List<SugarParameter> parameters = new List<SugarParameter>();
|
||||
List<DiffLogTableInfo> result = new List<DiffLogTableInfo>();
|
||||
var whereSql = string.Empty;
|
||||
List<IConditionalModel> cons = new List<IConditionalModel>();
|
||||
if (identity != null && identity > 0 && GetIdentityKeys().HasValue())
|
||||
{
|
||||
var fieldName = GetIdentityKeys().Last();
|
||||
|
||||
if (this.Context.CurrentConnectionConfig.DbType == DbType.PostgreSQL)
|
||||
{
|
||||
var fieldObjectType = this.EntityInfo.Columns.FirstOrDefault(x => x.DbColumnName == fieldName)
|
||||
.PropertyInfo.PropertyType;
|
||||
cons.Add(new ConditionalModel() { ConditionalType = ConditionalType.Equal, FieldName = fieldName, FieldValue = identity.ToString(),
|
||||
FieldValueConvertFunc = it => UtilMethods.ChangeType2(it, fieldObjectType) });
|
||||
}
|
||||
else
|
||||
cons.Add(new ConditionalModel() { ConditionalType = ConditionalType.Equal, FieldName = fieldName, FieldValue = identity.ToString() });
|
||||
}
|
||||
else
|
||||
{
|
||||
foreach (var item in this.EntityInfo.Columns.Where(it => it.IsIgnore == false && GetPrimaryKeys().Any(pk => pk.Equals(it.DbColumnName, StringComparison.CurrentCultureIgnoreCase))))
|
||||
{
|
||||
var fielddName = item.DbColumnName;
|
||||
var filedObject = this.EntityInfo.Columns.FirstOrDefault(it => it.PropertyName == item.PropertyName).PropertyInfo.GetValue(this.InsertObjs.Last(), null);
|
||||
var fieldValue = filedObject.ObjToString();
|
||||
if (filedObject != null && filedObject.GetType() != typeof(string) && this.Context.CurrentConnectionConfig.DbType == DbType.PostgreSQL)
|
||||
{
|
||||
cons.Add(new ConditionalModel() { ConditionalType = ConditionalType.Equal, FieldName = fielddName, FieldValue = fieldValue, FieldValueConvertFunc = it => UtilMethods.ChangeType2(it, filedObject.GetType()) });
|
||||
}
|
||||
else
|
||||
{
|
||||
cons.Add(new ConditionalModel() { ConditionalType = ConditionalType.Equal, FieldName = fielddName, FieldValue = fieldValue });
|
||||
}
|
||||
}
|
||||
}
|
||||
Check.Exception(cons.IsNullOrEmpty(), "Insertable.EnableDiffLogEvent need primary key");
|
||||
var sqlable = this.SqlBuilder.ConditionalModelToSql(cons);
|
||||
whereSql = sqlable.Key;
|
||||
parameters.AddRange(sqlable.Value);
|
||||
var dt = this.Context.Queryable<T>().Where(whereSql).AddParameters(parameters).ToDataTable();
|
||||
if (dt.Rows != null && dt.Rows.Count > 0)
|
||||
{
|
||||
foreach (DataRow row in dt.Rows)
|
||||
{
|
||||
DiffLogTableInfo item = new DiffLogTableInfo();
|
||||
item.TableDescription = this.EntityInfo.TableDescription;
|
||||
item.TableName = this.EntityInfo.DbTableName;
|
||||
item.Columns = new List<DiffLogColumnInfo>();
|
||||
foreach (DataColumn col in dt.Columns)
|
||||
{
|
||||
var sugarColumn = this.EntityInfo.Columns.Where(it => it.DbColumnName != null).First(it =>
|
||||
it.DbColumnName.Equals(col.ColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
DiffLogColumnInfo addItem = new DiffLogColumnInfo();
|
||||
addItem.Value = row[col.ColumnName];
|
||||
addItem.ColumnName = col.ColumnName;
|
||||
addItem.IsPrimaryKey = sugarColumn.IsPrimarykey;
|
||||
addItem.ColumnDescription = sugarColumn.ColumnDescription;
|
||||
item.Columns.Add(addItem);
|
||||
}
|
||||
result.Add(item);
|
||||
}
|
||||
return result;
|
||||
}
|
||||
else
|
||||
{
|
||||
DiffLogTableInfo diffTable = new DiffLogTableInfo();
|
||||
diffTable.TableName = this.EntityInfo.DbTableName;
|
||||
diffTable.TableDescription = this.EntityInfo.TableDescription;
|
||||
diffTable.Columns = this.EntityInfo.Columns.Where(it => it.IsIgnore == false).Select(it => new DiffLogColumnInfo()
|
||||
{
|
||||
ColumnDescription = it.ColumnDescription,
|
||||
ColumnName = it.DbColumnName,
|
||||
Value = it.PropertyInfo.GetValue(this.InsertObjs.Last(), null),
|
||||
IsPrimaryKey = it.IsPrimarykey
|
||||
}).ToList();
|
||||
return new List<DiffLogTableInfo>() { diffTable };
|
||||
}
|
||||
}
|
||||
|
||||
public IInsertable<T> CallEntityMethod(Expression<Action<T>> method)
|
||||
{
|
||||
if (this.InsertObjs.HasValue())
|
||||
{
|
||||
var oldColumns = this.InsertBuilder.DbColumnInfoList.Select(it => it.PropertyName).ToList();
|
||||
var expression = (LambdaExpression.Lambda(method).Body as LambdaExpression).Body;
|
||||
Check.Exception(!(expression is MethodCallExpression), method.ToString() + " is not method");
|
||||
var callExpresion = expression as MethodCallExpression;
|
||||
UtilMethods.DataInoveByExpresson(this.InsertObjs,callExpresion);
|
||||
this.InsertBuilder.DbColumnInfoList = new List<DbColumnInfo>();
|
||||
Init();
|
||||
this.InsertBuilder.DbColumnInfoList = this.InsertBuilder.DbColumnInfoList.Where(it => oldColumns.Contains(it.PropertyName)).ToList();
|
||||
}
|
||||
return this;
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,505 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Data;
|
||||
using System.Linq;
|
||||
using System.Linq.Expressions;
|
||||
using System.Text;
|
||||
using System.Text.RegularExpressions;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace SqlSugar
|
||||
{
|
||||
public partial class UpdateableProvider<T> : IUpdateable<T> where T : class, new()
|
||||
{
|
||||
|
||||
private bool UpdateObjectNotWhere()
|
||||
{
|
||||
return this.Context.CurrentConnectionConfig.DbType != DbType.MySql
|
||||
&& this.Context.CurrentConnectionConfig.DbType != DbType.MySqlConnector
|
||||
&& this.Context.CurrentConnectionConfig.DbType != DbType.SqlServer;
|
||||
}
|
||||
|
||||
private void AppendSets()
|
||||
{
|
||||
if (SetColumnsIndex > 0)
|
||||
{
|
||||
var keys = UpdateBuilder.SetValues.Select(it => SqlBuilder.GetNoTranslationColumnName(it.Key.ToLower())).ToList();
|
||||
var addKeys = keys.Where(k => !this.UpdateBuilder.DbColumnInfoList.Any(it => it.PropertyName.ToLower() == k || it.DbColumnName.ToLower() == k)).ToList();
|
||||
var addItems = this.EntityInfo.Columns.Where(it => !GetPrimaryKeys().Any(p => p.ToLower() == it.PropertyName?.ToLower() || p.ToLower() == it.DbColumnName?.ToLower()) && addKeys.Any(k => it.PropertyName?.ToLower() == k || it.DbColumnName?.ToLower() == k)).ToList();
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(addItems.Select(it => new DbColumnInfo() { PropertyName = it.PropertyName, DbColumnName = it.DbColumnName }));
|
||||
}
|
||||
SetColumnsIndex++;
|
||||
}
|
||||
private string _ExecuteCommand()
|
||||
{
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
Check.Exception(UpdateBuilder.WhereValues.IsNullOrEmpty() && GetPrimaryKeys().IsNullOrEmpty(), "You cannot have no primary key and no conditions");
|
||||
string sql = UpdateBuilder.ToSqlString();
|
||||
ValidateVersion();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
|
||||
private void _WhereColumn(string columnName)
|
||||
{
|
||||
var columnInfos = columns.Where(it => it.DbColumnName.Equals(columnName, StringComparison.OrdinalIgnoreCase) || it.PropertyName.Equals(columnName, StringComparison.OrdinalIgnoreCase)).ToList();
|
||||
if (!this.UpdateBuilder.DbColumnInfoList.Any(y => y.DbColumnName == columnInfos.First().DbColumnName))
|
||||
{
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(columnInfos);
|
||||
}
|
||||
}
|
||||
|
||||
private void AutoRemoveDataCache()
|
||||
{
|
||||
var moreSetts = this.Context.CurrentConnectionConfig.MoreSettings;
|
||||
var extService = this.Context.CurrentConnectionConfig.ConfigureExternalServices;
|
||||
if (moreSetts != null && moreSetts.IsAutoRemoveDataCache && extService != null && extService.DataInfoCacheService != null)
|
||||
{
|
||||
this.RemoveDataCache();
|
||||
}
|
||||
}
|
||||
internal void Init()
|
||||
{
|
||||
this.UpdateBuilder.TableName = EntityInfo.EntityName;
|
||||
if (IsMappingTable)
|
||||
{
|
||||
var mappingInfo = this.Context.MappingTables.SingleOrDefault(it => it.EntityName == EntityInfo.EntityName);
|
||||
if (mappingInfo != null)
|
||||
{
|
||||
this.UpdateBuilder.TableName = mappingInfo.DbTableName;
|
||||
}
|
||||
}
|
||||
//Check.Exception(UpdateObjs == null || UpdateObjs.Count() == 0, "UpdateObjs is null");
|
||||
int i = 0;
|
||||
foreach (var item in UpdateObjs)
|
||||
{
|
||||
List<DbColumnInfo> updateItem = new List<DbColumnInfo>();
|
||||
var isDic = item is Dictionary<string, object>;
|
||||
if (isDic)
|
||||
{
|
||||
SetUpdateItemByDic(i, item, updateItem);
|
||||
}
|
||||
else
|
||||
{
|
||||
DataAop(item);
|
||||
SetUpdateItemByEntity(i, item, updateItem);
|
||||
}
|
||||
++i;
|
||||
}
|
||||
this.columns = this.UpdateBuilder.DbColumnInfoList;
|
||||
|
||||
var ignoreColumns = EntityInfo.Columns.Where(it => it.IsOnlyIgnoreUpdate).ToList();
|
||||
if (ignoreColumns != null && ignoreColumns.Any())
|
||||
{
|
||||
this.IgnoreColumns(ignoreColumns.Select(it => it.PropertyName).ToArray());
|
||||
}
|
||||
}
|
||||
|
||||
private void DataAop(T item)
|
||||
{
|
||||
var dataEvent = this.Context.CurrentConnectionConfig.AopEvents?.DataExecuting;
|
||||
if (dataEvent != null && item != null)
|
||||
{
|
||||
foreach (var columnInfo in this.EntityInfo.Columns)
|
||||
{
|
||||
dataEvent(columnInfo.PropertyInfo.GetValue(item, null), new DataFilterModel() { OperationType = DataFilterType.UpdateByObject, EntityValue = item, EntityColumnInfo = columnInfo });
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void CheckTranscodeing(bool checkIsJson = true)
|
||||
{
|
||||
if (this.EntityInfo.Columns.Any(it => it.IsTranscoding))
|
||||
{
|
||||
Check.Exception(true, ErrorMessage.GetThrowMessage("UpdateColumns no support IsTranscoding", "SetColumns方式更新不支持IsTranscoding，你可以使用db.Updateable(实体)的方式更新"));
|
||||
}
|
||||
if (checkIsJson && this.EntityInfo.Columns.Any(it => it.IsJson))
|
||||
{
|
||||
Check.Exception(true, ErrorMessage.GetThrowMessage("UpdateColumns no support IsJson", "SetColumns方式更新不支持IsJson，你可以使用db.Updateable(实体)的方式更新"));
|
||||
}
|
||||
if (this.EntityInfo.Columns.Any(it => it.IsArray))
|
||||
{
|
||||
Check.Exception(true, ErrorMessage.GetThrowMessage("UpdateColumns no support IsArray", "SetColumns方式更新不支持IsArray，你可以使用db.Updateable(实体)的方式更新"));
|
||||
}
|
||||
}
|
||||
private void SetUpdateItemByDic(int i, T item, List<DbColumnInfo> updateItem)
|
||||
{
|
||||
foreach (var column in item as Dictionary<string, object>)
|
||||
{
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = column.Value,
|
||||
DbColumnName = column.Key,
|
||||
PropertyName = column.Key,
|
||||
PropertyType = column.Value == null ? DBNull.Value.GetType() : UtilMethods.GetUnderType(column.Value.GetType()),
|
||||
TableId = i
|
||||
};
|
||||
if (columnInfo.PropertyType.IsEnum())
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
updateItem.Add(columnInfo);
|
||||
}
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(updateItem);
|
||||
}
|
||||
private void SetUpdateItemByEntity(int i, T item, List<DbColumnInfo> updateItem)
|
||||
{
|
||||
foreach (var column in EntityInfo.Columns)
|
||||
{
|
||||
if (column.IsIgnore) continue;
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = column.PropertyInfo.GetValue(item, null),
|
||||
DbColumnName = GetDbColumnName(column.PropertyName),
|
||||
PropertyName = column.PropertyName,
|
||||
PropertyType = UtilMethods.GetUnderType(column.PropertyInfo),
|
||||
TableId = i
|
||||
};
|
||||
if (columnInfo.PropertyType.IsEnum() && columnInfo.Value != null)
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
if (column.IsJson)
|
||||
{
|
||||
columnInfo.IsJson = true;
|
||||
if (columnInfo.Value != null)
|
||||
columnInfo.Value = this.Context.Utilities.SerializeObject(columnInfo.Value);
|
||||
}
|
||||
if (column.IsArray)
|
||||
{
|
||||
columnInfo.IsArray = true;
|
||||
}
|
||||
var tranColumn = EntityInfo.Columns.FirstOrDefault(it => it.IsTranscoding && it.DbColumnName.Equals(column.DbColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (tranColumn != null && columnInfo.Value.HasValue())
|
||||
{
|
||||
columnInfo.Value = UtilMethods.EncodeBase64(columnInfo.Value.ToString());
|
||||
}
|
||||
updateItem.Add(columnInfo);
|
||||
}
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(updateItem);
|
||||
}
|
||||
|
||||
private void PreToSql()
|
||||
{
|
||||
|
||||
UpdateBuilder.PrimaryKeys = GetPrimaryKeys();
|
||||
if (this.IsWhereColumns)
|
||||
{
|
||||
foreach (var pkName in UpdateBuilder.PrimaryKeys)
|
||||
{
|
||||
if (WhereColumnList != null && WhereColumnList.Count() > 0)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
var isContains = this.UpdateBuilder.DbColumnInfoList.Select(it => it.DbColumnName.ToLower()).Contains(pkName.ToLower());
|
||||
Check.Exception(isContains == false, "Use UpdateColumns().WhereColumn() ,UpdateColumns need {0}", pkName);
|
||||
}
|
||||
}
|
||||
#region IgnoreColumns
|
||||
if (this.Context.IgnoreColumns != null && this.Context.IgnoreColumns.Any())
|
||||
{
|
||||
var currentIgnoreColumns = this.Context.IgnoreColumns.Where(it => it.EntityName == this.EntityInfo.EntityName).ToList();
|
||||
this.UpdateBuilder.DbColumnInfoList = this.UpdateBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !currentIgnoreColumns.Any(i => it.PropertyName.Equals(i.PropertyName, StringComparison.CurrentCulture));
|
||||
}).ToList();
|
||||
}
|
||||
#endregion
|
||||
if (this.IsSingle)
|
||||
{
|
||||
foreach (var item in this.UpdateBuilder.DbColumnInfoList)
|
||||
{
|
||||
if (this.UpdateBuilder.Parameters == null) this.UpdateBuilder.Parameters = new List<SugarParameter>();
|
||||
if (this.UpdateBuilder.SetValues.Any(it => this.SqlBuilder.GetNoTranslationColumnName(it.Key) == item.PropertyName))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
var parameter = new SugarParameter(this.SqlBuilder.SqlParameterKeyWord + item.DbColumnName, item.Value, item.PropertyType);
|
||||
if (item.IsJson)
|
||||
{
|
||||
parameter.IsJson = true;
|
||||
}
|
||||
if (item.IsArray)
|
||||
{
|
||||
parameter.IsArray = true;
|
||||
}
|
||||
this.UpdateBuilder.Parameters.Add(parameter);
|
||||
}
|
||||
}
|
||||
|
||||
#region Identities
|
||||
List<string> identities = GetIdentityKeys();
|
||||
if (identities != null && identities.Any())
|
||||
{
|
||||
this.UpdateBuilder.DbColumnInfoList.ForEach(it =>
|
||||
{
|
||||
var mappingInfo = identities.SingleOrDefault(i => it.DbColumnName.Equals(i, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (mappingInfo != null && mappingInfo.Any())
|
||||
{
|
||||
it.IsIdentity = true;
|
||||
}
|
||||
});
|
||||
}
|
||||
#endregion
|
||||
List<string> primaryKey = GetPrimaryKeys();
|
||||
if (primaryKey != null && primaryKey.Count > 0)
|
||||
{
|
||||
this.UpdateBuilder.DbColumnInfoList.ForEach(it =>
|
||||
{
|
||||
var mappingInfo = primaryKey.SingleOrDefault(i => it.DbColumnName.Equals(i, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (mappingInfo != null && mappingInfo.Any())
|
||||
{
|
||||
it.IsPrimarykey = true;
|
||||
}
|
||||
});
|
||||
}
|
||||
if (this.UpdateBuilder.Parameters.HasValue() && this.UpdateBuilder.SetValues.IsValuable())
|
||||
{
|
||||
this.UpdateBuilder.Parameters.RemoveAll(it => this.UpdateBuilder.SetValues.Any(v => (SqlBuilder.SqlParameterKeyWord + SqlBuilder.GetNoTranslationColumnName(v.Key)) == it.ParameterName));
|
||||
}
|
||||
}
|
||||
private string GetDbColumnName(string propertyName)
|
||||
{
|
||||
if (!IsMappingColumns)
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
if (this.Context.MappingColumns.Any(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)))
|
||||
{
|
||||
this.MappingColumnList = this.Context.MappingColumns.Where(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)).ToList();
|
||||
}
|
||||
if (MappingColumnList == null || !MappingColumnList.Any())
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
else
|
||||
{
|
||||
var mappInfo = this.MappingColumnList.FirstOrDefault(it => it.PropertyName.Equals(propertyName, StringComparison.CurrentCultureIgnoreCase));
|
||||
return mappInfo == null ? propertyName : mappInfo.DbColumnName;
|
||||
}
|
||||
}
|
||||
private List<string> GetPrimaryKeys()
|
||||
{
|
||||
if (this.WhereColumnList.HasValue())
|
||||
{
|
||||
return this.WhereColumnList;
|
||||
}
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetPrimaries(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => it.IsPrimarykey).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
protected virtual List<string> GetIdentityKeys()
|
||||
{
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetIsIdentities(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => it.IsIdentity).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
private void RestoreMapping()
|
||||
{
|
||||
if (IsAs)
|
||||
{
|
||||
this.Context.MappingTables = OldMappingTableList;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private void ValidateVersion()
|
||||
{
|
||||
var versionColumn = this.EntityInfo.Columns.FirstOrDefault(it => it.IsEnableUpdateVersionValidation);
|
||||
var pks = this.UpdateBuilder.DbColumnInfoList.Where(it => it.IsPrimarykey).ToList();
|
||||
if (versionColumn != null && this.IsVersionValidation)
|
||||
{
|
||||
Check.Exception(pks.IsNullOrEmpty(), "UpdateVersionValidation the primary key is required.");
|
||||
List<IConditionalModel> conModels = new List<IConditionalModel>();
|
||||
foreach (var item in pks)
|
||||
{
|
||||
conModels.Add(new ConditionalModel() { FieldName = item.DbColumnName, ConditionalType = ConditionalType.Equal, FieldValue = item.Value.ObjToString() });
|
||||
}
|
||||
var dbInfo = this.Context.Queryable<T>().Where(conModels).First();
|
||||
if (dbInfo != null)
|
||||
{
|
||||
var currentVersion = this.EntityInfo.Type.GetProperty(versionColumn.PropertyName).GetValue(UpdateObjs.Last(), null);
|
||||
var dbVersion = this.EntityInfo.Type.GetProperty(versionColumn.PropertyName).GetValue(dbInfo, null);
|
||||
Check.Exception(currentVersion == null, "UpdateVersionValidation entity property {0} is not null", versionColumn.PropertyName);
|
||||
Check.Exception(dbVersion == null, "UpdateVersionValidation database column {0} is not null", versionColumn.DbColumnName);
|
||||
if (versionColumn.PropertyInfo.PropertyType.IsIn(UtilConstants.IntType, UtilConstants.LongType))
|
||||
{
|
||||
if (Convert.ToInt64(dbVersion) != Convert.ToInt64(currentVersion))
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
else if (versionColumn.PropertyInfo.PropertyType.IsIn(UtilConstants.DateType))
|
||||
{
|
||||
if (dbVersion.ObjToDate() != currentVersion.ObjToDate())
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
else if (versionColumn.PropertyInfo.PropertyType.IsIn(UtilConstants.ByteArrayType))
|
||||
{
|
||||
if (UtilMethods.GetLong((byte[])dbVersion) != UtilMethods.GetLong((byte[])currentVersion))
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Check.ThrowNotSupportedException(string.Format("UpdateVersionValidation Not Supported Type [ {0} ] , {1}", versionColumn.PropertyInfo.PropertyType, versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
private void After(string sql)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = UpdateBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.AfterData = GetDiffTable(sql, parameters);
|
||||
diffModel.Time = this.Context.Ado.SqlExecutionTime;
|
||||
if (this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent != null)
|
||||
this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent(diffModel);
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
if (this.RemoveCacheFunc != null)
|
||||
{
|
||||
this.RemoveCacheFunc();
|
||||
}
|
||||
}
|
||||
private string _ExecuteCommandWithOptLock(T updateData)
|
||||
{
|
||||
Check.ExceptionEasy(UpdateParameterIsNull == true, "Optimistic lock can only be an entity update method", "乐观锁只能是实体更新方式");
|
||||
var verColumn = this.EntityInfo.Columns.FirstOrDefault(it => it.IsEnableUpdateVersionValidation);
|
||||
Check.ExceptionEasy(verColumn == null, $" {this.EntityInfo.EntityName } need IsEnableUpdateVersionValidation=true ", $"实体{this.EntityInfo.EntityName}没有找到版本标识特性 IsEnableUpdateVersionValidation");
|
||||
Check.ExceptionEasy(UpdateObjs.Length > 1, $"Optimistic lock can only handle a single update ", $"乐观锁只能处理单条更新");
|
||||
Check.ExceptionEasy(!verColumn.UnderType.IsIn(UtilConstants.StringType, UtilConstants.LongType, UtilConstants.GuidType, UtilConstants.DateType), $"Optimistic locks can only be guid, long, and string types", $"乐观锁只能是Guid、Long和字符串类型");
|
||||
var oldValue = verColumn.PropertyInfo.GetValue(updateData);
|
||||
var newValue = UtilMethods.GetRandomByType(verColumn.UnderType);
|
||||
verColumn.PropertyInfo.SetValue(updateData, newValue);
|
||||
var data = this.UpdateBuilder.DbColumnInfoList.First(it =>
|
||||
it.PropertyName.EqualCase(verColumn.PropertyName));
|
||||
data.Value = newValue;
|
||||
var pks = GetPrimaryKeys();
|
||||
Check.ExceptionEasy(pks.Count == 0, "need primary key or WhereColumn", "需要主键或者WhereColumn");
|
||||
this.Where(verColumn.DbColumnName, "=", oldValue);
|
||||
foreach (var p in pks)
|
||||
{
|
||||
var pkColumn = this.EntityInfo.Columns.FirstOrDefault(
|
||||
it => it.DbColumnName.EqualCase(p) || it.PropertyName.EqualCase(p));
|
||||
this.Where(pkColumn.DbColumnName, "=", pkColumn.PropertyInfo.GetValue(updateData));
|
||||
}
|
||||
return verColumn.PropertyName;
|
||||
}
|
||||
private void Before(string sql)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = UpdateBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.BeforeData = GetDiffTable(sql, parameters);
|
||||
diffModel.Sql = sql;
|
||||
diffModel.Parameters = parameters.ToArray();
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
}
|
||||
private bool IsPrimaryKey(DbColumnInfo it)
|
||||
{
|
||||
var result = GetPrimaryKeys().Any(p => p.Equals(it.DbColumnName, StringComparison.CurrentCultureIgnoreCase) || p.Equals(it.PropertyName, StringComparison.CurrentCultureIgnoreCase));
|
||||
return result;
|
||||
}
|
||||
|
||||
private List<DiffLogTableInfo> GetDiffTable(string sql, List<SugarParameter> parameters)
|
||||
{
|
||||
List<DiffLogTableInfo> result = new List<DiffLogTableInfo>();
|
||||
DataTable dt = null;
|
||||
if (this.UpdateParameterIsNull)
|
||||
{
|
||||
var whereSql = Regex.Replace(sql, ".* WHERE ", "", RegexOptions.Singleline);
|
||||
if (sql.Contains("WHERE (EXISTS"))
|
||||
{
|
||||
whereSql = Regex.Match(sql, @"\(EXISTS.+").Value;
|
||||
}
|
||||
dt = this.Context.Queryable<T>().Filter(null, true).Where(whereSql).AddParameters(parameters).ToDataTable();
|
||||
}
|
||||
else
|
||||
{
|
||||
if (this.UpdateObjs.ToList().Count == 0)
|
||||
{
|
||||
dt = new DataTable();
|
||||
}
|
||||
else
|
||||
{
|
||||
dt = this.Context.Queryable<T>().Filter(null, true).WhereClassByPrimaryKey(this.UpdateObjs.ToList()).ToDataTable();
|
||||
}
|
||||
}
|
||||
if (dt.Rows != null && dt.Rows.Count > 0)
|
||||
{
|
||||
foreach (DataRow row in dt.Rows)
|
||||
{
|
||||
DiffLogTableInfo item = new DiffLogTableInfo();
|
||||
item.TableDescription = this.EntityInfo.TableDescription;
|
||||
item.TableName = this.EntityInfo.DbTableName;
|
||||
item.Columns = new List<DiffLogColumnInfo>();
|
||||
foreach (DataColumn col in dt.Columns)
|
||||
{
|
||||
var sugarColumn = this.EntityInfo.Columns.Where(it => it.DbColumnName != null).First(it =>
|
||||
it.DbColumnName.Equals(col.ColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
DiffLogColumnInfo addItem = new DiffLogColumnInfo();
|
||||
addItem.Value = row[col.ColumnName];
|
||||
addItem.ColumnName = col.ColumnName;
|
||||
addItem.IsPrimaryKey = sugarColumn.IsPrimarykey;
|
||||
addItem.ColumnDescription = sugarColumn.ColumnDescription;
|
||||
item.Columns.Add(addItem);
|
||||
}
|
||||
result.Add(item);
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
private void ThrowUpdateByExpression()
|
||||
{
|
||||
Check.Exception(UpdateParameterIsNull == true, ErrorMessage.GetThrowMessage(" no support UpdateColumns and WhereColumns", "根据表达式更新 db.Updateable<T>() 禁止使用 UpdateColumns和WhereColumns,你可以使用 SetColumns Where 等。更新分为2种方式 1.根据表达式更新 2.根据实体或者集合更新， 具体用法请查看文档 "));
|
||||
}
|
||||
private void ThrowUpdateByObject()
|
||||
{
|
||||
Check.Exception(UpdateParameterIsNull == false, ErrorMessage.GetThrowMessage(" no support SetColumns and Where", "根据对像更新 db.Updateabe(对象) 禁止使用 SetColumns和Where ,你可以使用WhereColumns 和 UpdateColumns。 更新分为2种方式 1.根据表达式更新 2.根据实体或者集合更新 ， 具体用法请查看文档 "));
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
@ -9,7 +9,7 @@ using System.Threading.Tasks;
|
||||
|
||||
namespace SqlSugar
|
||||
{
|
||||
public class UpdateableProvider<T> : IUpdateable<T> where T : class, new()
|
||||
public partial class UpdateableProvider<T> : IUpdateable<T> where T : class, new()
|
||||
{
|
||||
#region Property
|
||||
public SqlSugarProvider Context { get; internal set; }
|
||||
@ -59,6 +59,20 @@ namespace SqlSugar
|
||||
var sqlObj = this.ToSql();
|
||||
this.Context.Queues.Add(sqlObj.Key, sqlObj.Value);
|
||||
}
|
||||
|
||||
public virtual int ExecuteCommandWithOptLock(bool IsVersionValidation=false)
|
||||
{
|
||||
var updateData = UpdateObjs.FirstOrDefault();
|
||||
if (updateData == null) return 0;
|
||||
var name=_ExecuteCommandWithOptLock(updateData);
|
||||
var result= this.ExecuteCommand();
|
||||
if (result == 0 && IsVersionValidation)
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", name));
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
public virtual int ExecuteCommand()
|
||||
{
|
||||
string sql = _ExecuteCommand();
|
||||
@ -70,10 +84,25 @@ namespace SqlSugar
|
||||
After(sql);
|
||||
return result;
|
||||
}
|
||||
|
||||
public bool ExecuteCommandHasChange()
|
||||
{
|
||||
return this.ExecuteCommand() > 0;
|
||||
}
|
||||
|
||||
public virtual async Task<int> ExecuteCommandWithOptLockAsync(bool IsVersionValidation = false)
|
||||
{
|
||||
var updateData = UpdateObjs.FirstOrDefault();
|
||||
if (updateData == null) return 0;
|
||||
var name=_ExecuteCommandWithOptLock(updateData);
|
||||
var result= await this.ExecuteCommandAsync();
|
||||
if (result == 0 && IsVersionValidation)
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", name));
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
public virtual async Task<int> ExecuteCommandAsync()
|
||||
{
|
||||
string sql = _ExecuteCommand();
|
||||
@ -85,12 +114,14 @@ namespace SqlSugar
|
||||
After(sql);
|
||||
return result;
|
||||
}
|
||||
|
||||
public async Task<bool> ExecuteCommandHasChangeAsync()
|
||||
{
|
||||
return await this.ExecuteCommandAsync() > 0;
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region Common
|
||||
public IUpdateable<T> With(string lockString)
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.DbType == DbType.SqlServer)
|
||||
@ -108,7 +139,7 @@ namespace SqlSugar
|
||||
};
|
||||
var tables = getTableNamesFunc(helper.GetTables());
|
||||
result.Tables = tables;
|
||||
result.updateobj= this;
|
||||
result.updateobj = this;
|
||||
return result;
|
||||
}
|
||||
public SplitTableUpdateByObjectProvider<T> SplitTable()
|
||||
@ -138,7 +169,7 @@ namespace SqlSugar
|
||||
this.RemoveCacheFunc = () =>
|
||||
{
|
||||
var cacheService = this.Context.CurrentConnectionConfig.ConfigureExternalServices.DataInfoCacheService;
|
||||
CacheSchemeMain.RemoveCacheByLike(cacheService,likeString);
|
||||
CacheSchemeMain.RemoveCacheByLike(cacheService, likeString);
|
||||
};
|
||||
return this;
|
||||
}
|
||||
@ -207,7 +238,7 @@ namespace SqlSugar
|
||||
}
|
||||
|
||||
|
||||
public IUpdateable<T> ReSetValue(Action<T> setValueExpression)
|
||||
public IUpdateable<T> ReSetValue(Action<T> setValueExpression)
|
||||
{
|
||||
ThrowUpdateByExpression();
|
||||
if (this.UpdateObjs.HasValue())
|
||||
@ -222,7 +253,8 @@ namespace SqlSugar
|
||||
this.UpdateBuilder.DbColumnInfoList = this.UpdateBuilder.DbColumnInfoList.Where(it => oldColumns.Contains(it.PropertyName)).ToList();
|
||||
}
|
||||
return this;
|
||||
}
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region Update by object
|
||||
public IUpdateable<T> CallEntityMethod(Expression<Action<T>> method)
|
||||
@ -464,473 +496,5 @@ namespace SqlSugar
|
||||
}
|
||||
#endregion
|
||||
|
||||
#region Helper
|
||||
|
||||
private bool UpdateObjectNotWhere()
|
||||
{
|
||||
return this.Context.CurrentConnectionConfig.DbType != DbType.MySql
|
||||
&& this.Context.CurrentConnectionConfig.DbType!=DbType.MySqlConnector
|
||||
&& this.Context.CurrentConnectionConfig.DbType != DbType.SqlServer;
|
||||
}
|
||||
|
||||
private void AppendSets()
|
||||
{
|
||||
if (SetColumnsIndex > 0)
|
||||
{
|
||||
var keys = UpdateBuilder.SetValues.Select(it => SqlBuilder.GetNoTranslationColumnName(it.Key.ToLower())).ToList();
|
||||
var addKeys = keys.Where(k => !this.UpdateBuilder.DbColumnInfoList.Any(it => it.PropertyName.ToLower() == k || it.DbColumnName.ToLower() == k)).ToList();
|
||||
var addItems = this.EntityInfo.Columns.Where(it => !GetPrimaryKeys().Any(p => p.ToLower() == it.PropertyName?.ToLower() || p.ToLower() == it.DbColumnName?.ToLower()) && addKeys.Any(k => it.PropertyName?.ToLower() == k || it.DbColumnName?.ToLower() == k)).ToList();
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(addItems.Select(it => new DbColumnInfo() { PropertyName = it.PropertyName, DbColumnName = it.DbColumnName }));
|
||||
}
|
||||
SetColumnsIndex++;
|
||||
}
|
||||
private string _ExecuteCommand()
|
||||
{
|
||||
PreToSql();
|
||||
AutoRemoveDataCache();
|
||||
Check.Exception(UpdateBuilder.WhereValues.IsNullOrEmpty() && GetPrimaryKeys().IsNullOrEmpty(), "You cannot have no primary key and no conditions");
|
||||
string sql = UpdateBuilder.ToSqlString();
|
||||
ValidateVersion();
|
||||
RestoreMapping();
|
||||
Before(sql);
|
||||
return sql;
|
||||
}
|
||||
|
||||
private void _WhereColumn(string columnName)
|
||||
{
|
||||
var columnInfos = columns.Where(it => it.DbColumnName.Equals(columnName, StringComparison.OrdinalIgnoreCase) || it.PropertyName.Equals(columnName, StringComparison.OrdinalIgnoreCase)).ToList();
|
||||
if (!this.UpdateBuilder.DbColumnInfoList.Any(y => y.DbColumnName == columnInfos.First().DbColumnName))
|
||||
{
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(columnInfos);
|
||||
}
|
||||
}
|
||||
|
||||
private void AutoRemoveDataCache()
|
||||
{
|
||||
var moreSetts = this.Context.CurrentConnectionConfig.MoreSettings;
|
||||
var extService = this.Context.CurrentConnectionConfig.ConfigureExternalServices;
|
||||
if (moreSetts != null && moreSetts.IsAutoRemoveDataCache && extService != null && extService.DataInfoCacheService != null)
|
||||
{
|
||||
this.RemoveDataCache();
|
||||
}
|
||||
}
|
||||
internal void Init()
|
||||
{
|
||||
this.UpdateBuilder.TableName = EntityInfo.EntityName;
|
||||
if (IsMappingTable)
|
||||
{
|
||||
var mappingInfo = this.Context.MappingTables.SingleOrDefault(it => it.EntityName == EntityInfo.EntityName);
|
||||
if (mappingInfo != null)
|
||||
{
|
||||
this.UpdateBuilder.TableName = mappingInfo.DbTableName;
|
||||
}
|
||||
}
|
||||
//Check.Exception(UpdateObjs == null || UpdateObjs.Count() == 0, "UpdateObjs is null");
|
||||
int i = 0;
|
||||
foreach (var item in UpdateObjs)
|
||||
{
|
||||
List<DbColumnInfo> updateItem = new List<DbColumnInfo>();
|
||||
var isDic = item is Dictionary<string, object>;
|
||||
if (isDic)
|
||||
{
|
||||
SetUpdateItemByDic(i, item, updateItem);
|
||||
}
|
||||
else
|
||||
{
|
||||
DataAop(item);
|
||||
SetUpdateItemByEntity(i, item, updateItem);
|
||||
}
|
||||
++i;
|
||||
}
|
||||
this.columns = this.UpdateBuilder.DbColumnInfoList;
|
||||
|
||||
var ignoreColumns = EntityInfo.Columns.Where(it => it.IsOnlyIgnoreUpdate).ToList();
|
||||
if (ignoreColumns != null && ignoreColumns.Any())
|
||||
{
|
||||
this.IgnoreColumns(ignoreColumns.Select(it => it.PropertyName).ToArray());
|
||||
}
|
||||
}
|
||||
|
||||
private void DataAop(T item)
|
||||
{
|
||||
var dataEvent = this.Context.CurrentConnectionConfig.AopEvents?.DataExecuting;
|
||||
if (dataEvent != null && item != null)
|
||||
{
|
||||
foreach (var columnInfo in this.EntityInfo.Columns)
|
||||
{
|
||||
dataEvent(columnInfo.PropertyInfo.GetValue(item, null), new DataFilterModel() { OperationType = DataFilterType.UpdateByObject, EntityValue = item, EntityColumnInfo = columnInfo });
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void CheckTranscodeing(bool checkIsJson = true)
|
||||
{
|
||||
if (this.EntityInfo.Columns.Any(it => it.IsTranscoding))
|
||||
{
|
||||
Check.Exception(true, ErrorMessage.GetThrowMessage("UpdateColumns no support IsTranscoding", "SetColumns方式更新不支持IsTranscoding，你可以使用db.Updateable(实体)的方式更新"));
|
||||
}
|
||||
if (checkIsJson && this.EntityInfo.Columns.Any(it => it.IsJson))
|
||||
{
|
||||
Check.Exception(true, ErrorMessage.GetThrowMessage("UpdateColumns no support IsJson", "SetColumns方式更新不支持IsJson，你可以使用db.Updateable(实体)的方式更新"));
|
||||
}
|
||||
if (this.EntityInfo.Columns.Any(it => it.IsArray))
|
||||
{
|
||||
Check.Exception(true, ErrorMessage.GetThrowMessage("UpdateColumns no support IsArray", "SetColumns方式更新不支持IsArray，你可以使用db.Updateable(实体)的方式更新"));
|
||||
}
|
||||
}
|
||||
private void SetUpdateItemByDic(int i, T item, List<DbColumnInfo> updateItem)
|
||||
{
|
||||
foreach (var column in item as Dictionary<string, object>)
|
||||
{
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = column.Value,
|
||||
DbColumnName = column.Key,
|
||||
PropertyName = column.Key,
|
||||
PropertyType = column.Value == null ? DBNull.Value.GetType() : UtilMethods.GetUnderType(column.Value.GetType()),
|
||||
TableId = i
|
||||
};
|
||||
if (columnInfo.PropertyType.IsEnum())
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
updateItem.Add(columnInfo);
|
||||
}
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(updateItem);
|
||||
}
|
||||
private void SetUpdateItemByEntity(int i, T item, List<DbColumnInfo> updateItem)
|
||||
{
|
||||
foreach (var column in EntityInfo.Columns)
|
||||
{
|
||||
if (column.IsIgnore) continue;
|
||||
var columnInfo = new DbColumnInfo()
|
||||
{
|
||||
Value = column.PropertyInfo.GetValue(item, null),
|
||||
DbColumnName = GetDbColumnName(column.PropertyName),
|
||||
PropertyName = column.PropertyName,
|
||||
PropertyType = UtilMethods.GetUnderType(column.PropertyInfo),
|
||||
TableId = i
|
||||
};
|
||||
if (columnInfo.PropertyType.IsEnum()&& columnInfo.Value!=null)
|
||||
{
|
||||
if (this.Context.CurrentConnectionConfig.MoreSettings?.TableEnumIsString == true)
|
||||
{
|
||||
columnInfo.PropertyType = UtilConstants.StringType;
|
||||
columnInfo.Value = columnInfo.Value.ToString();
|
||||
}
|
||||
else
|
||||
{
|
||||
columnInfo.Value = Convert.ToInt64(columnInfo.Value);
|
||||
}
|
||||
}
|
||||
if (column.IsJson)
|
||||
{
|
||||
columnInfo.IsJson = true;
|
||||
if (columnInfo.Value != null)
|
||||
columnInfo.Value = this.Context.Utilities.SerializeObject(columnInfo.Value);
|
||||
}
|
||||
if (column.IsArray)
|
||||
{
|
||||
columnInfo.IsArray = true;
|
||||
}
|
||||
var tranColumn = EntityInfo.Columns.FirstOrDefault(it => it.IsTranscoding && it.DbColumnName.Equals(column.DbColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (tranColumn != null && columnInfo.Value.HasValue())
|
||||
{
|
||||
columnInfo.Value = UtilMethods.EncodeBase64(columnInfo.Value.ToString());
|
||||
}
|
||||
updateItem.Add(columnInfo);
|
||||
}
|
||||
this.UpdateBuilder.DbColumnInfoList.AddRange(updateItem);
|
||||
}
|
||||
|
||||
private void PreToSql()
|
||||
{
|
||||
|
||||
UpdateBuilder.PrimaryKeys = GetPrimaryKeys();
|
||||
if (this.IsWhereColumns)
|
||||
{
|
||||
foreach (var pkName in UpdateBuilder.PrimaryKeys)
|
||||
{
|
||||
if (WhereColumnList != null&& WhereColumnList.Count()>0)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
var isContains = this.UpdateBuilder.DbColumnInfoList.Select(it => it.DbColumnName.ToLower()).Contains(pkName.ToLower());
|
||||
Check.Exception(isContains == false, "Use UpdateColumns().WhereColumn() ,UpdateColumns need {0}", pkName);
|
||||
}
|
||||
}
|
||||
#region IgnoreColumns
|
||||
if (this.Context.IgnoreColumns != null && this.Context.IgnoreColumns.Any())
|
||||
{
|
||||
var currentIgnoreColumns = this.Context.IgnoreColumns.Where(it => it.EntityName == this.EntityInfo.EntityName).ToList();
|
||||
this.UpdateBuilder.DbColumnInfoList = this.UpdateBuilder.DbColumnInfoList.Where(it =>
|
||||
{
|
||||
return !currentIgnoreColumns.Any(i => it.PropertyName.Equals(i.PropertyName, StringComparison.CurrentCulture));
|
||||
}).ToList();
|
||||
}
|
||||
#endregion
|
||||
if (this.IsSingle)
|
||||
{
|
||||
foreach (var item in this.UpdateBuilder.DbColumnInfoList)
|
||||
{
|
||||
if (this.UpdateBuilder.Parameters == null) this.UpdateBuilder.Parameters = new List<SugarParameter>();
|
||||
if (this.UpdateBuilder.SetValues.Any(it => this.SqlBuilder.GetNoTranslationColumnName(it.Key) == item.PropertyName))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
var parameter = new SugarParameter(this.SqlBuilder.SqlParameterKeyWord + item.DbColumnName, item.Value, item.PropertyType);
|
||||
if (item.IsJson)
|
||||
{
|
||||
parameter.IsJson = true;
|
||||
}
|
||||
if (item.IsArray)
|
||||
{
|
||||
parameter.IsArray = true;
|
||||
}
|
||||
this.UpdateBuilder.Parameters.Add(parameter);
|
||||
}
|
||||
}
|
||||
|
||||
#region Identities
|
||||
List<string> identities = GetIdentityKeys();
|
||||
if (identities != null && identities.Any())
|
||||
{
|
||||
this.UpdateBuilder.DbColumnInfoList.ForEach(it =>
|
||||
{
|
||||
var mappingInfo = identities.SingleOrDefault(i => it.DbColumnName.Equals(i, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (mappingInfo != null && mappingInfo.Any())
|
||||
{
|
||||
it.IsIdentity = true;
|
||||
}
|
||||
});
|
||||
}
|
||||
#endregion
|
||||
List<string> primaryKey = GetPrimaryKeys();
|
||||
if (primaryKey != null && primaryKey.Count > 0)
|
||||
{
|
||||
this.UpdateBuilder.DbColumnInfoList.ForEach(it =>
|
||||
{
|
||||
var mappingInfo = primaryKey.SingleOrDefault(i => it.DbColumnName.Equals(i, StringComparison.CurrentCultureIgnoreCase));
|
||||
if (mappingInfo != null && mappingInfo.Any())
|
||||
{
|
||||
it.IsPrimarykey = true;
|
||||
}
|
||||
});
|
||||
}
|
||||
if (this.UpdateBuilder.Parameters.HasValue() && this.UpdateBuilder.SetValues.IsValuable())
|
||||
{
|
||||
this.UpdateBuilder.Parameters.RemoveAll(it => this.UpdateBuilder.SetValues.Any(v => (SqlBuilder.SqlParameterKeyWord + SqlBuilder.GetNoTranslationColumnName(v.Key)) == it.ParameterName));
|
||||
}
|
||||
}
|
||||
private string GetDbColumnName(string propertyName)
|
||||
{
|
||||
if (!IsMappingColumns)
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
if (this.Context.MappingColumns.Any(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)))
|
||||
{
|
||||
this.MappingColumnList = this.Context.MappingColumns.Where(it => it.EntityName.Equals(EntityInfo.EntityName, StringComparison.CurrentCultureIgnoreCase)).ToList();
|
||||
}
|
||||
if (MappingColumnList == null || !MappingColumnList.Any())
|
||||
{
|
||||
return propertyName;
|
||||
}
|
||||
else
|
||||
{
|
||||
var mappInfo = this.MappingColumnList.FirstOrDefault(it => it.PropertyName.Equals(propertyName, StringComparison.CurrentCultureIgnoreCase));
|
||||
return mappInfo == null ? propertyName : mappInfo.DbColumnName;
|
||||
}
|
||||
}
|
||||
private List<string> GetPrimaryKeys()
|
||||
{
|
||||
if (this.WhereColumnList.HasValue())
|
||||
{
|
||||
return this.WhereColumnList;
|
||||
}
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetPrimaries(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => it.IsPrimarykey).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
protected virtual List<string> GetIdentityKeys()
|
||||
{
|
||||
if (this.Context.IsSystemTablesConfig)
|
||||
{
|
||||
return this.Context.DbMaintenance.GetIsIdentities(this.Context.EntityMaintenance.GetTableName(this.EntityInfo.EntityName));
|
||||
}
|
||||
else
|
||||
{
|
||||
return this.EntityInfo.Columns.Where(it => it.IsIdentity).Select(it => it.DbColumnName).ToList();
|
||||
}
|
||||
}
|
||||
private void RestoreMapping()
|
||||
{
|
||||
if (IsAs)
|
||||
{
|
||||
this.Context.MappingTables = OldMappingTableList;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private void ValidateVersion()
|
||||
{
|
||||
var versionColumn = this.EntityInfo.Columns.FirstOrDefault(it => it.IsEnableUpdateVersionValidation);
|
||||
var pks = this.UpdateBuilder.DbColumnInfoList.Where(it => it.IsPrimarykey).ToList();
|
||||
if (versionColumn != null && this.IsVersionValidation)
|
||||
{
|
||||
Check.Exception(pks.IsNullOrEmpty(), "UpdateVersionValidation the primary key is required.");
|
||||
List<IConditionalModel> conModels = new List<IConditionalModel>();
|
||||
foreach (var item in pks)
|
||||
{
|
||||
conModels.Add(new ConditionalModel() { FieldName = item.DbColumnName, ConditionalType = ConditionalType.Equal, FieldValue = item.Value.ObjToString() });
|
||||
}
|
||||
var dbInfo = this.Context.Queryable<T>().Where(conModels).First();
|
||||
if (dbInfo != null)
|
||||
{
|
||||
var currentVersion = this.EntityInfo.Type.GetProperty(versionColumn.PropertyName).GetValue(UpdateObjs.Last(), null);
|
||||
var dbVersion = this.EntityInfo.Type.GetProperty(versionColumn.PropertyName).GetValue(dbInfo, null);
|
||||
Check.Exception(currentVersion == null, "UpdateVersionValidation entity property {0} is not null", versionColumn.PropertyName);
|
||||
Check.Exception(dbVersion == null, "UpdateVersionValidation database column {0} is not null", versionColumn.DbColumnName);
|
||||
if (versionColumn.PropertyInfo.PropertyType.IsIn(UtilConstants.IntType, UtilConstants.LongType))
|
||||
{
|
||||
if (Convert.ToInt64(dbVersion) != Convert.ToInt64(currentVersion))
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
else if (versionColumn.PropertyInfo.PropertyType.IsIn(UtilConstants.DateType))
|
||||
{
|
||||
if (dbVersion.ObjToDate() != currentVersion.ObjToDate())
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
else if (versionColumn.PropertyInfo.PropertyType.IsIn(UtilConstants.ByteArrayType))
|
||||
{
|
||||
if (UtilMethods.GetLong((byte[])dbVersion) != UtilMethods.GetLong((byte[])currentVersion))
|
||||
{
|
||||
throw new VersionExceptions(string.Format("UpdateVersionValidation {0} Not the latest version ", versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Check.ThrowNotSupportedException(string.Format("UpdateVersionValidation Not Supported Type [ {0} ] , {1}", versionColumn.PropertyInfo.PropertyType, versionColumn.PropertyName));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
private void After(string sql)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = UpdateBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.AfterData = GetDiffTable(sql, parameters);
|
||||
diffModel.Time = this.Context.Ado.SqlExecutionTime;
|
||||
if (this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent != null)
|
||||
this.Context.CurrentConnectionConfig.AopEvents.OnDiffLogEvent(diffModel);
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
if (this.RemoveCacheFunc != null)
|
||||
{
|
||||
this.RemoveCacheFunc();
|
||||
}
|
||||
}
|
||||
|
||||
private void Before(string sql)
|
||||
{
|
||||
if (this.IsEnableDiffLogEvent)
|
||||
{
|
||||
var isDisableMasterSlaveSeparation = this.Ado.IsDisableMasterSlaveSeparation;
|
||||
this.Ado.IsDisableMasterSlaveSeparation = true;
|
||||
var parameters = UpdateBuilder.Parameters;
|
||||
if (parameters == null)
|
||||
parameters = new List<SugarParameter>();
|
||||
diffModel.BeforeData = GetDiffTable(sql, parameters);
|
||||
diffModel.Sql = sql;
|
||||
diffModel.Parameters = parameters.ToArray();
|
||||
this.Ado.IsDisableMasterSlaveSeparation = isDisableMasterSlaveSeparation;
|
||||
}
|
||||
}
|
||||
private bool IsPrimaryKey(DbColumnInfo it)
|
||||
{
|
||||
var result = GetPrimaryKeys().Any(p => p.Equals(it.DbColumnName, StringComparison.CurrentCultureIgnoreCase) || p.Equals(it.PropertyName, StringComparison.CurrentCultureIgnoreCase));
|
||||
return result;
|
||||
}
|
||||
|
||||
private List<DiffLogTableInfo> GetDiffTable(string sql, List<SugarParameter> parameters)
|
||||
{
|
||||
List<DiffLogTableInfo> result = new List<DiffLogTableInfo>();
|
||||
DataTable dt = null;
|
||||
if (this.UpdateParameterIsNull)
|
||||
{
|
||||
var whereSql = Regex.Replace(sql, ".* WHERE ", "", RegexOptions.Singleline);
|
||||
if (sql.Contains("WHERE (EXISTS"))
|
||||
{
|
||||
whereSql=Regex.Match(sql, @"\(EXISTS.+").Value;
|
||||
}
|
||||
dt = this.Context.Queryable<T>().Filter(null, true).Where(whereSql).AddParameters(parameters).ToDataTable();
|
||||
}
|
||||
else
|
||||
{
|
||||
if (this.UpdateObjs.ToList().Count == 0)
|
||||
{
|
||||
dt = new DataTable();
|
||||
}
|
||||
else
|
||||
{
|
||||
dt = this.Context.Queryable<T>().Filter(null, true).WhereClassByPrimaryKey(this.UpdateObjs.ToList()).ToDataTable();
|
||||
}
|
||||
}
|
||||
if (dt.Rows != null && dt.Rows.Count > 0)
|
||||
{
|
||||
foreach (DataRow row in dt.Rows)
|
||||
{
|
||||
DiffLogTableInfo item = new DiffLogTableInfo();
|
||||
item.TableDescription = this.EntityInfo.TableDescription;
|
||||
item.TableName = this.EntityInfo.DbTableName;
|
||||
item.Columns = new List<DiffLogColumnInfo>();
|
||||
foreach (DataColumn col in dt.Columns)
|
||||
{
|
||||
var sugarColumn = this.EntityInfo.Columns.Where(it => it.DbColumnName != null).First(it =>
|
||||
it.DbColumnName.Equals(col.ColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||
DiffLogColumnInfo addItem = new DiffLogColumnInfo();
|
||||
addItem.Value = row[col.ColumnName];
|
||||
addItem.ColumnName = col.ColumnName;
|
||||
addItem.IsPrimaryKey = sugarColumn.IsPrimarykey;
|
||||
addItem.ColumnDescription = sugarColumn.ColumnDescription;
|
||||
item.Columns.Add(addItem);
|
||||
}
|
||||
result.Add(item);
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
private void ThrowUpdateByExpression()
|
||||
{
|
||||
Check.Exception(UpdateParameterIsNull == true, ErrorMessage.GetThrowMessage(" no support UpdateColumns and WhereColumns", "根据表达式更新 db.Updateable<T>() 禁止使用 UpdateColumns和WhereColumns,你可以使用 SetColumns Where 等。更新分为2种方式 1.根据表达式更新 2.根据实体或者集合更新， 具体用法请查看文档 "));
|
||||
}
|
||||
private void ThrowUpdateByObject()
|
||||
{
|
||||
Check.Exception(UpdateParameterIsNull == false, ErrorMessage.GetThrowMessage(" no support SetColumns and Where", "根据对像更新 db.Updateabe(对象) 禁止使用 SetColumns和Where ,你可以使用WhereColumns 和 UpdateColumns。 更新分为2种方式 1.根据表达式更新 2.根据实体或者集合更新 ， 具体用法请查看文档 "));
|
||||
}
|
||||
#endregion
|
||||
}
|
||||
}
|
||||
|
||||
@ -12,6 +12,8 @@ namespace SqlSugar
|
||||
UpdateBuilder UpdateBuilder { get; set; }
|
||||
bool UpdateParameterIsNull { get; set; }
|
||||
|
||||
int ExecuteCommandWithOptLock(bool isThrowError = false);
|
||||
Task<int> ExecuteCommandWithOptLockAsync(bool isThrowError = false);
|
||||
int ExecuteCommand();
|
||||
bool ExecuteCommandHasChange();
|
||||
Task<int> ExecuteCommandAsync();
|
||||
|
||||
@ -133,6 +133,30 @@ namespace SqlSugar
|
||||
};
|
||||
}
|
||||
|
||||
internal static object GetRandomByType(Type underType)
|
||||
{
|
||||
if (underType == UtilConstants.GuidType)
|
||||
{
|
||||
return Guid.NewGuid();
|
||||
}
|
||||
else if (underType == UtilConstants.LongType)
|
||||
{
|
||||
return SnowFlakeSingle.Instance.NextId();
|
||||
}
|
||||
else if (underType == UtilConstants.StringType)
|
||||
{
|
||||
return Guid.NewGuid() + "";
|
||||
}
|
||||
else if (underType == UtilConstants.DateType)
|
||||
{
|
||||
return System.DateTime.Now;
|
||||
}
|
||||
else
|
||||
{
|
||||
return Guid.NewGuid() + "";
|
||||
}
|
||||
}
|
||||
|
||||
public static bool IsAsyncMethod(MethodBase method)
|
||||
{
|
||||
if (method == null)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user