设计模式6：Composite

Entry.java:

 

package gendwang.cisco.com;



public abstract class Entry 

{

	private int height = 0;

	private String name;

	private Type type;



	public Entry(String name)

	{

		this.name = name;

	}	

	

	public int getHeight() {

		return height;

	}



	public void setHeight(int height) {

		this.height = height;

	}



	public String getName() {

		return name;

	}



	public void setName(String name) {

		this.name = name;

	}



	public Type getType() {

		return type;

	}



	public void setType(Type type) {

		this.type = type;

	}

	

	public abstract void print();

}

File.java:

 

 

package gendwang.cisco.com;



public class File extends Entry

{

	public File(String name)

	{

		super(name);

		setType(Type.File);

	}

	

	public void print()

	{

		int numOfWhiteSpace = getHeight() * 2;

		for(int i = 0; i < numOfWhiteSpace; i++)

		{

			System.out.print(" ");

		}

		

		System.out.println(getName());

	}

}

Directory.java:

 

 

package gendwang.cisco.com;



import java.util.Iterator;

import java.util.LinkedList;

import java.util.List;



public class Directory extends Entry 

{

	private List<Entry> list = new LinkedList<Entry>();

	

	public Directory(String name)

	{

		super(name);

		setType(Type.Directory);

	}

	

	public void addEntry(Entry entry)

	{

		int myHeight = getHeight();

		incrementHeight(entry, myHeight + 1);

		

		list.add(entry);

	}

	

	public List<Entry> getChilden()

	{

		return list;

	}

	

	public void print()

	{

		int numOfWhiteSpace = getHeight() * 2;

		for(int i = 0; i < numOfWhiteSpace; i++)

		{

			System.out.print(" ");

		}

		System.out.println(getName());

		

		Iterator<Entry> itor = list.iterator();

		while(itor.hasNext())

		{

			Entry entry = itor.next();

			entry.print();

		}

	}

	

	private void incrementHeight(Entry entry, int added)

	{

		int height = entry.getHeight();

		entry.setHeight(height + added);

		

		if(entry.getType() == Type.Directory)

		{

			Directory dir = (Directory)entry;

			List<Entry> list = dir.getChilden();

			Iterator<Entry> itor = list.iterator();

			while(itor.hasNext())

			{

				Entry tmp = itor.next();

				incrementHeight(tmp, added);

			}

		}

	}

}

Type.java:

 

 

package gendwang.cisco.com;



public enum Type 

{

	File,

	Directory

}

Test.java:

 

 

package gendwang.cisco.com;



public class Test 

{

	public static void main(String[] args)

	{

		Directory dir1 = new Directory("tmp");

		Entry file1 = new File("hello.jpg");

		Entry file2 = new File("install.log");

		dir1.addEntry(file1);

		dir1.addEntry(file2);

		

		Directory dir2 = new Directory("system");

		

		Directory dir3 = new Directory("windows");

		dir3.addEntry(dir1);

		dir3.addEntry(dir2);

		

		Directory dir4 = new Directory("root");

		Entry file3 = new File("password");

		dir4.addEntry(file3);

		dir4.addEntry(dir3);

		

		dir4.print();

	}

}