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

Update SqlServer db.DbMain.Remark

Browse Source
This commit is contained in:
sunkaixuan 2023-07-26 02:20:00 +08:00
parent d238b944ba
commit 65f76387b4
1 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
Src/Asp.NetCore2/SqlSugar/Realization/SqlServer/DbMaintenance/SqlServerDbMaintenance.cs
View File

@ -321,6 +321,57 @@ namespace SqlSugar
#endregion
#region Methods
private bool IsAnySchemaTable(string tableName)
{
if (tableName == null||!tableName.Contains(".") )
{
return false;
}
var list = GetSchemas() ?? new List<string>();
list.Add("dbo");
var isAnySchemas = list.Any(it => it.EqualCase(tableName?.Split('.').FirstOrDefault()));
return isAnySchemas;
}
public override bool IsAnyColumnRemark(string columnName, string tableName)
{
if (IsAnySchemaTable(tableName))
{
var schema = tableName.Split('.').First();
tableName = tableName.Split('.').Last();
var temp = this.IsAnyColumnRemarkSql.Replace("'dbo'", $"'{schema}'");
string sql = string.Format(temp, columnName, tableName);
var dt = this.Context.Ado.GetDataTable(sql);
return dt.Rows != null && dt.Rows.Count > 0;
}
return base.IsAnyColumnRemark(columnName, tableName);
}
public override bool DeleteColumnRemark(string columnName, string tableName)
{
if (IsAnySchemaTable(tableName))
{
var schema = tableName.Split('.').First();
tableName = tableName.Split('.').Last();
var temp = this.DeleteColumnRemarkSql.Replace(",dbo,", $",{schema},");
string sql = string.Format(temp, columnName, tableName);
this.Context.Ado.ExecuteCommand(sql);
return true;
}
return base.DeleteColumnRemark(columnName, tableName);
}
public override bool AddColumnRemark(string columnName, string tableName, string description)
{
if (IsAnySchemaTable(tableName))
{
var schema = tableName.Split('.').First();
tableName = tableName.Split('.').Last();
var temp = this.AddColumnRemarkSql.Replace("N'dbo'", $"N'{schema}'");
string sql = string.Format(temp, columnName, tableName, description);
this.Context.Ado.ExecuteCommand(sql);
return true;
}
return base.AddColumnRemark(columnName, tableName, description);
}
public override void AddDefaultValue(EntityInfo entityInfo)
{
var dbColumns = this.GetColumnInfosByTableName(entityInfo.DbTableName, false);