C++primer 练习13.36

#pragma once
#include<string>
#include<set>
using namespace std;

class Message
{
friend class Folder;
friend void swap(Message &lhs, Message &rhs);
public:
explicit Message(const string &str=""):contents(str){}
Message(const Message&);
Message& operator=(const Message);//采用非引用类型，实现自赋值情况出现时，也能成功，但多了一次拷贝构造
~Message();
void save(Folder &);
void remove(Folder &);
private:
string contents;
set<Folder*> folders;
void add_to_Folders(const Message&);
void remove_from_Folders();
};

class Folder
{
public:
Folder(const Folder&);//拷贝构造函数
Folder& operator=(const Folder);//拷贝赋值运算符，采用非引用类型，实现出现自赋值情况时，也能成功，但多了一次拷贝构造
~Folder();                //析构函数
void addMsg(Message *);   //添加message
void remMsg(Message *);   //去除message
private:
set<Message*> Mes;
void addAllMsg();        //拷贝构造函数和拷贝赋值预算符的辅助函数
void removeAllMsg();     //拷贝赋值运算符和析构函数的辅助函数
};

void Folder::addMsg(Message *m)
{
if(Mes.find(m)==Mes.end())//防止在调用m->save(*this)时的从复插入（虽然重复插入对于set来说没什么关系）
Mes.insert(m);
if (m->folders.find(this) == m->folders.end())//给递归调用一个结束条件，防止无限调用
m->save(*this);
}

void Folder::remMsg(Message *m)
{
if (Mes.find(m) != Mes.end())//防止在调用m->remove(*this)时的从复删除（虽然重复删除对于set来说没什么关系）
Mes.erase(m);
if (m->folders.find(this) != m->folders.end())//给递归调用一个结束条件，防止无限调用
m->remove(*this);
}

void Folder::addAllMsg()        //拷贝构造函数和拷贝赋值预算符的辅助函数
{
for (auto &m : Mes)         //对Mes集合中的每一个指针对象都调用save()函数
m->save(*this);
}

Folder::Folder(const Folder &f)
{
Mes = f.Mes;              //将f中数据成员拷贝过来
addAllMsg();              //调用addALLMsg将这个Folder添加到所有对应的message中去
}

void Folder::removeAllMsg()   //拷贝赋值运算符和析构函数的辅助函数
{
for (auto &m : Mes)       //对Mes集合中的每一个指针对象都调用remover()函数
m->remove(*this);
}

Folder& Folder::operator=(const Folder f)//采用非引用类型是为了保护出现自赋值的情况时，也能正常运行
{
removeAllMsg();          //将数据成员中所包含的所有message都调用remove()函数，实现删除这个文件夹的作用
Mes = f.Mes;             //将新的数据成员拷贝过来
addAllMsg();             //将数据成员中所包含的所有message都调用remover()函数，完成为这个文件夹赋值右侧文件夹的步骤
return *this;
}

Folder::~Folder()
{
removeAllMsg();         //将数据成员中所包含的所有message都调用remove()函数，实现删除这个文件夹的作用
}

void Message::save(Folder &f)
{
folders.insert(&f);
f.addMsg(this);
}

void Message::remove(Folder &f)
{
folders.erase(&f);
f.remMsg(this);
}

void Message::add_to_Folders(const Message &m)
{
for (auto &f : m.folders)
f->addMsg(this);
}

Message::Message(const Message &m) :contents(m.contents), folders(m.folders)
{
add_to_Folders(m);
}

void Message::remove_from_Folders()
{
for (auto f : folders)
f->remMsg(this);
}

Message::~Message()
{
remove_from_Folders();
}

Message& Message::operator=(const Message rhs)
{
remove_from_Folders();
contents = rhs.contents;
folders = rhs.folders;
add_to_Folders(rhs);
return *this;
}

void swap(Message &lhs, Message &rhs)
{
set<Folder*> lfolders = lhs.folders;
set<Folder*> rfolders = rhs.folders;
for (auto f : rhs.folders)
f->remMsg(&rhs);
for (auto f : lhs.folders)
f->remMsg(&rhs);
lhs.folders = rhs.folders;
rhs.folders = lhs.folders;
swap(lhs.contents, rhs.contents);
for (auto f : lhs.folders)
f->addMsg(&lhs);
for (auto f : rhs.folders)
f->addMsg(&rhs);
}