深入Java设计模式之组合模式

package com.lzhsite.technology.designPattern.composite.GeneralImplement;
public interface Component
{
    public Component getComposite();
    public void sampleOperation();
}

package com.lzhsite.technology.designPattern.composite.GeneralImplement;
import java.util.Enumeration;
import java.util.Vector;
public class Composite implements Component
{
    private Vector componentVector = new Vector();
    public Component getComposite()
    {
        return this;
    }
    public void sampleOperation()
    {
        Enumeration enumeration = getChildren();
        while (enumeration.hasMoreElements())
        {
            ((Component) enumeration.nextElement()).sampleOperation();
        }
    }
    public void add(Component component)
    {
        componentVector.addElement(component);
    }
    public void remove(Component component)
    {
        componentVector.removeElement(component);
    }
    public Enumeration getChildren()
    {
        return componentVector.elements();
    }
}

package com.lzhsite.technology.designPattern.composite.GeneralImplement;
public class Leaf implements Component
{
    public Component getComposite()
    {
        return null;
    }
    public void sampleOperation()
    {
    }
}

package com.lzhsite.technology.designPattern.composite.FolderFileImplement;
public class ClientForFolderFileImplement
{
    public static void main(String[] args)
    {
        Root root1 = new Folder("C://");
        Root root2 = new Folder("D://");
        Root winDir = new Folder("Windows/");
        Root sysDir = new Folder("System/");
        Root userDir = new Folder("User/");
        Root defaultDir = new Folder("Default/");
        Root helloWorld = new File("HelloWorld.java");
        winDir.addfile(helloWorld);
        Root helloWorld1 = new File("HelloWorld1.java");
        Root helloWorld2 = new File("HelloWorld2.java");
        sysDir.addfile(helloWorld1);
        sysDir.addfile(helloWorld2);
        root1.addfile(winDir);
        root1.addfile(sysDir);
        Root HelloWorld3 = new File("HelloWorld3.java");
        Root HelloWorld4 = new File("HelloWorld4.java");
        Root HelloWorld5 = new File("HelloWorld5.java");
        userDir.addfile(defaultDir);
        userDir.addfile(HelloWorld4);
        userDir.addfile(HelloWorld5);
        defaultDir.addfile(HelloWorld3);
        root2.addfile(userDir);
        root1.display();
        root2.display();
    }
}

package com.lzhsite.technology.designPattern.composite.FolderFileImplement;
import java.util.List;
public interface Root
{
    public boolean addfile(Root file);
    public boolean removeFile(Root file);
    public List<Root> getFiles();
    public void display();
}

package com.lzhsite.technology.designPattern.composite.FolderFileImplement;
import java.util.ArrayList;
import java.util.List;
public class Folder implements Root {
  private String name;
  private List<Root> folders;
  public Folder(String name) {
    this.name = name;
    this.folders = new ArrayList<Root>();
  }
  public boolean addfile(Root file) {
    return folders.add(file);
  }
  public boolean removeFile(Root file) {
    return folders.remove(file);
  }
  public List<Root> getFiles() {
    return folders;
  }
  public String getName() {
    return name;
  }
  private static String getLevelStr(int level) {
    StringBuffer levelStr = new StringBuffer();
    for (int levelI = 0; levelI < level; levelI++) {
      levelStr.append("\t");
    }
    return levelStr.toString();
  }
    private int level = 0;
  public void printFileName(Root root) {
    if (root instanceof Folder) {
      System.out.print(getLevelStr(level));
      System.out.print("   |   ");
      System.out.println(((Folder) root).getName());
      for (Root sonRoot : root.getFiles()) {
        level++;
        printFileName(sonRoot);
        level--;
      }
    } else if (root instanceof File) {
      System.out.print(getLevelStr(level));
      root.display();
      return;
    }
  }
  public void display() {
    System.out.println(name);
    for (Root root : folders) {
      printFileName(root);
    }
  }
}

package com.lzhsite.technology.designPattern.composite.FolderFileImplement;
import java.util.List;
public class File implements Root
{
    private String name;
    public File(String name)
    {
        this.name = name;
    }
    public boolean addfile(Root file)
    {
        return false;
    }
    public boolean removeFile(Root file)
    {
        return false;
    }
    public List<Root> getFiles()
    {
        return null;
    }
    public void display()
    {
        System.out.println("   |   " + name);
    }
}