49 lines
1.9 KiB
C#
49 lines
1.9 KiB
C#
|
using System;
|
||
|
|
using Microsoft.EntityFrameworkCore;
|
||
|
|
using APT.Infrastructure.Api;
|
||
|
|
namespace APT.Infrastructure.EF
|
||
|
|
{
|
||
|
|
public class TenantEntityBuilder<TDbContext> : ITenantEntityBuilder<TDbContext>
|
||
|
|
where TDbContext : DbContext
|
||
|
|
{
|
||
|
|
private readonly IEntityScaner<TDbContext> entityScaner;
|
||
|
|
private readonly TenantOption tenantOption;
|
||
|
|
|
||
|
|
private readonly TenantInfo tenantInfo;
|
||
|
|
public TenantEntityBuilder(IEntityScaner<TDbContext> entityScaner, TenantOption tenantOption, TenantInfo tenantInfo)
|
||
|
|
{
|
||
|
|
this.tenantInfo = tenantInfo;
|
||
|
|
this.tenantOption = tenantOption;
|
||
|
|
this.entityScaner = entityScaner;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void UpdateEntities(ModelBuilder modelBuilder)
|
||
|
|
{
|
||
|
|
var dbsetProperties = entityScaner.ScanEntityTypes();
|
||
|
|
|
||
|
|
foreach (var property in dbsetProperties)
|
||
|
|
{
|
||
|
|
var entity = modelBuilder.Entity(property.PropertyType);
|
||
|
|
switch (this.tenantOption.ConnectionType)
|
||
|
|
{
|
||
|
|
case ConnectionResolverType.BySchema:
|
||
|
|
var tableName = this.tenantOption.TableNameFunc?.Invoke(this.tenantInfo, property.PropertyName)
|
||
|
|
?? property.PropertyName;
|
||
|
|
var schemaName = this.tenantOption.SchemaFunc?.Invoke(this.tenantInfo)
|
||
|
|
?? this.tenantInfo.Name;
|
||
|
|
entity.ToTable(tableName, schemaName);
|
||
|
|
break;
|
||
|
|
case ConnectionResolverType.ByTable:
|
||
|
|
tableName = this.tenantOption.TableNameFunc?.Invoke(this.tenantInfo, property.PropertyName)
|
||
|
|
?? $"{this.tenantInfo.Name}_{property.PropertyName}";
|
||
|
|
entity.ToTable(tableName);
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|