tk/FreeSql
1
0
Fork 0
mirror of https://github.com/nsnail/FreeSql.git synced 2025-06-19 20:38:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

为 InsertOrUpdateDictImpl 增加 WhereIdentityPrimary

Browse Source
This commit is contained in:
hyzx86
2023-06-07 19:23:48 +08:00
parent 181fb90ef6
commit 695251fc88
2 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
FreeSql/Extensions/FreeSqlGlobalExtensions.cs
View File

@ -1086,11 +1086,17 @@ SELECT ");
}
return pks.ToArray();
}
public static void SetTablePrimary(TableInfo table, params string[] primarys)
public static void SetTablePrimary(TableInfo table, bool isIdenity, params string[] primarys)
{
foreach (var primary in primarys)
{
if (table.ColumnsByCs.TryGetValue(string.Concat(primary), out var col)) col.Attribute.IsPrimary = true;
if (table.ColumnsByCs.TryGetValue(string.Concat(primary), out var col))
{
col.Attribute.IsPrimary = true;
if (isIdenity)
col.Attribute.IsIdentity = true;
}
else throw new Exception(CoreStrings.GetPrimarys_ParameterError_IsNotDictKey(primary));
}
table.Primarys = table.Columns.Where(a => a.Value.Attribute.IsPrimary).Select(a => a.Value).ToArray();
@ -1172,7 +1178,13 @@ SELECT ");
public InsertOrUpdateDictImpl WherePrimary(params string[] primarys)
{
UpdateDictImpl.SetTablePrimary(_insertOrUpdateProvider._table, primarys);
UpdateDictImpl.SetTablePrimary(_insertOrUpdateProvider._table, false, primarys);
_insertOrUpdateProvider._tempPrimarys = _insertOrUpdateProvider._table.Primarys;
return this;
}
public InsertOrUpdateDictImpl WhereIdentityPrimary(params string[] primarys)
{
UpdateDictImpl.SetTablePrimary(_insertOrUpdateProvider._table, true, primarys);
_insertOrUpdateProvider._tempPrimarys = _insertOrUpdateProvider._table.Primarys;
return this;
}