ow

CodeGenerator/Templates/RegistEntity.cshtml

@@ -0,0 +1,47 @@
+@{
+    ModelData modelData = Model;
+    SysTableSchemaViewEntity[] sysTableSchemas = modelData.sysTableSchemas;
+    NamespaceObject namespaceObject = modelData.NamespaceObject;
+}
+using CloudBuilder.Core.DependencyInjection.EFCore;
+using Microsoft.EntityFrameworkCore;
+using System.Reflection;
+
+namespace @namespaceObject.Namespace@{<text>.Entity</text>}
+
+{
+    public class @namespaceObject.Namespace.Replace(".","")RegistEntity : 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);
+        }
+
+        ///<summary>
+        ///如果存在多个主键的，要在RegistKeys方法注入
+        ///builder.Entity<xxxxxxEntity>().HasKey(c => new { c.TableName, c.FieldName });                
+        ///builder.Entity<xxxxxxEntity>(eb => { eb.HasNoKey(); });
+        ///</summary>
+        ///<param name="builder"></param>
+        public void RegistKeys(ModelBuilder builder)
+        {
+            //***BEGIN***
+
+            //***END***
+        }
+    }
+}
+