62 lines
2.3 KiB
C#
62 lines
2.3 KiB
C#
using ASC.Common;
|
|
using ASC.Core.Common.EF.Model;
|
|
|
|
using Microsoft.EntityFrameworkCore;
|
|
|
|
namespace ASC.Core.Common.EF.Context
|
|
{
|
|
public partial class TenantDbContext : BaseDbContext
|
|
{
|
|
public DbSet<DbTenant> Tenants { get; set; }
|
|
public DbSet<User> Users { get; set; }
|
|
public DbSet<UserSecurity> UserSecurity { get; set; }
|
|
public DbSet<DbTenantVersion> TenantVersion { get; set; }
|
|
public DbSet<DbTenantPartner> TenantPartner { get; set; }
|
|
public DbSet<DbTenantForbiden> TenantForbiden { get; set; }
|
|
public DbSet<TenantIpRestrictions> TenantIpRestrictions { get; set; }
|
|
public DbSet<DbCoreSettings> CoreSettings { get; set; }
|
|
|
|
public TenantDbContext() { }
|
|
public TenantDbContext(DbContextOptions<TenantDbContext> options)
|
|
: base(options)
|
|
{
|
|
Database.EnsureCreated();
|
|
}
|
|
protected override void OnModelCreating(ModelBuilder modelBuilder)
|
|
{
|
|
if (baseName == "default")
|
|
{
|
|
modelBuilder.MySqlAddUser();
|
|
modelBuilder.MySqlAddCoreSettings();
|
|
modelBuilder.MySqlAddDbTenant();
|
|
modelBuilder.MySqlAddUserSecurity();
|
|
modelBuilder.MySqlAddDbTenantForbiden();
|
|
modelBuilder.MySqlAddTenantIpRestrictions();
|
|
modelBuilder.MySqlAddDbTenantPartner();
|
|
modelBuilder.MySqlAddDbTenantVersion();
|
|
}
|
|
else
|
|
{
|
|
modelBuilder.PgSqlAddUser();
|
|
modelBuilder.PgSqlAddCoreSettings();
|
|
modelBuilder.PgSqlAddDbTenant();
|
|
modelBuilder.PgSqlAddUserSecurity();
|
|
modelBuilder.PgSqlAddDbTenantForbiden();
|
|
modelBuilder.PgSqlAddTenantIpRestrictions();
|
|
modelBuilder.PgSqlAddDbTenantPartner();
|
|
modelBuilder.PgSqlAddDbTenantVersion();
|
|
}
|
|
OnModelCreatingPartial(modelBuilder);
|
|
}
|
|
partial void OnModelCreatingPartial(ModelBuilder modelBuilder);
|
|
}
|
|
|
|
public static class TenantDbExtension
|
|
{
|
|
public static DIHelper AddTenantDbContextService(this DIHelper services)
|
|
{
|
|
return services.AddDbContextManagerService<TenantDbContext>();
|
|
}
|
|
}
|
|
}
|