使用EF Core 2.1检查表是否存在

时间:2018-11-26 01:39:45

标签: c# entity-framework entity-framework-core

在Entity Framework中,可以通过以下方式检查表的存在:

bool exists = context.Database
                 .SqlQuery<int?>(@"
                     SELECT 1 FROM sys.tables AS T
                     INNER JOIN sys.schemas AS S ON T.schema_id = S.schema_id
                     WHERE S.Name = 'SchemaName' AND T.Name = 'TableName'")
                 .SingleOrDefault() != null;

我正在使用 EF Core 2.1 ,并且方法SqlQuery不存在。

检查表是否存在的正确方法是什么?理想情况下,无需尝试访问该表并假设如果引发异常该表就不存在。

编辑:我的最终实现

public bool TableExists(string tableName)
{
    return TableExists("dbo", tableName);
}

public bool TableExists(string schema, string tableName)
{
    var connection = Context.Database.GetDbConnection();

    if (connection.State.Equals(ConnectionState.Closed))
        connection.Open();

    using (var command = connection.CreateCommand())
    {
        command.CommandText = @"
            SELECT 1 FROM INFORMATION_SCHEMA.TABLES 
            WHERE TABLE_SCHEMA = @Schema
            AND TABLE_NAME = @TableName";

        var schemaParam = command.CreateParameter();
        schemaParam.ParameterName = "@Schema";
        schemaParam.Value = schema;
        command.Parameters.Add(schemaParam);

        var tableNameParam = command.CreateParameter();
        tableNameParam.ParameterName = "@TableName";
        tableNameParam.Value = tableName;
        command.Parameters.Add(tableNameParam);

        return command.ExecuteScalar() != null;
    }
}

1 个答案:

答案 0 :(得分:6)

ExecuteSqlCommand

context.Database.ExecuteSqlCommand("...")

但是,仅限于返回一个整数,该整数指示受影响的行数。如果您执行的是SELECT,则行不会受到影响,因此它实际上并不能满足您的需求。

也有FromSql,但仅适用于表,不适用于数据库级别:

context.TableName.FromSql("SELECT ...")

对于您正在做的事情,更好的选择是从EF获取DbConnection,然后创建自己的DbCommand。这将返回您期望的结果:

var conn = context.Database.GetDbConnection();
if (conn.State.Equals(ConnectionState.Closed)) await conn.OpenAsync();
using (var command = conn.CreateCommand()) {
    command.CommandText = @"
    SELECT 1 FROM sys.tables AS T
        INNER JOIN sys.schemas AS S ON T.schema_id = S.schema_id
    WHERE S.Name = 'SchemaName' AND T.Name = 'TableName'";
    var exists = await command.ExecuteScalarAsync() != null;
}

这里有个警告:不要将DbConnection中的GetDbConnection()放在using语句中,也不要在完成后将其关闭。该连接用于该DbContext实例的生命周期。因此,如果关闭它,EF以后发出的任何请求都将失败。它也可能在您的代码之前已打开,这就是为什么我在调用OpenAsync()之前在其中进行测试以查看它是否已关闭的原因。