设计模式之Composite模式(组合模式)

呼呼呼，转眼九月30号了，没想到时间过得如此之快，来到新公司已经快4个月，这个月也提前转正了，虽然待遇没有那么高，但周围同事人都很好，相处的很舒服，氛围很好。工作之余还认识了我认为可以终其一生走下去的那个她，希望后面的日子多些顺利，少些磨难。
不多说，前面写了已经很多设计模式，虽然工作中大部分时间用不到，但确实有一种潜移默化的提高，还有对于算法的学习和提高，leetcode现在已经刷了600+，但可惜之前有很多题是只知道思路，就直接复制粘贴了，没有注重多多练习，现在还是需要多加巩固。

这次要写的就是组合模式，类似对文件和文件夹，菜单，以及其他树形结构数据的处理，即要抽象出容器和内容的相同点，在对待时可以进行同时处理，而不用区分两者。

下面是书上关于该模式的例子：计算机的文件系统，有文件和文件夹的概念，文件夹中可以放文件，也可以放文件夹，文件和文件夹也可被统称为“目录条目”，被当作同一对象看待（一致性）。有时，将文件夹和文件看作同一东西，将容器和内容看作同一东西，可以方便处理问题。容器中可以放小容器，可以放内容，小容器中又可以放入更小的容器，一种递归的结构。

还是直接上代码（这次的代码比较简单，而且快下班了，回家还有非常重要的事情要做嘿嘿嘿）

肯定要有的是文件类File，文件夹类Directory，为了抽象出公共性质，还要有其父类，也是个抽象类Entry，表示“目录条目”，UML图先偷一下书上的：
Entry类：

public abstract class Entry {

    public abstract String getName();

    public abstract int getSize();

    public Entry add(Entry entry) throws Exception {
        throw new Exception();
    }

    public void printList() {
        printList("");
    }

    protected abstract void printList(String prefix);

    public String toString() {
        return getName() + " (" + getSize() + ")";
    }

}

File类：

public class File extends Entry {

    private String name;

    private int size;

    public File(String name, int size) {
        this.name = name;
        this.size = size;
    }

    @Override
    public String getName() {
        return name;
    }

    @Override
    public int getSize() {
        return size;
    }

    @Override
    protected void printList(String prefix) {
        System.out.println(prefix + "/" + this);
    }
}

Directory类：

import java.util.ArrayList;
import java.util.Iterator;

public class Directory extends Entry {

    private String name;

    private ArrayList directory = new ArrayList();

    public Directory(String name) {
        this.name = name;
    }

    @Override
    public String getName() {
        return this.name;
    }

    @Override
    public int getSize() {
        int size = 0;
        Iterator it = directory.iterator();
        while(it.hasNext()) {
            Entry entry = (Entry) it.next();
            size += entry.getSize();
        }
        return size;
    }

    public Entry add(Entry entry) {
        directory.add(entry);
        return this;
    }

    @Override
    protected void printList(String prefix) {
        System.out.println(prefix + "/" + this);
        Iterator it = directory.iterator();
        while(it.hasNext()) {
            Entry entry = (Entry) it.next();
            entry.printList(prefix + "/" + name);
        }
    }
}

向文件中进行添加Entry所要抛出的异常类 FileTreatMentException：

public class FileTreatMentException extends RuntimeException {

    public FileTreatMentException() {

    }

    public FileTreatMentException(String msg) {
        super(msg);
    }
}

测试主类 Main:

public class Main {

    public static void main(String[] args) {
        try {

            System.out.println("Making root entries...");
            Directory rootDir = new Directory("root");
            Directory binDir = new Directory("bin");
            Directory tmpDir = new Directory("tmp");
            Directory usrDir = new Directory("usr");

            rootDir.add(binDir);
            rootDir.add(tmpDir);
            rootDir.add(usrDir);

            binDir.add(new File("vi", 10000));
            binDir.add(new File("latex", 20000));
            rootDir.printList();

            System.out.println("");
            System.out.println("Making user entries...");
            Directory yuki = new Directory("yuki");
            Directory hanako = new Directory("hanako");
            Directory tomura = new Directory("tomura");
            usrDir.add(yuki);
            usrDir.add(hanako);
            usrDir.add(tomura);
            yuki.add(new File("diary.html", 100));
            yuki.add(new File("Composite.java", 200));
            hanako.add(new File("memo,tex", 300));
            tomura.add(new File("game.doc", 400));
            tomura.add(new File("junk.mail", 500));
            rootDir.printList();

        } catch (FileTreatMentException e) {
            e.printStackTrace();
        }
    }
}

打印输出：

Making root entries...
/root (30000)
/root/bin (30000)
/root/bin/vi (10000)
/root/bin/latex (20000)
/root/tmp (0)
/root/usr (0)

Making user entries...
/root (31500)
/root/bin (30000)
/root/bin/vi (10000)
/root/bin/latex (20000)
/root/tmp (0)
/root/usr (1500)
/root/usr/yuki (300)
/root/usr/yuki/diary.html (100)
/root/usr/yuki/Composite.java (200)
/root/usr/hanako (300)
/root/usr/hanako/memo,tex (300)
/root/usr/tomura (900)
/root/usr/tomura/game.doc (400)
/root/usr/tomura/junk.mail (500)

后面计划：接下来有时间的话会继续记录一些，最近没啥学习的动力，也不知道学啥好，搞啥好，怎么说呢，希望一切顺其自然吧，什么都好，什么都好。