From be67b525fa1d8cc7ad3fec1239318f7e1100593a Mon Sep 17 00:00:00 2001
From: sunkaixuna <610262374@qq.com>
Date: Sun, 31 Oct 2021 09:50:39 +0800
Subject: [PATCH] Split table

---
 .../SqlSugar/IntegrationServices/SplitTableService.cs | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/Src/Asp.Net/SqlSugar/IntegrationServices/SplitTableService.cs b/Src/Asp.Net/SqlSugar/IntegrationServices/SplitTableService.cs
index d55593d16..5269a75ff 100644
--- a/Src/Asp.Net/SqlSugar/IntegrationServices/SplitTableService.cs
+++ b/Src/Asp.Net/SqlSugar/IntegrationServices/SplitTableService.cs
@@ -35,7 +35,16 @@ namespace SqlSugar
         }
         public virtual string GetTableName(ISqlSugarClient db, EntityInfo EntityInfo)
         {
-            return GetTableName(db, EntityInfo, SplitType.Day);
+            var splitTableAttribute = EntityInfo.Type.GetCustomAttribute<SplitTableAttribute>();
+            if (splitTableAttribute != null)
+            {
+                var type=(splitTableAttribute as SplitTableAttribute).SplitType;
+                return GetTableName(db, EntityInfo, type);
+            }
+            else
+            {
+                return GetTableName(db, EntityInfo, SplitType.Day);
+            }
         }
         public virtual string GetTableName(ISqlSugarClient db, EntityInfo EntityInfo, SplitType splitType)
         {