线程局部单例

Question

我想创建一个单例类，在使用它的每个线程中实例化一次。 我想将实例指针存储在 TLS 槽中。 我提出了以下解决方案，但我不确定当涉及线程本地存储时，多线程访问单例工厂是否有任何特殊考虑。 也许还有更好的解决方案来实现线程本地单例。

class ThreadLocalSingleton 
{
    static DWORD tlsIndex;
public:
    static ThreadLocalSingleton *getInstance()
    {
        ThreadLocalSingleton *instance = 
            static_cast<ThreadLocalSingleton*>(TlsGetValue(tlsIndex));
        if (!instance) {
            instance = new ThreadLocalSingleton();
            TlsSetValue(tlsIndex, instance);
        }
        return instance;
    }
};
DWORD ThreadLocalSingleton::tlsIndex = TlsAlloc();

Tls* 函数当然是 win32 特定的，但可移植性并不是这里的主要问题。 您对其他平台的想法仍然很有价值。

主要编辑：我最初询问过在这种情况下使用双重检查锁定的问题。 然而，正如 DavidK 指出的那样，无论如何，单例都是在每个线程的基础上创建的。

剩下的两个问题是：

1. 是否适合回复 TlsGetValue/TlsSetValue 以确保每个线程获得一个实例，并且该实例只为每个线程创建一次？

2. 是否可以注册一个回调，允许我在特定线程结束时清理与该线程关联的实例？

   是否可以

Answer 1

我们使用一个存储线程 ID 到数据的映射的类来实现线程本地存储。 这似乎工作得很好，然后可以将此类的实例放置在需要线程本地存储的任何地方。 通常客户端使用 的实例作为静态私有字段。

这是代码的粗略轮廓，

template <class T>
struct ThreadLocal {
    T & value()
    {
        LockGuard<CriticalSection> lock(m_cs);

        std::map<int, T>::iterator itr = m_threadMap.find(Thread::getThreadID());

        if(itr != m_threadMap.end())
            return itr->second;

        return m_threadMap.insert(
            std::map<int, T>::value_type(BWThread::getThreadID(), T()))
                .first->second;
    }

    CriticalSection     m_cs;
    std::map<int, T>    m_threadMap;
};

然后将其用作

class A {
    // ...

    void doStuff();
private:
   static ThreadLocal<Foo> threadLocalFoo;
};

ThreadLocal<Foo> A::threadLocalFoo;

void A::doStuff() {
    // ...
    threadLocalFoo.value().bar();
    // ...
}

这很简单，并且可以在任何可以获得线程 ID 的平台上工作。 请注意，关键部分仅用于返回/创建引用，一旦获得引用，所有调用都将在关键部分之外。

Answer 2

请查看本文，了解为什么双重检查锁定不起作用一般（尽管它可能在特殊情况下起作用）。

Answer 3

既然你的对象是线程本地的，为什么你需要锁定来保护它们呢？ 每个调用 getInstance() 的线程都将独立于任何其他线程，那么为什么不只检查单例是否存在并在需要时创建它呢？ 仅当多个线程尝试访问同一个单例时才需要锁定，这在您的设计中是不可能的，如上所述。

编辑：继续讨论另外两个问题...我看不出有任何理由为什么使用 TlsAlloc/TlsGetValue 等不能按您的预期工作。 由于保存指向单例的指针的内存只能由相关线程访问，因此对其进行延迟初始化不会出现任何问题。 然而，没有明确的回调接口来清理它们。

显而易见的解决方案是拥有一个由所有线程主函数调用的方法，该方法可以清除创建的单例（如果有）。

如果线程很可能创建单例，则更简单的模式可能是在线程主函数的开头创建单例并在末尾删除它。 然后，您可以通过在堆栈上创建单例或将其保存在 std::auto_ptr<> 中来使用 RAII，以便在线程结束时将其删除。 （除非线程异常终止，但如果发生这种情况，所有的赌注都会消失，泄漏的对象是最不重要的问题。）然后，您可以只传递单例，或者将其存储在 TLS 中，或者将其存储在类，如果大多数线程功能都在一个类中。