138 lines
4.1 KiB
C#
138 lines
4.1 KiB
C#
using BlueWest.Data;
|
|
using Microsoft.EntityFrameworkCore;
|
|
|
|
namespace BlueWest.WebApi.EF.Model
|
|
{
|
|
/// <summary>
|
|
/// Database model configuration extensions
|
|
/// </summary>
|
|
public static class ModelBuilderExtensions
|
|
{
|
|
#region Initialization
|
|
|
|
/// <summary>
|
|
/// Setup the database model
|
|
/// </summary>
|
|
/// <param name="modelBuilder"></param>
|
|
public static void ConfigureCurrentDbModel(this ModelBuilder modelBuilder)
|
|
{
|
|
modelBuilder
|
|
.ConfigureDatabaseKeys()
|
|
.CurrencyModel()
|
|
.ConfigureUserModel();
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region DatabasePK
|
|
|
|
private static ModelBuilder ConfigureDatabaseKeys(this ModelBuilder modelBuilder)
|
|
{
|
|
// User
|
|
modelBuilder
|
|
.Entity<User>(builder => builder
|
|
.HasKey(user => user.Id))
|
|
.Entity<User>().Property(user => user.Id)
|
|
.ValueGeneratedOnAdd();
|
|
|
|
// Country
|
|
modelBuilder
|
|
.Entity<Country>(builder => builder
|
|
.HasKey(country => country.Id))
|
|
.Entity<Country>().Property(country => country.Id)
|
|
.ValueGeneratedOnAdd();
|
|
|
|
// Currency
|
|
modelBuilder
|
|
.Entity<Currency>(builder => builder
|
|
.HasKey(currency => currency.Id))
|
|
.Entity<Currency>()
|
|
.Property(currency => currency.Id)
|
|
.ValueGeneratedOnAdd();
|
|
|
|
// Company PK
|
|
modelBuilder
|
|
.Entity<Company>(builder => builder
|
|
.HasKey(company => company.Id))
|
|
.Entity<Company>().Property(company => company.Id)
|
|
.ValueGeneratedOnAdd();
|
|
|
|
// Industry PK
|
|
modelBuilder
|
|
.Entity<Industry>(builder => builder
|
|
.HasKey(industry => industry.Id))
|
|
.Entity<Industry>().Property(industry => industry.Id)
|
|
.ValueGeneratedOnAdd();
|
|
|
|
|
|
// Product
|
|
modelBuilder
|
|
.Entity<Industry>(builder => builder
|
|
.HasKey(industry => industry.Id));
|
|
|
|
modelBuilder
|
|
.Entity<Industry>()
|
|
.Property(industry => industry.Id)
|
|
.ValueGeneratedOnAdd();
|
|
|
|
// FinanceOp
|
|
return
|
|
modelBuilder.Entity<FinanceOp>(builder => builder
|
|
.HasKey(financeOp => financeOp.Id));
|
|
|
|
|
|
}
|
|
#endregion
|
|
|
|
#region Database Models
|
|
|
|
private static ModelBuilder CurrencyModel(this ModelBuilder modelBuilder)
|
|
{
|
|
|
|
modelBuilder
|
|
.Entity<Currency>()
|
|
.HasMany(ub => ub.Countries)
|
|
.WithMany(au => au.Currencies);
|
|
|
|
modelBuilder.Entity<Country>()
|
|
.HasMany(ub => ub.Currencies)
|
|
.WithMany(au => au.Countries);
|
|
|
|
return modelBuilder;
|
|
|
|
}
|
|
|
|
private static ModelBuilder ConfigureUserModel(this ModelBuilder modelBuilder)
|
|
{
|
|
|
|
// User => Country
|
|
modelBuilder
|
|
.Entity<User>(builder => builder
|
|
.HasOne<Country>()
|
|
.WithMany(co => co.Users));
|
|
|
|
|
|
// Country => User
|
|
modelBuilder
|
|
.Entity<Country>(builder => builder
|
|
.HasMany<User>()
|
|
.WithOne(u => u.Country)
|
|
.HasForeignKey(x => x.CountryId));
|
|
|
|
|
|
// FinanceOp => User
|
|
modelBuilder
|
|
.Entity<FinanceOp>(builder => builder
|
|
.HasOne<User>()
|
|
.WithMany(ft => ft.FinanceTransactions)
|
|
.HasForeignKey(x => x.UserId));
|
|
|
|
|
|
return modelBuilder;
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
}
|
|
|