mirror of
https://github.com/RPCS3/discord-bot.git
synced 2025-03-03 23:37:03 +00:00
32 lines
1.0 KiB
C#
32 lines
1.0 KiB
C#
using System;
|
|
using System.Linq;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.EntityFrameworkCore.Metadata.Internal;
|
|
|
|
namespace CompatBot.Database
|
|
{
|
|
internal static class PrimaryKeyConvention
|
|
{
|
|
public static void ConfigureDefaultPkConvention(this ModelBuilder modelBuilder, string keyProperty = "Id")
|
|
{
|
|
if (string.IsNullOrEmpty(keyProperty))
|
|
throw new ArgumentException(nameof(keyProperty));
|
|
|
|
foreach (var entity in modelBuilder.Model.GetEntityTypes())
|
|
{
|
|
var pk = entity.GetKeys().FirstOrDefault(k => k.IsPrimaryKey());
|
|
pk?.SetName(keyProperty);
|
|
}
|
|
}
|
|
|
|
public static void ConfigureNoPkConvention(this ModelBuilder modelBuilder)
|
|
{
|
|
foreach (var entity in modelBuilder.Model.GetEntityTypes())
|
|
{
|
|
var pk = entity.GetKeys().FirstOrDefault(k => k.IsPrimaryKey());
|
|
if (pk != null)
|
|
entity.RemoveKey(pk.Properties);
|
|
}
|
|
}
|
|
}
|
|
} |