_102_Java_TreeSet的定制排序（CustomSort）

/*
     *TreeSet的定制排序
     *1、创建一个实现了Comparator接口的类对象。
     *2、将此对象做为形参传递给TreeSet的构造器中。
     **/

 --------------------------------------------------

--------------------------------------------------

 _003_Customer接口类

public class _003_Customer {

	private String name;
	private Integer id;
	
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public Integer getId() {
		return id;
	}
	public void setId(Integer id) {
		this.id = id;
	}
	public _003_Customer() {
	}
	public _003_Customer(String name, Integer id) {
		super();
		this.name = name;
		this.id = id;
	}
	@Override
	public String toString() {
		return "Customer_13_05 [name=" + name + ", id=" + id + "]";
	}
	@Override
	public int hashCode() {
		final int prime = 31;
		int result = 1;
		result = prime * result + ((id == null) ? 0 : id.hashCode());
		result = prime * result + ((name == null) ? 0 : name.hashCode());
		return result;
	}
	@Override
	public boolean equals(Object obj) {
		if (this == obj)
			return true;
		if (obj == null)
			return false;
		if (getClass() != obj.getClass())
			return false;
		_003_Customer other = (_003_Customer) obj;
		if (id == null) {
			if (other.id != null)
				return false;
		} else if (!id.equals(other.id))
			return false;
		if (name == null) {
			if (other.name != null)
				return false;
		} else if (!name.equals(other.name))
			return false;
		return true;
	}
	
}

import java.util.Comparator;
import java.util.Iterator;
import java.util.Set;
import java.util.TreeSet;

import org.junit.Test;

public class _003_TreeSet_CustomSort_13_05 {
	
	/*
	 *TreeSet的定制排序
	 *1、创建一个实现了Comparator接口的类对象。
	 *2、将此对象做为形参传递给TreeSet的构造器中。
	 **/
	
	@Test
	public void testA(){
		Comparator com=new Comparator() {
		/*向TreeSet中添加_003_Customer类的对象，
		 *在此compare方法中，指明是按照_003_Customer哪个属性排序的。*/
			@Override
			public int compare(Object o1, Object o2) {
				// TODO Auto-generated method stub
				if(o1 instanceof _003_Customer && o2 instanceof _003_Customer) {
					_003_Customer c1=(_003_Customer) o1;
					_003_Customer c2=(_003_Customer) o2;
					int i = c1.getId().compareTo(c2.getId());
					if(i==0) {
						return c1.getName().compareTo(c2.getName());
					}
					return i;
				}
				return 0;
			}
		};
		
		Set set = new TreeSet(com);
		set.add(new _003_Customer("AA",34));
		set.add(new _003_Customer("AA",34));
		set.add(new _003_Customer("AA",35));
		set.add(new _003_Customer("BB",35));
		set.add(new _003_Customer("CC",35));
		set.add(new _003_Customer("DD",35));
		
		Iterator it = set.iterator();
		while(it.hasNext()) {
			System.out.println(it.next());
		}
		
	}
	
}