在Entity Framework Core 5中删除表中的所有行

寒冰屋

已于 2022-05-29 14:58:03 修改

阅读量750

点赞数

分类专栏： ASP.NET CORE 文章标签： ef core 5

于 2022-05-29 14:52:56 首次发布

原文链接：https://www.codeproject.com/Tips/5320939/Delete-All-Rows-in-a-Table-in-Entity-Framework-Cor

版权

ASP.NET CORE 专栏收录该内容

1048 篇文章 53 订阅

订阅专栏

Delete/Truncate/Clear

背景

有时我们可能需要使用Entity Framework Core从表中删除所有记录。一种常用的方法是迭代每一行并使用DBSet.Remove()。

foreach (var item in Db.Users)
{
    Db.Users.Remove(item);
}
Db.SaveChanges();

在这篇文章中，我们将探索一些替代方案。

扩展助手

先决条件

需要安装NuGet包 Microsoft.EntityFrameworkCore.Relational

install Microsoft.EntityFrameworkCore.Relational

表名

获取映射实体的表名和模式的扩展方法。我们将使用这个名称和模式来创建一个Truncate/Delete脚本。

using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.EntityFrameworkCore.Metadata;

public class AnnotationHelper
{
    private static string GetName(IEntityType entityType, string defaultSchemaName = "dbo")
    {
        /*3.0.1 these were working*/
        //var schemaName = entityType.GetSchema();
        //var tableName = entityType.GetTableName();
        
        var schema = entityType.FindAnnotation("Relational:Schema").Value;
        string tableName = entityType.GetAnnotation("Relational:TableName").Value.ToString();
        string schemaName = schema == null ? defaultSchemaName : schema.ToString();
        string name = string.Format("[{0}].[{1}]", schemaName, tableName);
        return name;
    }

    public static string TableName<T>(DbContext dbContext) where T : class
    {
        var entityType = dbContext.Model.FindEntityType(typeof(T));
        return GetName(entityType);
    }

    public static string TableName<T>(DbSet<T> dbSet) where T : class
    {
        var entityType = dbSet.EntityType;
        return GetName(entityType);
    }
}

Delete/Truncate/Clear

有 DbSet<T> 对象的扩展方法

string Truncate<T>(this DbSet<T> dbSet)使用截断查询截断表
string Delete<T>(this DbSet<T> dbSet)使用删除查询删除表的所有行
void Clear<T>(this DbSet<T> dbSet)使用RemoveRange方法删除表的所有行

using Microsoft.EntityFrameworkCore;
using Microsoft.EntityFrameworkCore.Infrastructure;
using System.Data;
using System.Linq;

public static class EfHelper
{
    public static string Truncate<T>(this DbSet<T> dbSet) where T : class
    {
        string cmd = $"TRUNCATE TABLE {AnnotationHelper.TableName(dbSet)}";
        var context = dbSet.GetService<ICurrentDbContext>().Context;
        context.Database.ExecuteSqlRaw(cmd);
        return cmd;
    }

    public static string Delete<T>(this DbSet<T> dbSet) where T : class
    {
        string cmd = $"DELETE FROM {AnnotationHelper.TableName(dbSet)}";
        var context = dbSet.GetService<ICurrentDbContext>().Context;
        context.Database.ExecuteSqlRaw(cmd);
        return cmd;
    }

    public static void Clear<T>(this DbSet<T> dbSet) where T : class
    {
        if (dbSet.Any())
        {
            dbSet.RemoveRange(dbSet.ToList());
        }
    }
}

使用代码

数据库上下文

我们当前的数据库上下文类。

public class AppDbContext : DbContext
{
    public DbSet<User> Users { get; set; }
}

删除

Db.Users.Delete();

与数据库事务一起使用

string errorMsg = "Error";
User newUser = null;
Exception exception = null;
using (var tran = Db.Database.BeginTransaction())
{
    try
    {
        newUser = new User()
        {
            Name = "Name",
            Email = "Email",
            CreatedBy = "CreatedBy",
            CreatedOn = DateTime.Now
        };
        Db.Users.Add(newUser);
        Db.SaveChanges();

        Db.Users.Delete();

        throw new Exception(errorMsg);

        tran.Commit();
    }
    catch (Exception ex)
    {
        exception = ex;
        tran.Rollback();
    }
}