44 lines
1.4 KiB
C#
44 lines
1.4 KiB
C#
using CloudBuilder.Core.DependencyInjection.EFCore;
|
||
using Microsoft.EntityFrameworkCore;
|
||
using System.Reflection;
|
||
|
||
namespace CloudBuilder.AI.Entity
|
||
{
|
||
public class CloudBuilderAIRegistEntity : 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***
|
||
|
||
|
||
builder.Entity<AiSentenceViewEntity>(eb => { eb.HasNoKey(); });
|
||
//***END***
|
||
}
|
||
}
|
||
}
|
||
|