From 038cd245ecbfa2a337665c1047586190ef25f2af Mon Sep 17 00:00:00 2001
From: sunkaixuan <610262374@qq.com>
Date: Mon, 18 Mar 2024 16:55:41 +0800
Subject: [PATCH] Update Vastbase

---
 .../SqlSugar/Abstract/SugarProvider/SqlSugarAccessory.cs  | 8 +++-----
 .../PostgreSQL/DbMaintenance/PostgreSQLDbMaintenance.cs   | 2 +-
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/Src/Asp.NetCore2/SqlSugar/Abstract/SugarProvider/SqlSugarAccessory.cs b/Src/Asp.NetCore2/SqlSugar/Abstract/SugarProvider/SqlSugarAccessory.cs
index df45e91e5..8691d51b4 100644
--- a/Src/Asp.NetCore2/SqlSugar/Abstract/SugarProvider/SqlSugarAccessory.cs
+++ b/Src/Asp.NetCore2/SqlSugar/Abstract/SugarProvider/SqlSugarAccessory.cs
@@ -493,11 +493,9 @@ namespace SqlSugar
                     break;
                 case DbType.Vastbase:
                     config.DbType = DbType.PostgreSQL;
-                    if (this.TempItems == null) 
-                    {
-                        this.TempItems = new Dictionary<string, object>();
-                    }
-                    this.TempItems.Add("DbType.Vastbase", "DbType.Vastbase");
+                    if (this.CurrentConnectionConfig.MoreSettings==null) 
+                        this.CurrentConnectionConfig.MoreSettings = new ConnMoreSettings();
+                    this.CurrentConnectionConfig.MoreSettings.DatabaseModel = DbType.Vastbase;
                     break;
                 case DbType.OceanBase:
                     config.DbType = DbType.MySql; 
diff --git a/Src/Asp.NetCore2/SqlSugar/Realization/PostgreSQL/DbMaintenance/PostgreSQLDbMaintenance.cs b/Src/Asp.NetCore2/SqlSugar/Realization/PostgreSQL/DbMaintenance/PostgreSQLDbMaintenance.cs
index 3cfbd8680..c6a1e2fa2 100644
--- a/Src/Asp.NetCore2/SqlSugar/Realization/PostgreSQL/DbMaintenance/PostgreSQLDbMaintenance.cs
+++ b/Src/Asp.NetCore2/SqlSugar/Realization/PostgreSQL/DbMaintenance/PostgreSQLDbMaintenance.cs
@@ -381,7 +381,7 @@ WHERE tgrelid = '"+tableName+"'::regclass");
             });
             if (!GetDataBaseList(newDb).Any(it => it.Equals(databaseName, StringComparison.CurrentCultureIgnoreCase)))
             {
-                var isVast = this.Context?.TempItems?.ContainsKey("DbType.Vastbase")==true;
+                var isVast = this.Context?.CurrentConnectionConfig?.MoreSettings?.DatabaseModel==DbType.Vastbase;
                 var dbcompatibility = "";
                 if (isVast) 
                 {