hello-shop/samples/MultiTenancySample/MultiTenancySample.DatabaseIsolationService/EntityFrameworks/DataSeeding.cs
2024-07-05 19:25:58 +08:00

47 lines
1.7 KiB
C#

// Copyright (c) HelloShop Corporation. All rights reserved.
// See the license file in the project root for more information.
using Microsoft.EntityFrameworkCore;
using MultiTenancySample.DatabaseIsolationService.Entities;
using MultiTenancySample.ServiceDefaults;
namespace MultiTenancySample.DatabaseIsolationService.EntityFrameworks
{
public class DataSeeding(IDbContextFactory<DatabaseIsolationServiceDbContext> dbContext, ICurrentTenant currentTenant)
{
public async Task SeedDataAsync()
{
currentTenant.SetTenant("Tenant1");
DatabaseIsolationServiceDbContext dbContext1 = await dbContext.CreateDbContextAsync();
if (await dbContext1.Database.EnsureCreatedAsync())
{
await dbContext1.AddRangeAsync(new List<Product> {
new() { Id = Guid.NewGuid(), Name = "Product1"},
new() { Id = Guid.NewGuid(), Name = "Product3"},
new() { Id = Guid.NewGuid(), Name = "Product5"}
});
await dbContext1.SaveChangesAsync();
}
currentTenant.SetTenant("Tenant2");
DatabaseIsolationServiceDbContext dbContext2 = await dbContext.CreateDbContextAsync();
if (await dbContext2.Database.EnsureCreatedAsync())
{
await dbContext2.AddRangeAsync(new List<Product>
{
new() { Id = Guid.NewGuid(), Name = "Product2"},
new() { Id = Guid.NewGuid(), Name = "Product4"},
new() { Id = Guid.NewGuid(), Name = "Product6"}
});
await dbContext2.SaveChangesAsync();
}
}
}
}