Блог

Я пытаюсь создать класс с потокобезопасным shared_ptr. Мой вариант использования заключается в том, что shared_ptr принадлежит объекту класса и ведет себя как синглтон (функция CreateIfNotExist может быть запущена любым потоком в любой момент времени).

По сути, если указатель равен нулю, выигрывает первый поток, который задает его значение, а все остальные потоки, которые создают его одновременно, используют значение победившего потока.

Вот что у меня есть на данный момент (обратите внимание, что единственной рассматриваемой функцией является функция CreateIfNotExist(), остальное предназначено для тестирования):

 #include <memory>
#include <iostream>
#include <thread>
#include <vector>
#include <mutex>

struct A {
    A(int a) : x(a) {}
    int x;
};

struct B {
    B() : test(nullptr) {}

    void CreateIfNotExist(int val) {
        std::shared_ptr<A> newPtr = std::make_shared<A>(val);
        std::shared_ptr<A> _null = nullptr;
        std::atomic_compare_exchange_strong(amp;test, amp;_null, newPtr);
    }

    std::shared_ptr<A> test;
};

int gRet = -1;
std::mutex m;

void Func(B* b, int val) {
    b->CreateIfNotExist(val);
    int ret =  b->test->x;

    if(gRet == -1) {
        std::unique_lock<std::mutex> l(m);
        if(gRet == -1) {
            gRet = ret;
        }
    }

    if(ret != gRet) {
        std::cout << " FAILED " << std::endl;
    }
}

int main() {
    B b;

    std::vector<std::thread> threads;
    for(int i = 0; i < 10000;   i) {
        threads.clear();
        for(int i = 0; i < 8;   i) threads.emplace_back(amp;Func, amp;b, i);
        for(int i = 0; i < 8;   i) threads[i].join();
    }
}
  

Это правильный способ сделать это? Есть ли лучший способ гарантировать, что все потоки, вызывающие CreateIfNotExist() одновременно, используют один и тот же shared_ptr?

Возможно, что-то в этом роде:

 struct B {
  void CreateIfNotExist(int val) {
    std::call_once(test_init,
                   [this, val](){test = std::make_shared<A>(val);});
  }

  std::shared_ptr<A> test;
  std::once_flag test_init;
};