c++怎么实现一个Copy-on-Write的智能指针_c++共享数据延迟复制机制

穿越時空

发布： 2025-11-15 21:52:37

原创

692人浏览过

答案：通过引用计数与延迟复制实现COW智能指针，允许多个对象共享数据并在写时深拷贝，优化资源使用。

c++怎么实现一个copy-on-write的智能指针_c++共享数据延迟复制机制

在C++中实现一个支持Copy-on-Write（写时复制）的智能指针，核心目标是允许多个对象共享同一份数据，直到某个对象尝试修改数据时才真正复制一份独立副本。这种机制常用于优化资源使用，比如早期的std::string实现或自定义的大对象管理。

基本设计思路

要实现Copy-on-Write智能指针，需要以下关键组件：

引用计数：跟踪有多少个指针共享同一块数据。
可变标记（写保护）：标识当前共享的数据是否正被多个所有者共享，从而决定是否需要复制。
延迟复制：只有在写操作发生且引用计数大于1时才进行深拷贝。

与std::shared_ptr不同的是，我们不希望每次写都影响其他共享者，因此需在非 const 操作前检查是否“唯一拥有”。

简易 Copy-on-Write 智能指针实现

下面是一个简化但完整的 COW 指针模板示例：

立即学习“C++免费学习笔记（深入）”；

// cow_ptr.h

逻辑智能

InsiderX：打造每个团队都能轻松定制的智能体员工

查看详情

pragma once

include <memory>

include <atomic>

template <typename T> class cow_ptr { private: struct control_block { std::shared_ptr<T> data; mutable std::atomic<int> ref_count{1}; // 共享控制块本身的引用数

    explicit control_block(T* ptr) : data(ptr) {}

    void inc() const { ref_count.fetch_add(1, std::memory_order_relaxed); }
    bool dec() const { return ref_count.fetch_sub(1, std::memory_order_acq_rel) == 1; }
};

mutable std::shared_ptr<control_block> cb;

void ensure_unique() {
    if (!cb->data.unique()) {
        cb->data = std::make_shared<T>(*cb->data); // 深拷贝
    }
}

登录后复制

public: explicit cow_ptr(T* ptr) : cb(std::make_shared<control_block>(ptr)) {}

cow_ptr(const cow_ptr& other) : cb(other.cb) {
    cb->inc();
}

cow_ptr& operator=(const cow_ptr& other) {
    if (this != &other) {
        cow_ptr temp(other);
        std::swap(cb, temp.cb);
    }
    return *this;
}

// 只读访问 - 不触发复制
const T& operator*() const { return *cb->data; }
const T* operator->() const { return cb->data.get(); }

// 写访问 - 触发写时复制
T& operator*() {
    ensure_unique();
    return *cb->data;
}
T* operator->() {
    ensure_unique();
    return cb->data.get();
}

// 检查是否唯一引用（可用于调试）
bool is_unique() const { return cb->data.unique(); }

登录后复制

};