From 62b67ea93aa7897af17327aabd46a1d7cfaa2c06 Mon Sep 17 00:00:00 2001
From: sunkaixuan <610262374@qq.com>
Date: Wed, 26 Feb 2025 20:04:57 +0800
Subject: [PATCH] Update TDengine

---
 .../DbMaintenance/TDengineDbMaintenance.cs    |  2 +-
 .../Tools/SqlSugarExtensions.cs               | 29 +++++++++++++++++++
 .../Tools/UtilMethods.cs                      |  7 ++++-
 3 files changed, 36 insertions(+), 2 deletions(-)
 create mode 100644 Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/SqlSugarExtensions.cs

diff --git a/Src/Asp.NetCore2/SqlSugar.TDengineCore/TDengine/DbMaintenance/TDengineDbMaintenance.cs b/Src/Asp.NetCore2/SqlSugar.TDengineCore/TDengine/DbMaintenance/TDengineDbMaintenance.cs
index d5234dea6..d7640f725 100644
--- a/Src/Asp.NetCore2/SqlSugar.TDengineCore/TDengine/DbMaintenance/TDengineDbMaintenance.cs
+++ b/Src/Asp.NetCore2/SqlSugar.TDengineCore/TDengine/DbMaintenance/TDengineDbMaintenance.cs
@@ -1,4 +1,4 @@
- using SqlSugar.TDengineAdo;
+using SqlSugar.TDengineAdo;
 using System;
 using System.Collections.Generic;
 using System.Data;
diff --git a/Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/SqlSugarExtensions.cs b/Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/SqlSugarExtensions.cs
new file mode 100644
index 000000000..5860cac78
--- /dev/null
+++ b/Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/SqlSugarExtensions.cs
@@ -0,0 +1,29 @@
+using SqlSugar.TDengine;
+using System;
+using System.Collections.Generic;
+using System.Reflection;
+using System.Text;
+
+namespace SqlSugar 
+{
+     public static class SqlSugarExtensions
+    {
+        public static void MappingSTableName<T>(this ISqlSugarClient db,string newSTableName)
+        {
+            STableAttribute sTableAttribute = typeof(T).GetCustomAttribute<STableAttribute>();
+            if (db.TempItems == null) 
+            {
+                db.TempItems = new Dictionary<string, object>();
+            }
+            if (sTableAttribute != null)
+            {
+                var key = "GetCommonSTableAttribute_" + sTableAttribute.STableName;
+                if (db.TempItems.ContainsKey(key)) 
+                {
+                    db.TempItems.Remove(key);
+                }
+                db.TempItems.Add(key, newSTableName);
+            } 
+        }
+    }
+}
diff --git a/Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/UtilMethods.cs b/Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/UtilMethods.cs
index 41b0acf42..58b4360c4 100644
--- a/Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/UtilMethods.cs
+++ b/Src/Asp.NetCore2/SqlSugar.TDengineCore/Tools/UtilMethods.cs
@@ -19,7 +19,12 @@ namespace SqlSugar.TDengine
     {
         public static STableAttribute GetCommonSTableAttribute(ISqlSugarClient db,STableAttribute sTableAttribute)
         {
-            return sTableAttribute;
+            var key = "GetCommonSTableAttribute_" + sTableAttribute?.STableName;
+            if (db.TempItems?.ContainsKey(key) == true) 
+            {
+                sTableAttribute.STableName = db.TempItems[key] +"";
+            }
+            return sTableAttribute!;
         }
         public static long ToUnixTimestamp(DateTime dateTime)
         {