写时拷贝 code

写时拷贝解决2个问题：
1.析构多次
2.修改一个对象会影响其他变量

设置引用计数后，当一个对象析构，引用计数-1，没有减到0就不释放空间，能够解决问题1.
当对字符串进行修改时，判断其引用计数是否大于1（大于1说明至少2个对象指向该串，贸然修改就会出现问题2），此时使用剥削手法，为要修改的对象分配同样的空间，另外操作，即可解决问题2.

class String
{
public:
	
	int& GetRefCount()//取到头部的4个字节，方便赋值
	{
		return (*(int*)(str - 4));
	}

	String(const char *_str="")//构造函数，str头部多开4个字节来存储引用计数，
		:str(new char [strlen(_str)+5])//5中  4个字节存引用计数，1个字节存'\0'
	{
		str += 4;
		strcpy(str, _str);
		GetRefCount() = 1;

	}

	String(const String &s)//拷贝构造，只需让新对象的指针指向被拷对象的str，并让引用计数+1即可
		:str(s.str)
	{
		GetRefCount()++;
	}

	String& operator=(const String& s)//赋值运算符重载，返回值为string&可以连等
	{
		if (str != s.str)//防止出现s1=s1的情况
		{
			if (--GetRefCount() == 0)
			{
				delete[](str - 4);
			}
			str = s.str;
			GetRefCount()++;
		}
		return *this;

	}

	
	void copy_on_write()//判断修改对象字符串时会不会影响到其他对象
	{
		if (GetRefCount() > 1)//如果有不止1个对象指向该字符串，那么就要另外开辟空间
		{
			String tmp(str);//剥削手段
			swap(str, tmp.str);
			GetRefCount() = 1;
		}//对象tmp析构，不会影响s1对象
	}
		
	char* c_str()
	{
		return str;
	}
	
	char& operator[](size_t pos)
	{
		copy_on_write();
		return str[pos];
	}

	~String()
	{
		if (--GetRefCount() == 0)//说明当前析构的对象是最后一个指向该串的对象
		{
			delete[](str - 4);
		}
	}
private:
	char *str;
};
int main()
{
	String s1("hello");
	String s2(s1);
	s2[0] = 'x';
	String s3("world");
	s3 = s1;


	return 0;
}