GitHub
/
SqlSugar
6
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Code optimization

SqlSugar5
sunkaixuan 2 years ago
parent
c535a76ee8
commit
50a7eedb43
2 changed files with 106 additions and 42 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 100
      Src/Asp.Net/SqlSugar/Abstract/InsertableProvider/InsertableHelper.cs
  2. 48
      Src/Asp.Net/SqlSugar/Abstract/InsertableProvider/InsertableProvider.cs

100
Src/Asp.Net/SqlSugar/Abstract/InsertableProvider/InsertableHelper.cs
View File

@ -512,5 +512,105 @@ namespace SqlSugar
}
#endregion
#region Insert PkList
private List<Type> InsertPkListWithFunc<Type>(EntityColumnInfo pkInfo)
{
InsertBuilder.IsReturnPkList = true;
InsertBuilder.IsNoPage = true;
string sql = _ExecuteCommand();
sql = this.InsertBuilder.ConvertInsertReturnIdFunc(SqlBuilder.GetTranslationColumnName(pkInfo.DbColumnName), sql);
var result = Ado.SqlQuery<Type>(sql, InsertBuilder.Parameters == null ? null : InsertBuilder.Parameters.ToArray());
After(sql, null);
return result;
}
private List<Type> InsertPkListNoFunc<Type>(EntityColumnInfo pkInfo)
{
if (this.Ado.Transaction != null)
{
return ReturnDefaultIdentity<Type>(pkInfo);
}
else
{
try
{
this.Context.Ado.BeginTran();
var result = ReturnDefaultIdentity<Type>(pkInfo);
this.Context.Ado.CommitTran();
return result;
}
catch (Exception ex)
{
this.Context.Ado.RollbackTran();
throw ex;
}
}
}
private List<Type> InsertPkListIdentityCount1<Type>(EntityColumnInfo pkInfo)
{
if (pkInfo.UnderType == UtilConstants.IntType)
{
return new List<Type> { (Type)(object)this.ExecuteReturnIdentity() };
}
else
{
return new List<Type> { (Type)(object)this.ExecuteReturnBigIdentity() };
}
}
private List<Type> InsertPkListLong<Type>()
{
var list = this.ExecuteReturnSnowflakeIdList();
try
{
return list.Cast<Type>().ToList();
}
catch
{
Check.ExceptionEasy($"long to ExecuteReturnPkList<{typeof(Type).Name}> error ", $" long 转换成ExecuteReturnPkList<{typeof(Type).Name}>失败");
return null;
}
}
private List<Type> InsertPkListGuid<Type>(EntityColumnInfo pkInfo)
{
Check.ExceptionEasy(pkInfo.UnderType.Name != typeof(Type).Name, $"{pkInfo.UnderType.Name} to ExecuteReturnPkList<{typeof(Type).Name}> error ", $" {pkInfo.UnderType.Name} 转换成ExecuteReturnPkList<{typeof(Type).Name}>失败");
this.ExecuteCommand();
List<Type> result = new List<Type>();
if (InsertBuilder.DbColumnInfoList.HasValue())
{
foreach (var item in InsertBuilder.DbColumnInfoList)
{
var isPk = item.DbColumnName.EqualCase(pkInfo.DbColumnName);
if (isPk)
{
result.Add((Type)item.Value);
}
}
}
return result;
}
private List<Type> ReturnDefaultIdentity<Type>(EntityColumnInfo pkInfo)
{
List<Type> result = new List<Type>();
foreach (var item in this.InsertObjs)
{
var insertable = this.Context.Insertable(item)
.InsertColumns(this.InsertBuilder.DbColumnInfoList.Select(it => it.DbColumnName).Distinct().ToArray());
if (pkInfo.UnderType == UtilConstants.IntType)
{
result.Add((Type)(object)insertable.ExecuteReturnIdentity());
}
else
{
result.Add((Type)(object)insertable.ExecuteReturnBigIdentity());
}
}
return result;
}
#endregion
}
}

48
Src/Asp.Net/SqlSugar/Abstract/InsertableProvider/InsertableProvider.cs
View File

@ -79,62 +79,26 @@ namespace SqlSugar
var isIdEntity = pkInfo.IsIdentity|| (pkInfo.OracleSequenceName.HasValue()&&this.Context.CurrentConnectionConfig.DbType==DbType.Oracle);
if (pkInfo.UnderType == UtilConstants.LongType)
{
var list = this.ExecuteReturnSnowflakeIdList();
try
{
return list.Cast<Type>().ToList();
}
catch
{
Check.ExceptionEasy($"long to ExecuteReturnPkList<{typeof(Type).Name}> error ", $" long 转换成ExecuteReturnPkList<{typeof(Type).Name}>失败");
return null;
}
return InsertPkListLong<Type>();
}
else if (isIdEntity&&this.InsertObjs.Length==1)
{
if (pkInfo.UnderType == UtilConstants.IntType)
{
return new List<Type> { (Type)(object)this.ExecuteReturnIdentity() };
}
else
{
return new List<Type> { (Type)(object)this.ExecuteReturnBigIdentity() };
}
return InsertPkListIdentityCount1<Type>(pkInfo);
}
else if (isIdEntity && this.InsertBuilder.ConvertInsertReturnIdFunc == null)
{
Check.ExceptionEasy("The current database does not support batch auto increment", "当前数据库不支持批量返回自增");
return null;
return InsertPkListNoFunc<Type>(pkInfo);
}
else if (isIdEntity && this.InsertBuilder.ConvertInsertReturnIdFunc != null)
{
InsertBuilder.IsReturnPkList = true;
InsertBuilder.IsNoPage = true;
string sql = _ExecuteCommand();
sql = this.InsertBuilder.ConvertInsertReturnIdFunc(SqlBuilder.GetTranslationColumnName(pkInfo.DbColumnName),sql);
var result = Ado.SqlQuery<Type>(sql, InsertBuilder.Parameters == null ? null : InsertBuilder.Parameters.ToArray());
After(sql, null);
return result;
return InsertPkListWithFunc<Type>(pkInfo);
}
else
{
Check.ExceptionEasy(pkInfo.UnderType.Name != typeof(Type).Name,$"{pkInfo.UnderType.Name} to ExecuteReturnPkList<{typeof(Type).Name}> error ", $" {pkInfo.UnderType.Name} 转换成ExecuteReturnPkList<{typeof(Type).Name}>失败");
this.ExecuteCommand();
List<Type> result = new List<Type>();
if (InsertBuilder.DbColumnInfoList.HasValue())
{
foreach (var item in InsertBuilder.DbColumnInfoList)
{
var isPk = item.DbColumnName.EqualCase(pkInfo.DbColumnName);
if (isPk)
{
result.Add((Type)item.Value);
}
}
}
return result;
return InsertPkListGuid<Type>(pkInfo);
}
}
public virtual int ExecuteReturnIdentity()
{
if (this.InsertObjs.Count() == 1 && this.InsertObjs.First() == null)

Write Preview
Loading…
Cancel
Save