lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-04-24 18:04:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Synchronization code

Browse Source
This commit is contained in:
sunkaixuan 2023-11-25 01:45:04 +08:00
parent 00058680ac
commit 6be9aff42a
3 changed files with 28 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Src/Asp.NetCore2/SqlSugar/Abstract/DbMaintenanceProvider/Methods.cs
View File

@ -665,6 +665,10 @@ namespace SqlSugar
#endregion
#region Private
public virtual List<DbTableInfo> GetSchemaTables(EntityInfo entityInfo)
{
return null;
}
protected List<T> GetListOrCache<T>(string cacheKey, string sql)
{
return this.Context.Utilities.GetReflectionInoCacheInstance().GetOrCreate(cacheKey,

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

@ -321,21 +321,27 @@ namespace SqlSugar
#endregion
#region Methods
public override bool DropColumn(string tableName, string columnName)
public override List<DbTableInfo> GetSchemaTables(EntityInfo entityInfo)
{
if (Regex.IsMatch(tableName,@"^\w+$") && Regex.IsMatch(columnName,@"^\w+$"))
{
var sql = $"SELECT distinct dc.name AS ConstraintName \r\nFROM sys.default_constraints dc\r\nJOIN sys.columns c ON dc.parent_column_id = c.column_id\r\nWHERE dc.parent_object_id = OBJECT_ID('{tableName}')\r\nAND c.name = '{columnName}';";
var checks=this.Context.Ado.SqlQuery<string>(sql);
foreach (var checkName in checks)
if (entityInfo.DbTableName.Contains(".") && this.Context.CurrentConnectionConfig.DbType == DbType.SqlServer)
{
var schema = entityInfo.DbTableName.Split('.').First();
var isAny = GetSchemas().Any(it => it.EqualCase(schema))||schema.EqualCase("dbo");
if (isAny)
{
if (checkName?.ToUpper()?.StartsWith("DF__")==true)
{
this.Context.Ado.ExecuteCommand($"ALTER TABLE {SqlBuilder.GetTranslationColumnName(tableName)} DROP CONSTRAINT {checkName}");
}
var tableInfos = this.Context.Ado.SqlQuery<DbTableInfo>(@"SELECT schem.name+'.'+tb.name Name,tb.Description from
( SELECT obj.name,Convert(nvarchar(max),prop.value)as Description,obj.schema_id FROM sys.objects obj
LEFT JOIN sys.extended_properties prop
ON obj.object_id=prop.major_id
and prop.minor_id=0
AND (prop.Name='MS_Description' OR prop.Name is null)
WHERE obj.type IN('U')) tb
inner join sys.schemas as schem
on tb.schema_id=schem.schema_id ");
return tableInfos;
}
}
return base.DropColumn(tableName, columnName);
return null;
}
public override List<string> GetDbTypes()
{

10
Src/Asp.NetCore2/SqlSugar/SpliteTable/SplitTableContext.cs
View File

@ -58,14 +58,18 @@ namespace SqlSugar
}
public List<SplitTableInfo> GetTables()
{
if (StaticConfig.SplitTableGetTablesFunc != null)
if (StaticConfig.SplitTableGetTablesFunc != null)
{
return StaticConfig.SplitTableGetTablesFunc();
}
var oldIsEnableLogEvent = this.Context.Ado.IsEnableLogEvent;
this.Context.Ado.IsEnableLogEvent = false;
var tableInfos = this.Context.DbMaintenance.GetTableInfoList(false);
List<SplitTableInfo> result = Service.GetAllTables(this.Context,EntityInfo,tableInfos);
List<DbTableInfo> tableInfos =((DbMaintenanceProvider)this.Context.DbMaintenance).GetSchemaTables(EntityInfo);
if (tableInfos == null)
{
tableInfos = this.Context.DbMaintenance.GetTableInfoList(false);
}
List<SplitTableInfo> result = Service.GetAllTables(this.Context, EntityInfo, tableInfos);
this.Context.Ado.IsEnableLogEvent = oldIsEnableLogEvent;
return result;
}