ow

CodeGenerator/CloudBuilder.CodeGenerator/Entity/CodeGeneratorRegistEntity.auto.cs

@@ -0,0 +1,39 @@
+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***
+        }
+    }
+}