Composite模式,将对象组合成树形结构以表示“部分-整体”的层次结构。Composite使得用户对单个对象和组合对象的使用具有一致性。
// component
class FileComponent {
public:
virtual ~FileComponent();
virtual int GetSize();
virtual void Display();
virtual FileComponent* GetComposite() { return this; }
virtual void Add(FileComponent*);
virtual void Remove(FileComponent*);
virtual Iterator<FileComponent*>* CreateIterator();
protected:
FileComponent(const char* name);
private:
const char* name;
};
// composite
class FolderComposite: public FileComponent {
public:
FolderComposite(const char* name);
virtual ~FolderComposite();
virtual int GetSize();
virtual void Display();
virtual void Add(FileComponent*);
virtual void Remove(FileComponent*);
virtual List<FileComponent*>* GetFileList();
private:
List<FileComponent*> fileList;
};
int FolderComposite::GetSize() {
List<FileComponent*>* itr = fileList->begin();
int totalSize = 0;
for (; itr != fileList->end(); itr++) {
totalSize += (*itr)->GetSize();
}
return totalSize;
}
void Display() {
printf("d %s %d\n", name, GetSize());
List<FileComponent*>* itr = fileList->begin();
for (; itr != fileList->end(); itr++) {
(*itr)->Display();
}
}
// leaf
class DocumentLeaf: public FileComponent {
public:
DocumentLeaf(const char* name);
virtual ~DocumentLeaf();
virtual int GetSize();
virtual void Display();
virtual FileComponent* GetComposite() { return 0; }
};
void DocumentLeaf::Display() {
printf("- %s %d\n", name, GetSize());
}
// test
void Test() {
FolderComposite* folder1 = new FolderComposite("folder1");
FolderComposite* folder2 = new FolderComposite("folder2");
DocumentLeaf* doc1 = new DocumentLeaf("doc1");
DocumentLeaf* doc2 = new DocumentLeaf("doc2");
DocumentLeaf* doc3 = new DocumentLeaf("doc3");
folder1->Add(doc1);
folder1->Add(doc2);
folder2->Add(doc3);
FolderComposite* rootFolder = new FolderComposite("root");
rootFolder->Add(folder1);
rootFolder->Add(folder2);
rootFolder->Display();
// todo ... destroy
}
PS. 我的设计模式系列blog,《设计模式》专栏,通过简单的示例演示设计模式,对于初学者很容易理解入门。深入学习请看GoF的《设计模式》。