13章 Message和Folder的实现

#include<iostream>
#include<algorithm>
#include<memory>//the hearder file of shared_ptr
#include <vector>
#include <list>
#include <map>
#include <string>
#include <set>

using namespace std;

class Message;
class Folder {
    friend class Message;
    friend void print(const Folder&);
public:
    Folder(){}
    Folder(const Folder&);//拷贝构造函数需要把指向自己的指针放进Message维护的set中
    Folder& operator=(const Folder&);//做析构函数和拷贝构造函数的工作
    ~Folder();//需要将自己所拥有的mes中的this指针删除

private:
    set<Message*> mes;

    void add_all_mes(const Folder&);
    void rev_all_mes();//

    void addmes(Message *const);
    void revmes(Message *const);//单向操作而已
};


class Message {
    friend void print(const Folder&);
    friend class Folder;
public:
    Message(const std::string &t = "") :content(t) {}
    Message(const Message&);
    Message& operator=(const Message&);
    ~Message();

    void save(Folder&);
    void remove(Folder&);//非常量引用

private:
    std::string content;
    std::set<Folder*> folders;
    //下面两个是单向操作
    void add_to_floders(const Message&);
    void rev_from_folders();
};

void Folder::add_all_mes(const Folder&f){//构造函数调用
    //mes = f.mes;
    for (const auto&pm : f.mes)pm->folders.insert(this);//this 和 f的区别
}

void Folder::rev_all_mes() {//析构函数调用，只需要管好别人的事，自己的东西委托析构函数处理
    for (const auto&pm : mes)pm->folders.erase(this);
}

void Folder::addmes(Message*const pm) {
    mes.insert(pm);
}

void Folder::revmes(Message*const pm) {//注意const应该修饰指针，即应该是顶层const
    mes.erase(pm);
}

Folder::Folder(const Folder& rhs) :mes(rhs.mes) {
    add_all_mes(rhs);
}

Folder::~Folder() {
    rev_all_mes();
}

Folder& Folder::operator=(const Folder& rhs) {
    rev_all_mes();

    mes = rhs.mes;
    add_all_mes(rhs);
    return *this;
}



//与Folder类不同，下面的两个函数是双向操作，即同时更新Folder和Message两个类，且调用Folder的私有函数
void Message::save(Folder&f) {
    folders.insert(&f);
    f.addmes(this);
}

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

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

void Message::rev_from_folders() {
    for (auto &f : folders)f->revmes(this);
}

Message::Message(const Message&rhs) :content(rhs.content), folders(rhs.folders) {
    add_to_floders(rhs);//传入的虽然是引用的参数，压入对应的folders事实上是this指针
}

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

Message& Message::operator=(const Message&rhs) {
    rev_from_folders();//先移除，因为底层用set实现，插入相同元素会被忽略，考虑到自赋值,应该先remove一波再add
    add_to_floders(rhs);

    content = rhs.content;
    folders = rhs.folders;

    return *this;
}

void print(const Folder& f) {
    if (f.mes.empty()) { cout << "empty!" << endl;return; }
    cout << (*(f.mes.begin()))->content << endl;
}

int f() { return 0; }
int main(void) {
    Folder f;
    auto p = new Message("xixi");
    p->save(f);
    print(f);
    delete(p);
    print(f);

    cout << endl;
    return 0;
}

大概是第一次写有点逻辑的C++，写了两个小时--过程略刺激