中介者模式(Mediator)

#include <iostream>
#include <string>
using namespace std;

class Colleague;

class Mediator {
public:
    virtual void send(string message, Colleague *pColleague) = 0;
};

class Colleague {
public:
    Colleague(Mediator *pMediator) { m_pMediator = pMediator; }
    void notify(string message) { cout << "Got message : " << message << endl; }
protected:
    Mediator *m_pMediator;
};

class ConcreteColleague1 : public Colleague {
public:
    ConcreteColleague1(Mediator *pMediator) : Colleague(pMediator) {}
    void send(string message) { m_pMediator->send(message, this); }
    void notify(string message) { cout << "Colleague1 got message : " << message << endl; }
};

class ConcreteColleague2 : public Colleague {
public:
    ConcreteColleague2(Mediator *pMediator) : Colleague(pMediator) {}
    void send(string message) { m_pMediator->send(message, this); }
    void notify(string message) { cout << "Colleague2 got message : " << message << endl; }
};

class ConcreteMediator : public Mediator {
public:
    void setColleague1(ConcreteColleague1 *pColleague1) {
        m_pColleague1 = pColleague1;
    }
    void setColleague2(ConcreteColleague2 *pColleague2) {
        m_pColleague2 = pColleague2;
    }
    void send(string message, Colleague *pColleague) {
        if(pColleague == m_pColleague1) {
            m_pColleague2->notify(message);
        }
        else {
            m_pColleague1->notify(message);
        }
    }

private:
    ConcreteColleague1 *m_pColleague1;
    ConcreteColleague2 *m_pColleague2;
};

int main() {
    ConcreteMediator *pcm = new ConcreteMediator();;
    ConcreteColleague1 *pc1 = new ConcreteColleague1(pcm);
    ConcreteColleague2 *pc2 = new ConcreteColleague2(pcm);

    pcm->setColleague1(pc1);
    pcm->setColleague2(pc2);

    pc1->send("message from pc1");
    
    return 0;
}