Set接口的小练习

package Collection;
public class MyDate {
        private  int year;
        private  int mouth;
        private  int day;

    @Override
    public String toString() {
        return this.year+":"+this.mouth+":"+this.day;
    }

    public MyDate(int year, int mouth, int day) {
            this.year = year;
            this.mouth = mouth;
            this.day = day;
        }

        public int getYear() {
            return year;
        }

        public void setYear(int year) {
            this.year = year;
        }

        public int getMouth() {
            return mouth;
        }

        public void setMouth(int mouth) {
            this.mouth = mouth;
        }

        public int getDay() {
            return day;
        }

        public void setDay(int day) {
            this.day = day;
        }

}

MyDate设置一下生日的属性,Eployee类中设置相应的属性

 

/**
 * 如果使用测试类,要求当前类只能有一个空参构造器,否则不可以使用测试,只能通过main方法进行编译
 *
 */


import org.junit.Test;

import java.util.Arrays;
import java.util.Comparator;
import java.util.TreeSet;

public class Employee implements Comparable{
    private String name;
    private int age;
    private MyDate birthday;
    @Override
    public String toString() {
        return "姓名:"+this.name+"年龄:"+this.age+"生日:"+this.birthday;
    }

    public Employee(String name, int age, MyDate birthday) {
        this.name = name;
        this.age = age;
        this.birthday = birthday;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public int getAge() {
        return age;
    }

    public void setAge(int age) {
        this.age = age;
    }

    public MyDate getBirthday() {
        return birthday;
    }

    public void setBirthday(MyDate birthday) {
        this.birthday = birthday;
    }
@Test
    public void test2() {
        Employee employee1=new Employee("hpp",23,new MyDate(1998,3,18));
        Employee employee2=new Employee("ni",3,new MyDate(1997,1,20));
        Employee employee3=new Employee("ns",43,new MyDate(1992,8,1));
        Employee employee4=new Employee("tang",18,new MyDate(2000,5,15));
        Employee employee5=new Employee("an",25,new MyDate(1999,4,25));
        TreeSet treeSet = new TreeSet();
        treeSet.add(employee1);
        treeSet.add(employee2);
        treeSet.add(employee3);
        treeSet.add(employee4);
        treeSet.add(employee5);
        for (Object obj:treeSet){
            System.out.println(obj);
        }
    }
    //按照name进行排序
    @Override
    public int compareTo(Object o) {
        if (o instanceof Employee){
            Employee employee=(Employee) o;
            return this.getName().compareTo(employee.getName());
        }
        throw new RuntimeException("类型不匹配");
    }
    //按照生日日期先后排序

    public static void main(String[] args) {
        Employee employee1 = new Employee("hpp", 23, new MyDate(1998, 3, 18));
        Employee employee2 = new Employee("ni", 3, new MyDate(1997, 1, 20));
        Employee employee3 = new Employee("ns", 43, new MyDate(1992, 8, 1));
        Employee employee4 = new Employee("tang", 18, new MyDate(2000, 5, 15));
        Employee employee5 = new Employee("an", 25, new MyDate(1999, 4, 25));
        TreeSet treeSet2 = new TreeSet(new Comparator() {//直接新建一个Comparator类型的匿名对象
            @Override
            public int compare(Object o1, Object o2) {
                if (o1 instanceof Employee && o2 instanceof Employee) {
                    Employee employee1 = (Employee) o1;
                    Employee employee2 = (Employee) o2;
                    if (employee1.birthday.getMouth() == employee2.birthday.getMouth()) {
                        if (employee1.birthday.getDay() > employee2.birthday.getDay()) {
                            return 1;
                        } else if (employee1.birthday.getDay() < employee2.birthday.getDay()) {
                            return -1;
                        } else {
                            return 0;
                        }
                    } else if (employee1.birthday.getMouth() > employee2.birthday.getMouth()) {
                        return 1;
                    } else {
                        return -1;
                    }
                }
                throw new RuntimeException("类型错误");
            }
        });
        treeSet2.add(employee1);
        treeSet2.add(employee2);
        treeSet2.add(employee3);
        treeSet2.add(employee4);
        treeSet2.add(employee5);
        for (Object obj:treeSet2){
            System.out.println(obj);
        }
    }
}