如何检查当前操作是否在对象的范围内？

Question

类似于在 TransactionScope 范围内建立的数据库连接的相应行为，我希望能够在我的类中测试它是否在另一个类的实例中？

我的场景

我有一个类 (EntityBase)，所有数据库对象都基于该类构建，并且具有 AddToDatabase、UpdateInDatabase、DeleteFromDatabase 等方法。

我想创建一个 BulkLoader 类，它将以批量方式处理插入。 然后，这将有一个完整的方法，如 TransactionScope 类，这将是数据加载到数据库的点。

这将需要更改 EntityBase 类，以便在 BulkLoader 类的范围内实例化它并与该类交互时，它的行为会有所不同。

示例代码：

using (BulkLoader bulk = new BulkLoader(connection, transaction))
{
    foreach (IToDatabase assignment in assignmentsCollection)
    {
        assignment.WriteToDataBase(connection, transaction);
    }

    bulk.Complete();
}

class ClientAssignment : IToDatabase
{
     public int WriteToDataBase(IDbConnection connection, 
         IDbTransaction transaction)
     {
          foreach (EntityBase account in accountsCollection)
          {
                account.AddToDataBase(connection, transaction);
          }

          foreach (EntityBase person in personsCollection)
          {
                person.AddToDataBase(connection, transaction);
          }
     }
}

class EntityBase
{
    public virtual int AddToDatabase(IDbConnection connection, 
        IDbTransaction transaction)
    {
        // question relates to this bit
        if (inBulkLoaderClass)
        {
            // interact with bulk loader
        }
        else 
        {
            // do existing code
        }
    }
}

Answer 1

您可以使用 ThreadStaticAttribute：

public class BulkLoader : IDisposable
{
    [ThreadStatic]
    private static BulkLoader currentBulkLoader;

    public BulkLoader()
    {
        if (InBulkLoader)
        {
            throw new InvalidOperationException();
        }
        currentBulkLoader = this;
    }

    public void Dispose()
    {
        currentBulkLoader = null;
    }

    public static bool InBulkLoader
    {
         get { return currentBulkLoader != null; }
    }

    public static BulkLoader CurrentBulkLoader
    {
         get { return currentBulkLoader; }
    }
}

我一般不热衷于线程静态，但我认为在这种情况下可能没问题。 构造函数中的检查可确保您永远不会处于“嵌套”批量加载器中。 如果您希望能够嵌套它们，请保留一个计数器（在构造函数中递增，在 Dispose 调用中递减）。

Answer 2

好吧，您可以对环境对象（特别是线程静态字段）做一些事情，但这并不漂亮。 我认为如果逻辑不同，它首先应该是一个单独的方法。 另外，请注意，一种观点是实体对象无论如何都不应该与数据库对话——它们的工作是表示对象的状态； 存储库类（或类似类）的工作就是担心持久性。 那么你的存储库将有两种策略来应对事情。 也许存储库可以（可选）实现一个 IBulkRepository 接口，等等。

Answer 3

我同意马克的观点。 我们也采取了与马克几乎相同的方法。 我想这是解决此类问题的标准和最佳方法。 虽然我不太熟悉这里的线程静态业务，所以不能对此提出更多建议。