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

Update DuckDb

Browse Source
This commit is contained in:
sunkaixuan
2025-04-20 14:24:16 +08:00
parent e605ff71a1
commit eb28bf6212
1 changed files with 8 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
Src/Asp.NetCore2/SqlSugar.DuckDBCore/DuckDB/DbMaintenance/DuckDBDbMaintenance.cs
View File

@@ -22,7 +22,7 @@ namespace SqlSugar.DuckDB
{ {
string schema = GetSchema(); string schema = GetSchema();
string sql = @"select cast (pclass.oid as int4) as TableId,cast(ptables.tablename as varchar) as TableName, string sql = @"select cast (pclass.oid as int4) as TableId,cast(ptables.tablename as varchar) as TableName,
pcolumn.column_name as DbColumnName,pcolumn.udt_name as DataType, pcolumn.column_name as DbColumnName,pcolumn.data_type as DataType,
CASE WHEN pcolumn.numeric_scale >0 THEN pcolumn.numeric_precision ELSE pcolumn.character_maximum_length END as Length, CASE WHEN pcolumn.numeric_scale >0 THEN pcolumn.numeric_precision ELSE pcolumn.character_maximum_length END as Length,
pcolumn.column_default as DefaultValue, pcolumn.column_default as DefaultValue,
pcolumn.numeric_scale as DecimalDigits, pcolumn.numeric_scale as DecimalDigits,
@@ -396,16 +396,10 @@ namespace SqlSugar.DuckDB
} }
try try
{ {
string sql = $@"select string sql = $@"
kcu.column_name as key_column SELECT name AS key_column
from information_schema.table_constraints tco FROM pragma_table_info('{GetSchema()}.{tableName.Trim('"')}')
join information_schema.key_column_usage kcu WHERE pk > 0";
on kcu.constraint_name = tco.constraint_name
and kcu.constraint_schema = tco.constraint_schema
and kcu.constraint_name = tco.constraint_name
where tco.constraint_type = 'PRIMARY KEY'
and kcu.table_schema='{GetSchema()}' and
upper(kcu.table_name)=upper('{tableName.TrimEnd('"').TrimStart('"')}')";
List<string> pkList = new List<string>(); List<string> pkList = new List<string>();
if (isCache) if (isCache)
{ {
@@ -415,17 +409,14 @@ namespace SqlSugar.DuckDB
{ {
pkList = this.Context.Ado.SqlQuery<string>(sql); pkList = this.Context.Ado.SqlQuery<string>(sql);
} }
if (pkList.Count >1)
{
foreach (var item in result) foreach (var item in result)
{ {
if (pkList.Select(it=>it.ToUpper()).Contains(item.DbColumnName.ToUpper())) if (pkList.Select(it => it.ToUpper()).Contains(item.DbColumnName.ToUpper()))
{ {
item.IsPrimarykey = true; item.IsPrimarykey = true;
} }
} }
} }
}
catch catch
{ {