library/cpp/threading/thread_local/generic.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

#include "generic.h"

#include "thread_local.h"

namespace {
    class TThreadLocalStorage
        : public NThreading::IGenericLocalStorage
    {
    public:
        TData* GetData() const override {
            return Data_.Get();
        }
    private:
        NThreading::TThreadLocalValue<TData, NThreading::EThreadLocalImpl::StdThreadLocal> Data_;
    };

    std::atomic<size_t>& DefaultFactoryUsageCounter() {
        static std::atomic<size_t> v;
        return v;
    }

    auto& genericLocalStorageFactory() {
        static NThreading::TGenericLocalStorageFactory factory = [] {
            DefaultFactoryUsageCounter() += 1;
            return MakeHolder<TThreadLocalStorage>();
        };

        return factory;
    }
}

namespace NThreading {
    void SetGenericLocalStorageFactory(TGenericLocalStorageFactory factory) {
        Y_ENSURE(DefaultFactoryUsageCounter() == 0, "There are some thread local values allocated with default factory");

        genericLocalStorageFactory() = factory;
    }

    THolder<IGenericLocalStorage> MakeGenericLocalStorage() {
        return genericLocalStorageFactory()();
    }
}