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***
|
|
}
|
|
}
|
|
}
|