使用Guava的ComparisonChain实现自定义的排序

 可以看到使用比较器前,先要写一个实体类,还要实现comparable接口,实现compareTo方法。这个方法一般会返回-1 0 1三个int类型数字,分别表示,对象和传入的对象比较,排序应该在传入的对象之后。因为比较器通常是为了集合排序而实现的。

ComparisonChain 的好处就在于,可以用method().method().这样的链式比较,而不用if else这种可读性很差的面向过程的方式去判断大小。比较链的用法是当第一个compare没有比出大小时,转向第二个属性进行比较,如果第二个也没有比较出大小,转向第三个属性进行比较。。以此类推,一般来说前几个属性足够比较了,不会比到最后一个属性。

package comparable;

import com.google.common.collect.ComparisonChain;
import lombok.Data;

import java.util.*;

/**
 * @author xfyou
 * @date 2018/9/30
 */
@Data
public class Course implements Comparable<Course> {

    /**
     * 学号
     */
    private String studentNumber;

    /**
     * 姓名
     */
    private String studentName;

    /**
     * 分数
     */
    private int score;

    @Override
    public int compareTo(Course course) {
        return ComparisonChain.start()
                .compare(course.score, this.score)
                .compare(this.studentNumber, course.studentNumber)
                .result();
    }

    public static void main(String[] args) {
        TreeSet<Course> coursesSet = new TreeSet<>();

        Course course3 = new Course();
        course3.setScore(95);
        course3.setStudentName("Smith");
        course3.setStudentNumber("1003");
        coursesSet.add(course3);

        Course course1 = new Course();
        course1.setScore(90);
        course1.setStudentName("Frank");
        course1.setStudentNumber("1110");
        coursesSet.add(course1);

        Course course2 = new Course();
        course2.setScore(90);
        course2.setStudentName("Tom");
        course2.setStudentNumber("0002");
        coursesSet.add(course2);

        // 编译时使用iterator实现
        for (Course course : coursesSet) {
            System.out.println(course);
        }
    }

}

按照分数由大到小,分数相同的情况下按照学号由小到大来排序。运行结果如下:

Course(studentNumber=1003, studentName=Smith, score=95)
Course(studentNumber=0002, studentName=Tom, score=90)
Course(studentNumber=1110, studentName=Frank, score=90)

 

posted @ 2018-10-08 10:17  FrankYou  阅读(1944)  评论(0编辑  收藏  举报