mirror of
https://gitee.com/dotnetchina/SqlSugar.git
synced 2026-01-28 19:21:25 +08:00
Update DuckDb
This commit is contained in:
sunkaixuan
@@ -76,7 +76,146 @@ namespace SqlSugar.DuckDB
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
public override void ExistLogic(EntityInfo entityInfo)
|
||||||
|
{
|
||||||
|
if (entityInfo.Columns.HasValue() && entityInfo.IsDisabledUpdateAll == false)
|
||||||
|
{
|
||||||
|
//Check.Exception(entityInfo.Columns.Where(it => it.IsPrimarykey).Count() > 1, "Multiple primary keys do not support modifications");
|
||||||
|
|
||||||
|
var tableName = GetTableName(entityInfo);
|
||||||
|
var dbColumns = this.Context.DbMaintenance.GetColumnInfosByTableName(tableName, false);
|
||||||
|
ConvertColumns(dbColumns);
|
||||||
|
var entityColumns = entityInfo.Columns.Where(it => it.IsIgnore == false).ToList();
|
||||||
|
var dropColumns = dbColumns
|
||||||
|
.Where(dc => !entityColumns.Any(ec => dc.DbColumnName.Equals(ec.OldDbColumnName, StringComparison.CurrentCultureIgnoreCase)))
|
||||||
|
.Where(dc => !entityColumns.Any(ec => dc.DbColumnName.Equals(ec.DbColumnName, StringComparison.CurrentCultureIgnoreCase)))
|
||||||
|
.ToList();
|
||||||
|
var addColumns = entityColumns
|
||||||
|
.Where(ec => ec.OldDbColumnName.IsNullOrEmpty() || !dbColumns.Any(dc => dc.DbColumnName.Equals(ec.OldDbColumnName, StringComparison.CurrentCultureIgnoreCase)))
|
||||||
|
.Where(ec => !dbColumns.Any(dc => ec.DbColumnName.Equals(dc.DbColumnName, StringComparison.CurrentCultureIgnoreCase))).ToList();
|
||||||
|
var alterColumns =new List<EntityColumnInfo>();
|
||||||
|
|
||||||
|
alterColumns.RemoveAll(entityColumnInfo =>
|
||||||
|
{
|
||||||
|
var bigStringArray = StaticConfig.CodeFirst_BigString.Replace("varcharmax", "nvarchar(max)").Split(',');
|
||||||
|
var dbColumnInfo = dbColumns.FirstOrDefault(dc => dc.DbColumnName.EqualCase(entityColumnInfo.DbColumnName));
|
||||||
|
var isMaxString = (dbColumnInfo?.Length == -1 && dbColumnInfo?.DataType?.EqualCase("nvarchar") == true);
|
||||||
|
var isRemove =
|
||||||
|
dbColumnInfo != null
|
||||||
|
&& bigStringArray.Contains(entityColumnInfo.DataType)
|
||||||
|
&& isMaxString;
|
||||||
|
return isRemove;
|
||||||
|
});
|
||||||
|
var renameColumns = entityColumns
|
||||||
|
.Where(it => !string.IsNullOrEmpty(it.OldDbColumnName))
|
||||||
|
.Where(entityColumn => dbColumns.Any(dbColumn => entityColumn.OldDbColumnName.Equals(dbColumn.DbColumnName, StringComparison.CurrentCultureIgnoreCase)))
|
||||||
|
.ToList();
|
||||||
|
|
||||||
|
|
||||||
|
var isMultiplePrimaryKey = dbColumns.Where(it => it.IsPrimarykey).Count() > 1 || entityColumns.Where(it => it.IsPrimarykey).Count() > 1;
|
||||||
|
|
||||||
|
|
||||||
|
var isChange = false;
|
||||||
|
foreach (var item in addColumns)
|
||||||
|
{
|
||||||
|
this.Context.DbMaintenance.AddColumn(tableName, EntityColumnToDbColumn(entityInfo, tableName, item));
|
||||||
|
isChange = true;
|
||||||
|
}
|
||||||
|
if (entityInfo.IsDisabledDelete == false)
|
||||||
|
{
|
||||||
|
foreach (var item in dropColumns)
|
||||||
|
{
|
||||||
|
this.Context.DbMaintenance.DropColumn(tableName, item.DbColumnName);
|
||||||
|
isChange = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
foreach (var item in alterColumns)
|
||||||
|
{
|
||||||
|
|
||||||
|
if (this.Context.CurrentConnectionConfig.DbType == DbType.Oracle)
|
||||||
|
{
|
||||||
|
var entityColumnItem = entityColumns.FirstOrDefault(y => y.DbColumnName == item.DbColumnName);
|
||||||
|
if (entityColumnItem != null && !string.IsNullOrEmpty(entityColumnItem.DataType))
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
this.Context.DbMaintenance.UpdateColumn(tableName, EntityColumnToDbColumn(entityInfo, tableName, item));
|
||||||
|
isChange = true;
|
||||||
|
}
|
||||||
|
foreach (var item in renameColumns)
|
||||||
|
{
|
||||||
|
this.Context.DbMaintenance.RenameColumn(tableName, item.OldDbColumnName, item.DbColumnName);
|
||||||
|
isChange = true;
|
||||||
|
}
|
||||||
|
var isAddPrimaryKey = false;
|
||||||
|
foreach (var item in entityColumns)
|
||||||
|
{
|
||||||
|
var dbColumn = dbColumns.FirstOrDefault(dc => dc.DbColumnName.Equals(item.DbColumnName, StringComparison.CurrentCultureIgnoreCase));
|
||||||
|
if (dbColumn == null) continue;
|
||||||
|
bool pkDiff, idEntityDiff;
|
||||||
|
KeyAction(item, dbColumn, out pkDiff, out idEntityDiff);
|
||||||
|
if (dbColumn != null && pkDiff && !idEntityDiff && isMultiplePrimaryKey == false)
|
||||||
|
{
|
||||||
|
var isAdd = item.IsPrimarykey;
|
||||||
|
if (isAdd)
|
||||||
|
{
|
||||||
|
isAddPrimaryKey = true;
|
||||||
|
this.Context.DbMaintenance.AddPrimaryKey(tableName, item.DbColumnName);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
this.Context.DbMaintenance.DropConstraint(tableName, string.Format("PK_{0}_{1}", tableName, item.DbColumnName));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if ((pkDiff || idEntityDiff) && isMultiplePrimaryKey == false)
|
||||||
|
{
|
||||||
|
ChangeKey(entityInfo, tableName, item);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (isAddPrimaryKey == false && entityColumns.Count(it => it.IsPrimarykey) == 1 && dbColumns.Count(it => it.IsPrimarykey) == 0)
|
||||||
|
{
|
||||||
|
var addPk = entityColumns.First(it => it.IsPrimarykey);
|
||||||
|
this.Context.DbMaintenance.AddPrimaryKey(tableName, addPk.DbColumnName);
|
||||||
|
}
|
||||||
|
if (isMultiplePrimaryKey)
|
||||||
|
{
|
||||||
|
var oldPkNames = dbColumns.Where(it => it.IsPrimarykey).Select(it => it.DbColumnName.ToLower()).OrderBy(it => it).ToList();
|
||||||
|
var newPkNames = entityColumns.Where(it => it.IsPrimarykey).Select(it => it.DbColumnName.ToLower()).OrderBy(it => it).ToList();
|
||||||
|
if (oldPkNames.Count == 0 && newPkNames.Count > 1)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
this.Context.DbMaintenance.AddPrimaryKeys(tableName, newPkNames.ToArray());
|
||||||
|
}
|
||||||
|
catch (Exception ex)
|
||||||
|
{
|
||||||
|
Check.Exception(true, ErrorMessage.GetThrowMessage("The current database does not support changing multiple primary keys. " + ex.Message, "当前数据库不支持修改多主键," + ex.Message));
|
||||||
|
throw ex;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (!Enumerable.SequenceEqual(oldPkNames, newPkNames))
|
||||||
|
{
|
||||||
|
Check.Exception(true, ErrorMessage.GetThrowMessage("Modification of multiple primary key tables is not supported. Delete tables while creating", "不支持修改多主键表，请删除表在创建"));
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
if (isChange && IsBackupTable)
|
||||||
|
{
|
||||||
|
this.Context.DbMaintenance.BackupTable(tableName, tableName + DateTime.Now.ToString("yyyyMMddHHmmss"), MaxBackupDataRows);
|
||||||
|
}
|
||||||
|
ExistLogicEnd(entityColumns);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
private bool IsNoSamePrecision(EntityColumnInfo ec, DbColumnInfo dc)
|
||||||
|
{
|
||||||
|
if (this.Context.CurrentConnectionConfig.MoreSettings?.EnableCodeFirstUpdatePrecision == true)
|
||||||
|
{
|
||||||
|
return ec.DecimalDigits != dc.DecimalDigits && ec.UnderType.IsIn(UtilConstants.DobType, UtilConstants.DecType);
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
protected override void ChangeKey(EntityInfo entityInfo, string tableName, EntityColumnInfo item)
|
protected override void ChangeKey(EntityInfo entityInfo, string tableName, EntityColumnInfo item)
|
||||||
{
|
{
|
||||||
this.Context.DbMaintenance.UpdateColumn(tableName, EntityColumnToDbColumn(entityInfo, tableName, item));
|
this.Context.DbMaintenance.UpdateColumn(tableName, EntityColumnToDbColumn(entityInfo, tableName, item));
|
||||||
|
|||||||
21
Src/Asp.NetCore2/SqlSugar.DuckDBCore/Extensions.cs
Normal file
21
Src/Asp.NetCore2/SqlSugar.DuckDBCore/Extensions.cs
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Text;
|
||||||
|
|
||||||
|
namespace SqlSugar.DuckDB
|
||||||
|
{
|
||||||
|
public static class Extensions
|
||||||
|
{
|
||||||
|
public static bool EqualCase(this string thisValue, string equalValue)
|
||||||
|
{
|
||||||
|
if (thisValue != null && equalValue != null)
|
||||||
|
{
|
||||||
|
return thisValue.ToLower() == equalValue.ToLower();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return thisValue == equalValue;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user