40 lines
1.2 KiB
C#
40 lines
1.2 KiB
C#
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***
|
||
}
|
||
}
|
||
}
|