设计模式之组合模式-树形结构的处理

一、描述：

     对于树形结构，当容器对象（如文件夹）的某一个方法被调用时，将遍历整个树形结构，寻找也包含这个方法的成员对象（可以是容器对象，也可以是叶子对象）并调用执行，牵一而动百，其中使用了递归调用的机制来对整个结构进行处理。由于容器对象和叶子对象在功能上的区别，在使用这些对象的代码中必须有区别地对待容器对象和叶子对象，而实际上大多数情况下我们希望一致地处理它们，因为对于这些对象的区别对待将会使得程序非常复杂。组合模式为解决此类问题而诞生，它可以让叶子对象和容器对象的使用具有一致性

二、定义：

    组合多个对象形成树形结构以表示具有“整体—部分”关系的层次结构。组合模式对单个对象（即叶子对象）和组合对象（即容器对象）的使用具有一致性，组合模式又可以称为“整体—部分”(Part-Whole)模式，它是一种对象结构型模式。

    在组合模式中引入了抽象构件类Component，它是所有容器类和叶子类的公共父类，客户端针对Component进行编程。组合模式结构如图：

在组合模式结构图中包含如下几个角色：
● Component（抽象构件）：它可以是接口或抽象类，为叶子构件和容器构件对象声明接口，在该角色中可以包含所有子类共有行为的声明和实现。在抽象构件中定义了访问及管理它的子构件的方法，如增加子构件、删除子构件、获取子构件等。
● Leaf（叶子构件）：它在组合结构中表示叶子节点对象，叶子节点没有子节点，它实现了在抽象构件中定义的行为。对于那些访问及管理子构件的方法，可以通过异常等方式进行处理。
● Composite（容器构件）：它在组合结构中表示容器节点对象，容器节点包含子节点，其子节点可以是叶子节点，也可以是容器节点，它提供一个集合用于存储子节点，实现了在抽象构件中定义的行为，包括那些访问及管理子构件的方法，在其业务方法中可以递归调用其子节点的业务方法。
     组合模式的关键是定义了一个抽象构件类，它既可以代表叶子，又可以代表容器，而客户端针对该抽象构件类进行编程，无须知道它到底表示的是叶子还是容器，可以对其进行统一处理。同时容器对象与抽象构件类之间还建立一个聚合关联关系，在容器对象中既可以包含叶子，也可以包含容器，以此实现递归组合，形成一个树形结构。

    如果不使用组合模式，客户端代码将过多地依赖于容器对象复杂的内部实现结构，容器对象内部实现结构的变化将引起客户代码的频繁变化，带来了代码维护复杂、可扩展性差等弊端。组合模式的引入将在一定程度上解决这些问题。、

三、代码示例：

