From 68706d7d79c7b9a0e762be3e97588fe3838b5f81 Mon Sep 17 00:00:00 2001
From: sunkaixuan <610262374@qq.com>
Date: Mon, 12 Aug 2024 17:39:47 +0800
Subject: [PATCH] Update split table

---
 .../QueryableProvider/QueryableProvider.cs      | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/Src/Asp.NetCore2/SqlSugar/Abstract/QueryableProvider/QueryableProvider.cs b/Src/Asp.NetCore2/SqlSugar/Abstract/QueryableProvider/QueryableProvider.cs
index 94b5a1304..92e49f93d 100644
--- a/Src/Asp.NetCore2/SqlSugar/Abstract/QueryableProvider/QueryableProvider.cs
+++ b/Src/Asp.NetCore2/SqlSugar/Abstract/QueryableProvider/QueryableProvider.cs
@@ -1500,7 +1500,21 @@ namespace SqlSugar
                 }
                 else
                 {
-                    return this.Select<TResult>(this.SqlBuilder.SqlSelectAll);
+                    if (this.EntityInfo?.Type?.GetCustomAttribute<SplitTableAttribute>() != null&& this.QueryBuilder?.SelectValue?.ToString()=="*")
+                    {
+                        var columnAarray = this.Context.EntityMaintenance.GetEntityInfo<T>().Columns;
+                        var sql = string.Empty;
+                        var columns= columnAarray.Where(it => typeof(TResult).GetProperties().Any(s => s.Name.EqualCase(it.PropertyName))).Where(it => it.IsIgnore == false).ToList();
+                        if (columns.Any())
+                        {
+                            sql = string.Join(",", columns.Select(it => $"{SqlBuilder.GetTranslationColumnName(it.DbColumnName)} AS  {SqlBuilder.GetTranslationColumnName(it.PropertyName)} "));
+                        }
+                        return this.Select<TResult>(sql);
+                    }
+                    else
+                    {
+                        return this.Select<TResult>(this.SqlBuilder.SqlSelectAll);
+                    }
                 }
             }
             else
@@ -1687,6 +1701,7 @@ namespace SqlSugar
                 //}
                 var unionall = this.Context._UnionAll(tableQueryables.ToArray());
                 unionall.QueryBuilder.Includes = this.QueryBuilder.Includes;
+                unionall.QueryBuilder.EntityType = typeof(T);
                 if (unionall.QueryBuilder.Includes?.Any()==true) 
                 {
                     unionall.QueryBuilder.NoCheckInclude = true;