sqlsugar/Src/Asp.Net/SqlSugar/SpliteTable/SplitTableContext.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;

namespace SqlSugar
{
    public class SplitTableContextResult<T> 
    {
        public List<T> Items { get;  set; }
        public SplitTableContext Helper { get;  set; }

        public string  GetTableName()
        {
            return GetTableNames().First();
        }
        public string [] GetTableNames() 
        {
            List<string> result = new List<string>();
            var attribute = typeof(T).GetCustomAttribute<SplitTableAttribute>() as SplitTableAttribute;
            Check.Exception(attribute == null, $" {typeof(T).Name} need SplitTableAttribute");
            foreach (var item in Items)
            {
                result.Add(Helper.GetTableName(Helper.GetValue(attribute.SplitType,item)));
            }
            return  result.Distinct().ToArray();
        }
        public string[] GetTableNames(SplitType splitType)
        {
            List<string> result = new List<string>();
            foreach (var item in Items)
            {
                result.Add(Helper.GetTableName(Helper.GetValue(splitType, item)));
            }
            return result.ToArray();
        }
    }
    public class SplitTableContext
    {
        internal SqlSugarProvider Context { get; set; }
        internal EntityInfo EntityInfo { get; set; }
        internal ISplitTableService Service { get; set; }
        private SplitTableContext() { }
        internal SplitTableContext(SqlSugarProvider context) 
        {
            this.Context = context;
            if (this.Context.CurrentConnectionConfig.ConfigureExternalServices != null&&this.Context.CurrentConnectionConfig.ConfigureExternalServices.SplitTableService!=null)
            {
                Service = this.Context.CurrentConnectionConfig.ConfigureExternalServices.SplitTableService;
            }
            else
            {
                Service = new DateSplitTableService();
            }
        }
        public List<SplitTableInfo> GetTables()
        {
            if (StaticConfig.SplitTableGetTablesFunc != null)
            {
                return GetCustomGetTables();
            }
            var oldIsEnableLogEvent = this.Context.Ado.IsEnableLogEvent;
            this.Context.Ado.IsEnableLogEvent = false;
            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;
        }

        private  List<SplitTableInfo> GetCustomGetTables()
        {
            var oldIsEnableLogEvent = this.Context.Ado.IsEnableLogEvent;
            this.Context.Ado.IsEnableLogEvent = false;
            var tables = StaticConfig.SplitTableGetTablesFunc();
            List<SplitTableInfo> result = Service.GetAllTables(this.Context, EntityInfo, tables.Select(it=>new DbTableInfo() { Name=it.TableName }).ToList());
            this.Context.Ado.IsEnableLogEvent = oldIsEnableLogEvent;
            return result.ToList();
        }

        public string GetDefaultTableName()
        {
           return Service.GetTableName(this.Context,EntityInfo);
        }
        public string GetTableName(SplitType splitType) 
        {
            return Service.GetTableName(this.Context,EntityInfo, splitType);
        }
        public string GetTableName(SplitType splitType, object fieldValue)
        {
            return Service.GetTableName(this.Context,EntityInfo, splitType, fieldValue);
        }
        public string GetTableName(object fieldValue)
        {
            var attribute = EntityInfo.Type.GetCustomAttribute<SplitTableAttribute>() as SplitTableAttribute;
            Check.Exception(attribute == null, $" {EntityInfo.EntityName} need SplitTableAttribute");
            return Service.GetTableName(this.Context, EntityInfo, attribute.SplitType, fieldValue);
        }
        public object GetValue(SplitType splitType, object entityValue)
        {
            return Service.GetFieldValue(this.Context,EntityInfo, splitType, entityValue);
        }
        internal void CheckPrimaryKey()
        {
            Check.Exception(EntityInfo.Columns.Any(it => it.IsIdentity == true), ErrorMessage.GetThrowMessage("Split table can't IsIdentity=true", "分表禁止使用自增列"));
        }
    }
}