如何在 Entity Framework 6 中使用 DbDataReader 获取表名？

Question

我的应用程序中有一个自定义 DbDataReader，它会覆盖 GetDateTime 方法来更改 DateTimeKind。

public class MyDbDataReader : BaseDbDataReader
{
    private string _dbName;

    public MyDbDataReader(string dbName, DbDataReader source) : base(source)
    {
        _dbName = dbName;
    }

    public override DateTime GetDateTime(int ordinal)
    {
        var tableName = base.GetSchemaTable().TableName; //this doesn't work
        return DateTime.SpecifyKind(base.GetDateTime(ordinal), base.GetName(ordinal).EndsWith("UTC", StringComparison.OrdinalIgnoreCase) ? DateTimeKind.Utc : DateTimeKind.Local);
    }
}

这是我的 BaseDbDataReader：

public abstract class BaseDbDataReader : DbDataReader
{
    readonly DbDataReader source;
    public BaseDbDataReader(DbDataReader source)
    {
        this.source = source;
    }

    ...
    public override DataTable GetSchemaTable() { return source.GetSchemaTable(); }
}

这个 dbReader 用在我的自定义拦截器中：

public class MyInterceptor : DbCommandInterceptor
{

    public override void ReaderExecuted(DbCommand command, DbCommandInterceptionContext<DbDataReader> interceptionContext)
    {           
        base.ReaderExecuted(command, interceptionContext);
        if (!command.Connection.Database.Equals("...") &&
            !(interceptionContext.Result is MyDbDataReader) &&
            interceptionContext.Result != null &&
            interceptionContext.Exception == null)
        {               
            interceptionContext.Result = new MyDbDataReader(command.Connection.Database, interceptionContext.Result);
        }
    }
}

我想要的只是在 GetDateTime 方法中获取 TableName。但是 GetSchemaTable 方法返回一个难以理解的结果，其中 TableName 属性等于“SchemaTable”。我在这里做错了什么以及如何获得正确的表名（如“用户”）。

注意：我不使用 SqlCommand 和 SqlCommand.ExecuteReader 来执行查询。我只是使用 dbSet。即dbContext.Users.Where(x => x.Id = 1).Single();

Answer 1

TableName 属性将始终返回 "SchemaTable"，因为它没有其他有意义的名称可以返回。

根据the link you found，每列的表名应在模式表的BaseTableName 列中返回。但是只有在指定了CommandBehavior.KeyInfo 标志时才会返回。

挖掘源代码，您似乎需要使用ReaderExecuting 方法并接管执行命令的责任才能做到这一点：

public override void ReaderExecuting(
    DbCommand command, 
    DbCommandInterceptionContext<DbDataReader> interceptionContext)
{
    // Skip processing for a specific database:
    if (command.Connection.Database.Equals("...")) return;
    
    var commandBehavior = interceptionContext.CommandBehavior | CommandBehavior.KeyInfo;
    
    try
    {
        var reader = command.ExecuteReader(commandBehavior);
        var result = new MyDbDataReader(command.Connection.Database, reader);
        interceptionContext.Result = result;
    }
    catch (Exception ex)
    {
        interceptionContext.Exception = ex;
    }
}

理论上，您应该能够从架构表中提取BaseTableName 列：

public override DateTime GetDateTime(int ordinal)
{
    var schemaTable = base.GetSchemaTable();
    var tableName = schemaTable.Rows[ordinal].Field<string>("BaseTableName");
    ...
}