lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-09-20 02:29:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update DuckdB

Browse Source
This commit is contained in:
sunkaixuan
2025-04-20 14:48:26 +08:00
parent eb28bf6212
commit fadafc463f
1 changed files with 55 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
Src/Asp.NetCore2/SqlSugar.DuckDBCore/DuckDB/DuckDBProvider.cs
View File

@@ -5,6 +5,7 @@ using System.Data.Common;
using System.Linq; using System.Linq;
using System.Text; using System.Text;
using System.Threading.Tasks; using System.Threading.Tasks;
using System.Transactions;
using DuckDB; using DuckDB;
using DuckDB.NET.Data; using DuckDB.NET.Data;
using DuckDB.NET.Native; using DuckDB.NET.Native;
@@ -51,21 +52,37 @@ namespace SqlSugar.DuckDB
{ {
base.BeginTran(); base.BeginTran();
} }
/// <summary>
/// Only SqlServer
/// </summary>
/// <param name="iso"></param>
/// <param name="transactionName"></param>
public override void BeginTran(IsolationLevel iso, string transactionName)
{
base.BeginTran(iso);
}
public override IDataAdapter GetAdapter() public override IDataAdapter GetAdapter()
{ {
return new DuckDBDataAdapter(); return new DuckDBDataAdapter();
} }
public override DbCommand GetCommand(string sql, SugarParameter[] parameters) public override DbCommand GetCommand(string sql, SugarParameter[] parameters)
{ {
var helper = new DuckDBInsertBuilder();
helper.Context = this.Context;
List<SugarParameter> orderParameters = new List<SugarParameter>();
if (parameters.HasValue())
{
foreach (var p in parameters)
{
if (!p.ParameterName.StartsWith(this.SqlParameterKeyWord))
{
p.ParameterName = this.SqlParameterKeyWord + p.ParameterName;
}
}
orderParameters = parameters.Where(it => sql.Contains(it.ParameterName))
.Select(it => new { p = it, sort = GetSortId(sql, it) })
.OrderBy(it => it.sort)
.Where(it => it.sort != 0)
.Select(it => it.p)
.ToList();
foreach (var param in parameters.OrderByDescending(it => it.ParameterName.Length))
{
sql = sql.Replace(param.ParameterName, "?");
}
}
DuckDBCommand sqlCommand = new DuckDBCommand(sql, (DuckDBConnection)this.Connection); DuckDBCommand sqlCommand = new DuckDBCommand(sql, (DuckDBConnection)this.Connection);
sqlCommand.CommandType = this.CommandType; sqlCommand.CommandType = this.CommandType;
sqlCommand.CommandTimeout = this.CommandTimeOut; sqlCommand.CommandTimeout = this.CommandTimeOut;
@@ -73,14 +90,33 @@ namespace SqlSugar.DuckDB
{ {
sqlCommand.Transaction = (DuckDBTransaction)this.Transaction; sqlCommand.Transaction = (DuckDBTransaction)this.Transaction;
} }
if (parameters.HasValue()) if (orderParameters.HasValue())
{ {
IDataParameter[] ipars = ToIDbDataParameter(parameters); DuckDBParameter[] ipars = (DuckDBParameter[])ToIDbDataParameter(orderParameters.ToArray());
sqlCommand.Parameters.AddRange((DuckDBParameter[])ipars); sqlCommand.Parameters.AddRange(ipars);
} }
CheckConnection(); CheckConnection();
return sqlCommand; return sqlCommand;
} }
private static int GetSortId(string sql, SugarParameter it)
{
return new List<int>() {
0,
sql.IndexOf(it.ParameterName+")"),
sql.IndexOf(it.ParameterName+" "),
sql.IndexOf(it.ParameterName+"="),
sql.IndexOf(it.ParameterName+"+"),
sql.IndexOf(it.ParameterName+"-"),
sql.IndexOf(it.ParameterName+";"),
sql.IndexOf(it.ParameterName+","),
sql.IndexOf(it.ParameterName+"*"),
sql.IndexOf(it.ParameterName+"/"),
sql.IndexOf(it.ParameterName+"|"),
sql.IndexOf(it.ParameterName+"&"),
sql.EndsWith(it.ParameterName)?sql.IndexOf(it.ParameterName):0
}.Max();
}
public override void SetCommandToAdapter(IDataAdapter dataAdapter, DbCommand command) public override void SetCommandToAdapter(IDataAdapter dataAdapter, DbCommand command)
{ {
((DuckDBDataAdapter)dataAdapter).SelectCommand = (DuckDBCommand)command; ((DuckDBDataAdapter)dataAdapter).SelectCommand = (DuckDBCommand)command;
@@ -140,7 +176,8 @@ namespace SqlSugar.DuckDB
//{ //{
// sqlParameter.DbType = ((DuckDBType)parameter.CustomDbType); // sqlParameter.DbType = ((DuckDBType)parameter.CustomDbType);
//} //}
} sqlParameter.ParameterName = null;
}
return result; return result;
} }
@@ -162,6 +199,10 @@ namespace SqlSugar.DuckDB
parameter.Value = Convert.ToInt64(parameter.Value); parameter.Value = Convert.ToInt64(parameter.Value);
} }
} }
public override void BeginTran(System.Data.IsolationLevel iso, string transactionName)
{
base.BeginTran();
}
} }
} }