C++引用计数原理以及简易版引用技数String的实现

1. 引用计数的作用（为什么使用引用计数）

引用计数是一项允许多个等值对象共享同一实值的技术。

此技术一般由下面两个方向驱动：

a. 引用计数可以简化堆对象的薄记工作。一旦某个对象由new 分配出来，则记录该对象的所有者是一件重要的事情，只有该对象的所有者有权删除该对象。引用计数可以消除记录对象所有权的负荷，因为当对象运用了引用计数，他便拥有自己的所有权。

b. 引用计数可以使许多等值对象共享同一个实值。如果许多对象拥有相同的值，那么存储多次是一种资源浪费。如果能让这些对象共享同一实值，不仅可以节省内存，也可以提升程序的速度，因为不需要构造和析构同值对象的多余副本。

2. 一个非引用计数的String的简易实现

class RcString {
public:
    RcString(const char* value = "");
    RcString(const RcString& rhs);
    ~RcString();
    RcString& operator=(const RcString& rhs);

    char* get() {
        return _data;
    }

private:
    char* _data;
};

增加get方法主要是为了测试。

RcString::RcString(const char *value) {
    _data = new char[strlen(value) + 1];
    strcpy(_data, value);
}

RcString::RcString(const RcString &rhs) {
    _data = new char[strlen(rhs._data) + 1];
    strcpy(_data, rhs._data);
}

RcString & RcString::operator=(const RcString &rhs) {
    if (this == &rhs) return *this;
    delete [] _data;
    _data = new char[strlen(rhs._data) + 1];
    strcpy(_data, rhs._data);
    return *this;
}

RcString::~RcString() {
    delete [] _data;
}

3.  引用计数String的实现（怎样实现引用计数String)

我们需要的是为每个共享值提供一个引用计数，而不是为了每个对象提供一个引用计数，因此引用计数变量应该与共享值耦合。其引用计数的String实现如下：

class RcString {
public:
    RcString(const char* value = "");
    RcString(const RcString& rhs);
    ~RcString();
    RcString& operator=(const RcString& rhs);

    char* get() {
        return _data->_data_ref;
    }

private:
    struct StringValue {
        StringValue(const char* value);
        ~StringValue();
        char* _data_ref;
        int _refcount;
    };
    StringValue* _data;
};

RcString::RcString(const char *value) : _data(new StringValue(value)){

}

RcString::RcString(const RcString &rhs) : _data(rhs._data) {
    ++_data->_refcount;
}

RcString & RcString::operator=(const RcString &rhs) {
    if (_data == rhs._data) return *this;

    if (--_data->_refcount == 0) delete _data;
    _data = rhs._data;
    ++_data->_refcount;
    return *this;
}

RcString::~RcString() {
    if (--_data->_refcount == 0) {
        delete _data;
    }
}

RcString::StringValue::StringValue(const char *value) : _refcount(1) {
    _data_ref = new char[strlen(value) + 1];
    strcpy(_data_ref, value);
}

RcString::StringValue::~StringValue() {
    delete [] _data_ref;
}

 3.1 引用计数String的写时复制（COW）

上述引用计数String未能重载[]操作符，因此为了完善该引用计数String，增加[]操作符，其实现如下：

    const char& operator[](int index) const {
        return _data->_data_ref[index];
    }

    char& operator[](int index) {
        // TODO
    }

针对const char& 的方法 只能表示只读属性，所以实现较为简单，直接返回相应值即可。下面一起分析下char& operator[]方法存在的问题及解决方案。

存在的问题：由于char& operator[]方法返回的是char& 类型，因此可对某个string进行写操作，但该写操作会传递到所有共享StringValue值的对象，这便于设计意图不符。

那么如何解决这个问题呢？ 能否靠编译器自动识别我们是正在写一个String？答案是否定的，编译器无法识别一个调用是const 还是非const。

那么我们可以采取一个不那么友好的方法：将char& operator[]方法均标记为写，因此我们可以在该方法内为某个String 生成一份StringValue的副本，仅仅返回该副本即可，这样便不会影响共享该值的其他String对象。其具体实现代码如下：

char & RcString::operator[](int index) {
    if (_data->_refcount > 1) {
        --_data->_refcount;
        _data = new StringValue(_data->_data_ref);
    }
    
    return _data->_data_ref[index];
}

参考： more effective C++