using CloudBuilder.Core.DependencyInjection.EFCore;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;

namespace CloudBuilder.CodeGenerator.Entity
{
    public partial class CodeGeneratorRegistEntity : IMasterRegistEntityType<ModelBuilder>
    {
        /// <summary>
        /// 注入模块内的所有实体类
        /// </summary>
        /// <param name="builder"></param>
        public void RegistEntity(ModelBuilder builder)
        {
            string _assemblyName = Assembly.GetExecutingAssembly().GetName().Name;
            //注入所有Entity实体类
            AddEntityType(builder, _assemblyName);
            //如果Entity实体类不是单一主键，要单独重写注入
            RegistKeys(builder);
        }

        public void AddEntityType(ModelBuilder builder, string _assemblyName)
        {
            RegistEntityHelper.AddMasterEntityType(builder, _assemblyName);
        }

        public void RegistKeys(ModelBuilder builder)
        {
            //***BEGIN***
            builder.Entity<SysTableSchemaViewEntity>(eb => { eb.HasNoKey(); });
            //***END***
        }
    }
}