lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-04-24 18:04:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Code optimization

Browse Source
This commit is contained in:
sunkaixuan 2022-08-24 00:36:42 +08:00
parent c535a76ee8
commit 50a7eedb43
2 changed files with 106 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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)