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());


    @Override
    public 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("測(cè)試名稱1");

	RecordDto record02= new RecordDto();
	record02.setOrderNum(0);
	record02.setSName("測(cè)試名稱2");

	RecordDto record03 = new RecordDto();
	record03.setOrderNum(1);
	record03.setSName("測(cè)試名稱3");

	RecordDto record04= new RecordDto();
	record04.setOrderNum(2);
	record04.setSName("測(cè)試名稱4");

	RecordDto record05= new RecordDto ();
	record05.setOrderNum(0);
	record05.setSName("測(cè)試名稱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());
	});
}

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());
}