需求背景

最近在對一個集合列表的數據進行排序，需求是要集合數據按照一個排序狀態值進行排序，而這個狀態值，不是按照從小到大這樣的順序排序的，而是要按照特定的順序，比如按照1, 0, 2的順序排的，所以需要自定義一個排序工具類來實現

解決方案

寫一個排序工具類

package cn.nzp.ems.ecb.server.business.screen;import java.io.Serializable;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;public class CustomizeNumComparator implements Comparator<Integer>, Serializable {private static final long serialVersionUID = 6076418974912880405L;List<Integer> numList = Stream.of(1, 0, 2).collect(Collectors.toList());@Overridepublic int compare(Integer o1, Integer o2) {int i1 = numList.indexOf(o1);int i2 = numList.indexOf(o2);return (i1 == -1 || i2 == -1) ? (i2 - i1) : i1 - i2;}
}

測試一下排序

@Test
public void testCompare() {List<RecordDto> list = Lists.newArrayList();RecordDto record01= new RecordDto();record01.setOrderNum(2);record01.setSName("測試名稱1");RecordDto record02= new RecordDto();record02.setOrderNum(0);record02.setSName("測試名稱2");RecordDto record03 = new RecordDto();record03.setOrderNum(1);record03.setSName("測試名稱3");RecordDto record04= new RecordDto();record04.setOrderNum(2);record04.setSName("測試名稱4");RecordDto record05= new RecordDto ();record05.setOrderNum(0);record05.setSName("測試名稱5");list.add(record01);list.add(record02);list.add(record03);list.add(record04);list.add(record05);list = list.stream().sorted(Comparator.comparing(RecordDto::getOrderNum, new CustomizeNumComparator())).collect(Collectors.toList());list.stream().forEach(e -> {System.out.println(e.getName() +", " +e.getOrderNum());});
}

跑一下測試結果

如果不想寫一個排序工具類，也可以創建一個Comparator即可

public void testCompare() {List<Integer> numList = Stream.of(1, 0, 2).collect(Collectors.toList());Comparator<RecordDto> comparingIndex = Comparator.comparing(record -> {final int index = numList.indexOf(record.getOrderNum());return index == -1 ? Long.MAX_VALUE : index;});list = list.stream().sorted(comparingIndex.thenComparing(RecordDto::getCreateTime, Comparator.nullsLast(Date::compareTo))).collect(Collectors.toList());
}