下面通过简单的示例代码来分析组合模式的各个角色的用途和实现。对于组合模式中的抽象构件角色，其典型代码如下所示：

    abstract class Component {          public abstract void add(Component c); //增加成员          public abstract void remove(Component c); //删除成员          public abstract Component getChild(int i); //获取成员          public abstract void operation();  //业务方法      }  

一般将抽象构件类设计为接口或抽象类，将所有子类共有方法的声明和实现放在抽象构件类中。对于客户端而言，将针对抽象构件编程，而无须关心其具体子类是容器构件还是叶子构件。

如果继承抽象构件的是叶子构件，则其典型代码如下所示： class Leaf extends Component {

public void add(Component c) {//异常处理或错误提示}public void remove(Component c) {//异常处理或错误提示}public Component getChild(int i) {//异常处理或错误提示return null;}public void operation() {//叶子构件具体业务方法的实现}} 

作为抽象构件类的子类，在叶子构件中需要实现在抽象构件类中声明的所有方法，包括业务方法以及管理和访问子构件的方法，但是叶子构件不能再包含子构件，因此在叶子构件中实现子构件管理和访问方法时需要提供异常处理或错误提示。当然，这无疑会给叶子构件的实现带来麻烦。

如果继承抽象构件的是容器构件，则其典型代码如下所示：

    class Composite：Component {          private ArrayList<Component> list = new ArrayList<Component>();                    public void add(Component c) {              list.add(c);          }                    public void remove(Component c) {              list.remove(c);          }                    public Component getChild(int i) {              return (Component)list.get(i);          }                    public void operation() {              //容器构件具体业务方法的实现              //递归调用成员构件的业务方法              for(Object obj in list) {                  ((Component)obj).operation();              }          }         }  

     在容器构件中实现了在抽象构件中声明的所有方法，既包括业务方法，也包括用于访问和管理成员子构件的方法，如add()、remove()和getChild()等方法。需要注意的是在实现具体业务方法时，由于容器构件充当的是容器角色，包含成员构件，因此它将调用其成员构件的业务方法。在组合模式结构中，由于容器构件中仍然可以包含容器构件，因此在对容器构件进行处理时需要使用递归算法，即在容器构件的operation()方法中递归调用其成员构件的operation()方法。

四、案例：

    某软件公司欲开发一个杀毒(AntiVirus)软件，该软件既可以对某个文件夹(Folder)杀毒，也可以对某个指定的文件(File)进行杀毒。该杀毒软件还可以根据各类文件的特点，为不同类型的文件提供不同的杀毒方式，例如图像文件(ImageFile)和文本文件(TextFile)的杀毒方式就有所差异。现需要提供该杀毒软件的整体框架设计方案：

     我们可以看出，在图中包含文件（灰色节点）和文件夹（白色节点）两类不同的元素，其中在文件夹中可以包含文件，还可以继续包含子文件夹，但是在文件中不能再包含子文件或者子文件夹。在此，我们可以称文件夹为容器(Container)，而不同类型的各种文件是其成员，也称为叶子(Leaf)，一个文件夹也可以作为另一个更大的文件夹的成员。如果我们现在要对某一个文件夹进行操作，如查找文件，那么需要对指定的文件夹进行遍历，如果存在子文件夹则打开其子文件夹继续遍历，如果是文件则判断之后返回查找结果。

    为了让系统具有更好的灵活性和可扩展性，客户端可以一致地对待文件和文件夹，公司开发人员使用组合模式来进行杀毒软件的框架设计，其基本结构如图所示：

     AbstractFile充当抽象构件类，Folder充当容器构件类，ImageFile、TextFile和VideoFile充当叶子构件类。完整代码如下所示：

 class Program    {        static void Main(string[] args)        {            AbstractFile file1, file2, file3, file4, folder1, folder2, folder3;            folder1 = new Folder("资料文件夹");            folder2 = new Folder("图像文件夹");            folder3 = new Folder("文本文件夹");            file1 = new ImageFile("小龙女.jpg");            file2 = new ImageFile("张无忌.jpg");            file3 = new TextFile("九阴真经.txt");            file4 = new TextFile("葵花宝典.doc");            folder2.Add(file1);            folder2.Add(file2);            folder3.Add(file3);            folder3.Add(file4);            folder1.Add(folder2);            folder1.Add(folder3);            //从“资料文件夹”节点开始进行杀毒操作              folder1.KillVirus();            Console.ReadLine();        }    }    //抽象文件类：抽象构件      public abstract class AbstractFile    {        public abstract void Add(AbstractFile file);        public abstract void Remove(AbstractFile file);        public abstract AbstractFile GetChild(int i);        public abstract void KillVirus();    }    //图像文件类：叶子构件      public class ImageFile : AbstractFile    {        private string name;        public ImageFile(string name)        {            this.name = name;        }        public override void Add(AbstractFile file)        {            Console.WriteLine("对不起，不支持该方法");        }        public override void Remove(AbstractFile file)        {            Console.WriteLine("对不起，不支持该方法");        }        public override AbstractFile GetChild(int i)        {            Console.WriteLine("对不起，不支持该方法");            return null;        }        //模拟杀毒          public override void KillVirus()        {            Console.WriteLine("----对图像文件'" + name + "'进行杀毒");        }    }    //文本文件类：叶子构件      public class TextFile : AbstractFile    {        private string name;        public TextFile(string name)        {            this.name = name;        }        public override void Add(AbstractFile file)        {            Console.WriteLine("对不起，不支持该方法");        }        public override void Remove(AbstractFile file)        {            Console.WriteLine("对不起，不支持该方法");        }        public override AbstractFile GetChild(int i)        {            Console.WriteLine("对不起，不支持该方法");            return null;        }        //模拟杀毒          public override void KillVirus()        {            Console.WriteLine("----对文本文件'" + name + "'进行杀毒");        }    }    //文件夹类：容器构件      public class Folder : AbstractFile    {        //定义集合fileList，用于存储AbstractFile类型的成员          private List<AbstractFile> fileList = new List<AbstractFile>();        private string name;        public Folder(string name)        {            this.name = name;        }        public override void Add(AbstractFile file)        {            fileList.Add(file);        }        public override void Remove(AbstractFile file)        {            fileList.Remove(file);        }        public override AbstractFile GetChild(int i)        {            return fileList[i];        }        //模拟杀毒          public override void KillVirus()        {            Console.WriteLine("****对文件夹'" + name + "'进行杀毒");            //递归调用成员构件的killVirus()方法              foreach (var item in fileList)            {                item.KillVirus();            }        }    }

或在AbstractFIle中使用virtual标记所有方法， 然后在子类中重写需要的方法：

 class Program    {        static void Main(string[] args)        {            AbstractFile file1, file2, file3, file4, folder1, folder2, folder3;            folder1 = new Folder("资料文件夹");            folder2 = new Folder("图像文件夹");            folder3 = new Folder("文本文件夹");            file1 = new ImageFile("小龙女.jpg");            file2 = new ImageFile("张无忌.jpg");            file3 = new TextFile("九阴真经.txt");            file4 = new TextFile("葵花宝典.doc");            folder2.Add(file1);            folder2.Add(file2);            folder3.Add(file3);            folder3.Add(file4);            folder1.Add(folder2);            folder1.Add(folder3);            //从“资料文件夹”节点开始进行杀毒操作              folder1.KillVirus();            Console.ReadLine();        }    }    //抽象文件类：抽象构件      public abstract class AbstractFile    {        public virtual void Add(AbstractFile file)        {            Console.WriteLine("对不起，不支持该方法");        }        public virtual void Remove(AbstractFile file)        {            Console.WriteLine("对不起，不支持该方法");        }        public virtual AbstractFile GetChild(int i)        {            Console.WriteLine("对不起，不支持该方法");            return null;        }        public virtual void KillVirus()        {        }    }    //图像文件类：叶子构件      public class ImageFile : AbstractFile    {        private string name;        public ImageFile(string name)        {            this.name = name;        }        //模拟杀毒          public override void KillVirus()        {            Console.WriteLine("----对图像文件'" + name + "'进行杀毒");        }    }    //文本文件类：叶子构件      public class TextFile : AbstractFile    {        private string name;        public TextFile(string name)        {            this.name = name;        }            //模拟杀毒          public override void KillVirus()        {            Console.WriteLine("----对文本文件'" + name + "'进行杀毒");        }    }    //文件夹类：容器构件      public class Folder : AbstractFile    {        //定义集合fileList，用于存储AbstractFile类型的成员          private List<AbstractFile> fileList = new List<AbstractFile>();        private string name;        public Folder(string name)        {            this.name = name;        }        public override void Add(AbstractFile file)        {            fileList.Add(file);        }        public override void Remove(AbstractFile file)        {            fileList.Remove(file);        }        public override AbstractFile GetChild(int i)        {            return fileList[i];        }        //模拟杀毒          public override void KillVirus()        {            Console.WriteLine("****对文件夹'" + name + "'进行杀毒");            //递归调用成员构件的killVirus()方法              foreach (var item in fileList)            {                item.KillVirus();            }        }    }

     由于在本实例中使用了组合模式，在抽象构件类中声明了所有方法，包括用于管理和访问子构件的方法，如add()方法和remove()方法等，因此在ImageFile等叶子构件类中实现这些方法时必须进行相应的异常处理或错误提示。在容器构件类Folder的killVirus()方法中将递归调用其成员对象的killVirus()方法，从而实现对整个树形结构的遍历。
     如果需要更换操作节点，例如只需对文件夹“文本文件”进行杀毒，客户端代码只需修改一行即可，将代码：
folder1.killVirus();
改为：
folder3.killVirus();

在具体实现时，我们可以创建图形化界面让用户选择所需操作的根节点，无须修改源代码，符合“开闭原则”，客户端无须关心节点的层次结构，可以对所选节点进行统一处理，提高系统的灵活性。

五、透明组合模式与安全组合模式：

    通过引入组合模式，公司设计的杀毒软件具有良好的可扩展性，在增加新的文件类型时，无须修改现有类库代码，只需增加一个新的文件类作为AbstractFile类的子类即可；

1、透明组合模式：

    透明组合模式中，抽象构件Component中声明了所有用于管理成员对象的方法，包括add()、remove()以及getChild()等方法，这样做的好处是确保所有的构件类都有相同的接口。在客户端看来，叶子对象与容器对象所提供的方法是一致的，客户端可以相同地对待所有的对象。透明组合模式也是组合模式的标准形式。上面案例使用的就是透明模式

    透明组合模式的缺点是不够安全，因为叶子对象和容器对象在本质上是有区别的。叶子对象不可能有下一个层次的对象，即不可能包含成员对象，因此为其提供add()、remove()以及getChild()等方法是没有意义的，这在编译阶段不会出错，但在运行阶段如果调用这些方法可能会出错（如果没有提供相应的错误处理代码）。

2.安全组合模式：

     安全组合模式中，在抽象构件Component中没有声明任何用于管理成员对象的方法，而是在Composite类中声明并实现这些方法。这种做法是安全的，因为根本不向叶子对象提供这些管理成员对象的方法，对于叶子对象，客户端不可能调用到这些方法：

    安全组合模式的缺点是不够透明，因为叶子构件和容器构件具有不同的方法，且容器构件中那些用于管理成员对象的方法没有在抽象构件类中定义，因此客户端不能完全针对抽象编程，必须有区别地对待叶子构件和容器构件。在实际应用中，安全组合模式的使用频率也非常高，在Java AWT中使用的组合模式就是安全组合模式。

六、总结：

1. 主要优点
组合模式的主要优点如下：
(1) 组合模式可以清楚地定义分层次的复杂对象，表示对象的全部或部分层次，它让客户端忽略了层次的差异，方便对整个层次结构进行控制。
(2) 客户端可以一致地使用一个组合结构或其中单个对象，不必关心处理的是单个对象还是整个组合结构，简化了客户端代码。
(3) 在组合模式中增加新的容器构件和叶子构件都很方便，无须对现有类库进行任何修改，符合“开闭原则”。
(4) 组合模式为树形结构的面向对象实现提供了一种灵活的解决方案，通过叶子对象和容器对象的递归组合，可以形成复杂的树形结构，但对树形结构的控制却非常简单。
2. 主要缺点
组合模式的主要缺点如下：
在增加新构件时很难对容器中的构件类型进行限制。有时候我们希望一个容器中只能有某些特定类型的对象，例如在某个文件夹中只能包含文本文件，使用组合模式时，不能依赖类型系统来施加这些约束，因为它们都来自于相同的抽象层，在这种情况下，必须通过在运行时进行类型检查来实现，这个实现过程较为复杂。
3. 适用场景
在以下情况下可以考虑使用组合模式：
(1) 在具有整体和部分的层次结构中，希望通过一种方式忽略整体与部分的差异，客户端可以一致地对待它们。
(2) 在一个使用面向对象语言开发的系统中需要处理一个树形结构。
(3) 在一个系统中能够分离出叶子对象和容器对象，而且它们的类型不固定，需要增加一些新的类型。