98 lines
4.5 KiB
C#
98 lines
4.5 KiB
C#
using Microsoft.EntityFrameworkCore;
|
|
using Microsoft.EntityFrameworkCore.Infrastructure;
|
|
using Microsoft.EntityFrameworkCore.Metadata;
|
|
|
|
namespace Pf2eModel;
|
|
|
|
/// <summary>
|
|
/// Класс, обеспецивающий общий способ для отчистки данных таблиц.
|
|
///
|
|
/// Источник: https://www.codeproject.com/Articles/5339402/Delete-All-Rows-in-Entity-Framework-Core-6
|
|
/// </summary>
|
|
public static class EfCleanHelper
|
|
{
|
|
/// <summary>
|
|
/// Определение полного названия таблицы по служебному классу.
|
|
/// </summary>
|
|
/// <param name="entityType">Служебный класс, определяющий таблицу базы данных.</param>
|
|
/// <param name="defaultSchemaName">Схема таблицы по умолчанию (<c>"public"</c> для PostgresSQL)</param>
|
|
/// <returns></returns>
|
|
private static string GetName(IEntityType entityType, string defaultSchemaName = "public")
|
|
{
|
|
object? schema = entityType.FindAnnotation("Relational:Schema")?.Value;
|
|
|
|
string tableName =
|
|
entityType.GetAnnotation("Relational:TableName")?.Value?.ToString()
|
|
?? throw new($"Не могу определить название таблицы для типа {entityType}");
|
|
|
|
string schemaName = schema?.ToString() ?? defaultSchemaName;
|
|
|
|
string name = $"\"{schemaName}\".\"{tableName}\"";
|
|
return name;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Определение полного названия таблицы по типу сущности.
|
|
/// </summary>
|
|
/// <typeparam name="T">Тип искомой сущности.</typeparam>
|
|
/// <param name="dbContext">Контекст, в котором производится поиск таблицы.</param>
|
|
/// <returns>Строковое представление названия таблицы.</returns>
|
|
public static string TableName<T>(DbContext dbContext)
|
|
where T : class
|
|
{
|
|
IEntityType entityType =
|
|
dbContext.Model.FindEntityType(typeof(T))
|
|
?? throw new($"Не могу определить название таблицы для класса {typeof(T)}");
|
|
|
|
return GetName(entityType);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Определение полного названия таблицы по типу сущности.
|
|
/// </summary>
|
|
/// <typeparam name="T">Тип искомой сущности.</typeparam>
|
|
/// <param name="dbSet">CLR-представление таблицы, название которой требуется определить.</param>
|
|
/// <returns>Строковое представление названия таблицы.</returns>
|
|
public static string TableName<T>(DbSet<T> dbSet)
|
|
where T : class
|
|
{
|
|
IEntityType entityType = dbSet.EntityType;
|
|
return GetName(entityType);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Отчистка содержимого таблицы.
|
|
/// </summary>
|
|
/// <typeparam name="T">Тип сущности, таблицу которой необходимо отчистить.</typeparam>
|
|
/// <param name="dbSet">CLR-представление таблицы, которую требуется отчистить.</param>
|
|
/// <returns>Строковое представление выполненной SQL-команды.</returns>
|
|
public static string Truncate<T>(this DbSet<T> dbSet)
|
|
where T : class
|
|
{
|
|
string cmd = $"TRUNCATE TABLE {TableName(dbSet)} CASCADE;";
|
|
DbContext context = dbSet.GetService<ICurrentDbContext>().Context;
|
|
context.Database.ExecuteSqlRaw(cmd);
|
|
return cmd;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Получение SQL-команды отчистки содержимого всех таблиц базы данных.
|
|
/// </summary>
|
|
/// <param name="context">Контекст базы данных, в которой необходимо отчистить все строки.</param>
|
|
/// <returns>Строковое представление выполненных SQL-команды.</returns>
|
|
public static string TruncateAll(this DbContext context)
|
|
{
|
|
IEnumerable<IEntityType> tables = context.Model.GetEntityTypes();
|
|
List<string> cmdList = [];
|
|
|
|
foreach (IEntityType entityType in tables)
|
|
{
|
|
string cmd = $"TRUNCATE TABLE {GetName(entityType)} CASCADE;";
|
|
context.Database.ExecuteSqlRaw(cmd);
|
|
cmdList.Add(cmd);
|
|
}
|
|
|
|
return string.Join('\n', cmdList);
|
|
}
|
|
}
|