lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2026-01-22 02:51:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add overload

Browse Source
This commit is contained in:
sunkaixuan
2023-03-30 18:39:32 +08:00
parent fc0781a5ff
commit 90357bb9e9
2 changed files with 50 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -5,6 +5,7 @@ using System.Linq;
using System.Linq.Expressions;
using System.Reflection;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace SqlSugar
@@ -36,7 +37,7 @@ namespace SqlSugar
#region Core
public void AddQueue()
{
if (this.InsertObjs!=null&&this.InsertObjs.Length > 0&& this.InsertObjs[0]!=null)
if (this.InsertObjs != null && this.InsertObjs.Length > 0 && this.InsertObjs[0] != null)
{
var sqlObj = this.ToSql();
this.Context.Queues.Add(sqlObj.Key, sqlObj.Value);
@@ -71,17 +72,17 @@ namespace SqlSugar
RestoreMapping();
return new KeyValuePair<string, List<SugarParameter>>(sql, InsertBuilder.Parameters);
}
public async Task<List<Type>> ExecuteReturnPkListAsync<Type>()
public async Task<List<Type>> ExecuteReturnPkListAsync<Type>()
{
return await Task.Run(() => ExecuteReturnPkList<Type>());
}
public virtual List<Type> ExecuteReturnPkList<Type>()
public virtual List<Type> ExecuteReturnPkList<Type>()
{
var pkInfo= this.EntityInfo.Columns.FirstOrDefault(it => it.IsPrimarykey == true);
Check.ExceptionEasy(pkInfo==null,"ExecuteReturnPkList need primary key", "ExecuteReturnPkList需要主键");
Check.ExceptionEasy(this.EntityInfo.Columns.Count(it => it.IsPrimarykey == true)>1, "ExecuteReturnPkList ，Only support technology single primary key", "ExecuteReturnPkList只支技单主键");
var isIdEntity = pkInfo.IsIdentity|| (pkInfo.OracleSequenceName.HasValue()&&this.Context.CurrentConnectionConfig.DbType==DbType.Oracle);
if (isIdEntity&&this.InsertObjs.Length==1)
var pkInfo = this.EntityInfo.Columns.FirstOrDefault(it => it.IsPrimarykey == true);
Check.ExceptionEasy(pkInfo == null, "ExecuteReturnPkList need primary key", "ExecuteReturnPkList需要主键");
Check.ExceptionEasy(this.EntityInfo.Columns.Count(it => it.IsPrimarykey == true) > 1, "ExecuteReturnPkList ，Only support technology single primary key", "ExecuteReturnPkList只支技单主键");
var isIdEntity = pkInfo.IsIdentity || (pkInfo.OracleSequenceName.HasValue() && this.Context.CurrentConnectionConfig.DbType == DbType.Oracle);
if (isIdEntity && this.InsertObjs.Length == 1)
{
return InsertPkListIdentityCount1<Type>(pkInfo);
}
@@ -114,7 +115,7 @@ namespace SqlSugar
if (InsertBuilder.IsOleDb)
{
var isAuto = false;
if (this.Context.Ado.IsAnyTran() == false && this.Context.CurrentConnectionConfig.IsAutoCloseConnection)
if (this.Context.Ado.IsAnyTran() == false && this.Context.CurrentConnectionConfig.IsAutoCloseConnection)
{
isAuto = this.Context.CurrentConnectionConfig.IsAutoCloseConnection;
this.Context.CurrentConnectionConfig.IsAutoCloseConnection = false;
@@ -146,7 +147,7 @@ namespace SqlSugar
if (InsertBuilder.IsOleDb)
{
var isAuto = false;
if (this.Context.Ado.IsAnyTran()==false&&this.Context.CurrentConnectionConfig.IsAutoCloseConnection)
if (this.Context.Ado.IsAnyTran() == false && this.Context.CurrentConnectionConfig.IsAutoCloseConnection)
{
isAuto = this.Context.CurrentConnectionConfig.IsAutoCloseConnection;
this.Context.CurrentConnectionConfig.IsAutoCloseConnection = false;
@@ -161,25 +162,25 @@ namespace SqlSugar
}
else
{
result= Convert.ToInt64(Ado.GetScalar(sql, InsertBuilder.Parameters == null ? null : InsertBuilder.Parameters.ToArray()));
result = Convert.ToInt64(Ado.GetScalar(sql, InsertBuilder.Parameters == null ? null : InsertBuilder.Parameters.ToArray()));
}
After(sql, result);
return result;
}
public virtual long ExecuteReturnSnowflakeId()
public virtual long ExecuteReturnSnowflakeId()
{
if (this.InsertObjs.Length > 1)
if (this.InsertObjs.Length > 1)
{
return this.ExecuteReturnSnowflakeIdList().First();
}
var id = SnowFlakeSingle.instance.getID();
var entity = this.Context.EntityMaintenance.GetEntityInfo<T>();
var snowProperty=entity.Columns.FirstOrDefault(it => it.IsPrimarykey && it.PropertyInfo.PropertyType == UtilConstants.LongType);
Check.Exception(snowProperty==null, "The entity sets the primary key and is long");
var snowProperty = entity.Columns.FirstOrDefault(it => it.IsPrimarykey && it.PropertyInfo.PropertyType == UtilConstants.LongType);
Check.Exception(snowProperty == null, "The entity sets the primary key and is long");
Check.Exception(snowProperty.IsIdentity == true, "SnowflakeId IsIdentity can't true");
foreach (var item in this.InsertBuilder.DbColumnInfoList.Where(it=>it.PropertyName==snowProperty.PropertyName))
foreach (var item in this.InsertBuilder.DbColumnInfoList.Where(it => it.PropertyName == snowProperty.PropertyName))
{
item.Value = id;
snowProperty?.PropertyInfo.SetValue(this.InsertObjs.First(), id);
@@ -187,7 +188,7 @@ namespace SqlSugar
this.ExecuteCommand();
return id;
}
public List<long> ExecuteReturnSnowflakeIdList()
public List<long> ExecuteReturnSnowflakeIdList()
{
List<long> result = new List<long>();
var entity = this.Context.EntityMaintenance.GetEntityInfo<T>();
@@ -208,6 +209,11 @@ namespace SqlSugar
this.ExecuteCommand();
return result;
}
public Task<long> ExecuteReturnSnowflakeIdAsync(CancellationToken token)
{
this.Context.Ado.CancellationToken= token;
return ExecuteReturnSnowflakeIdAsync();
}
public async Task<long> ExecuteReturnSnowflakeIdAsync()
{
var id = SnowFlakeSingle.instance.getID();
@@ -245,6 +251,12 @@ namespace SqlSugar
return result;
}
public Task<List<long>> ExecuteReturnSnowflakeIdListAsync(CancellationToken token)
{
this.Ado.CancellationToken= token;
return ExecuteReturnSnowflakeIdListAsync();
}
public virtual T ExecuteReturnEntity()
{
ExecuteCommandIdentityIntoEntity();
@@ -302,7 +314,11 @@ namespace SqlSugar
this.Context.EntityMaintenance.GetProperty<T>(identityKey).SetValue(result, setValue, null);
return idValue > 0;
}
public Task<int> ExecuteCommandAsync(CancellationToken token)
{
this.Context.Ado.CancellationToken= token;
return ExecuteCommandAsync();
}
public async Task<int> ExecuteCommandAsync()
{
if (this.InsertObjs.Count() == 1 && this.InsertObjs.First() == null)
@@ -315,6 +331,11 @@ namespace SqlSugar
if (result == -1) return this.InsertObjs.Count();
return result;
}
public Task<int> ExecuteReturnIdentityAsync(CancellationToken token)
{
this.Ado.CancellationToken= token;
return ExecuteReturnIdentityAsync();
}
public virtual async Task<int> ExecuteReturnIdentityAsync()
{
if (this.InsertObjs.Count() == 1 && this.InsertObjs.First() == null)
@@ -404,6 +425,11 @@ namespace SqlSugar
this.Context.EntityMaintenance.GetProperty<T>(identityKey).SetValue(result, setValue, null);
return idValue > 0;
}
public Task<long> ExecuteReturnBigIdentityAsync(CancellationToken token)
{
this.Context.Ado.CancellationToken= token;
return ExecuteReturnBigIdentityAsync();
}
public virtual async Task<long> ExecuteReturnBigIdentityAsync()
{
if (this.InsertObjs.Count() == 1 && this.InsertObjs.First() == null)

6
Src/Asp.Net/SqlSugar/Interface/Insertable.cs
View File

@@ -3,6 +3,7 @@ using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
namespace SqlSugar
@@ -12,20 +13,25 @@ namespace SqlSugar
InsertBuilder InsertBuilder { get; set; }
int ExecuteCommand();
Task<int> ExecuteCommandAsync();
Task<int> ExecuteCommandAsync(CancellationToken token);
List<Type> ExecuteReturnPkList<Type>();
Task<List<Type>> ExecuteReturnPkListAsync<Type>();
long ExecuteReturnSnowflakeId();
List<long> ExecuteReturnSnowflakeIdList();
Task<long> ExecuteReturnSnowflakeIdAsync();
Task<long> ExecuteReturnSnowflakeIdAsync(CancellationToken token);
Task<List<long>> ExecuteReturnSnowflakeIdListAsync();
Task<List<long>> ExecuteReturnSnowflakeIdListAsync(CancellationToken token);
int ExecuteReturnIdentity();
Task<int> ExecuteReturnIdentityAsync();
Task<int> ExecuteReturnIdentityAsync(CancellationToken token);
T ExecuteReturnEntity();
Task<T> ExecuteReturnEntityAsync();
bool ExecuteCommandIdentityIntoEntity();
Task<bool> ExecuteCommandIdentityIntoEntityAsync();
long ExecuteReturnBigIdentity();
Task<long> ExecuteReturnBigIdentityAsync();
Task<long> ExecuteReturnBigIdentityAsync(CancellationToken token);
IInsertable<T> AS(string tableName);
IInsertable<T> AsType(Type tableNameType);
IInsertable<T> With(string lockString);