温馨提示×

C++弱引用在多线程环境下的使用

c++
小樊
81
2024-09-10 16:58:31
栏目: 编程语言

在C++中,弱引用(weak_ptr)是一种智能指针,它可以存储对对象的引用,但不会影响该对象的生命周期

以下是在多线程环境下使用弱引用的示例:

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

std::shared_ptr<int> sharedPtr;
std::weak_ptr<int> weakPtr;
std::mutex mtx;

void thread1() {
    std::unique_lock<std::mutex> lock(mtx);
    sharedPtr = std::make_shared<int>(42);
    weakPtr = sharedPtr;
    lock.unlock();
}

void thread2() {
    std::this_thread::sleep_for(std::chrono::milliseconds(50)); // 等待一段时间以确保thread1已经设置了weakPtr
    std::unique_lock<std::mutex> lock(mtx);
    if (auto sharedPtr2 = weakPtr.lock()) {
        std::cout << "Value: " << *sharedPtr2<< std::endl;
    } else {
        std::cout << "Object has been destroyed"<< std::endl;
    }
    lock.unlock();
}

int main() {
    std::thread t1(thread1);
    std::thread t2(thread2);
    t1.join();
    t2.join();
    return 0;
}

在这个示例中,我们创建了两个线程。第一个线程创建了一个shared_ptr并将其存储到全局变量sharedPtr中,然后将weak_ptr指向该对象。第二个线程尝试从weak_ptr中获取shared_ptr,如果成功,则输出对象的值;否则,输出“Object has been destroyed”。

注意,我们使用互斥锁(std::mutex)来确保在访问weak_ptr时不会发生数据竞争。这是因为weak_ptrlock()方法不是线程安全的,所以我们需要在访问它之前锁定互斥锁。

总之,在多线程环境下使用弱引用时,需要确保对弱引用的访问是线程安全的。这通常意味着使用互斥锁或其他同步原语来保护对弱引用的访问。

0