1. 程式人生 > >java8 stream sort自定義複雜排序

java8 stream sort自定義複雜排序

java8

java 8 自定義排序

需求

今天在專案中遇到個需求,按照物件中的三個屬性進行排序。 具體要求: 前提:物件 Obj [a=a,b=b,c=c] 1、 優先順序為a > b > c 2、 a屬性為中文,固定排序規則為:政府,合作,基金 … … 3、 b的為BigDecimal型別,固定的排序規則為:A,B,C 字典排序 4、 c為java.util.Date型別,規則為:降序排序

其實看這個需求,第3點和第4點不是什麼問題,但是第1點,會考慮下怎麼實現好。 直接上方案吧!

方案一

新建一張排序表,至少要有欄位 【名稱—中文名稱(政府、合作、基金等)】【排序編號—(1、2、3)】,在Obj表中的a欄位存排序表的id。 此時可以直接用sql語句 ORDER BY 排序即可。 優點:可動態配置。

方案二

完全用java程式碼操作,和sql無關,上程式碼:

  • Obj.java 和 Sort.java
package TestSort;
import java.math.BigDecimal;
public class Obj {
	private String name;
	private BigDecimal price;
	public Obj(String name, BigDecimal price){
		this.name = name;
		this.price = price;
	}
	
	public Obj(){
		
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public BigDecimal getPrice() {
		return price;
	}
	public void setPrice(BigDecimal price) {
		this.price = price;
	}
	@Override
	public String toString() {
		return "Obj [name=" + name + ", price=" + price + "]";
	}	
}




package TestSort;
import java.math.BigDecimal;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
public class Sort {
	
	public static void main(String[] args) {
		List<Obj> list =  Arrays.asList(
				new Obj("政府", null),
				new Obj("政府", new BigDecimal("1216.23")),
				new Obj("商業", new BigDecimal("123.23")),
				new Obj("PPD", new BigDecimal("123.23")),
				new Obj("合作", new BigDecimal("127.23")),
				new Obj("合作", new BigDecimal("125.23")),
				new Obj("諮詢", null),
				new Obj(null, null)
				);
		
		/*Comparator<Obj> byName =  Comparator.comparing(Obj::getName).reversed();
		
		Comparator<Obj> finalByPrice= byName.thenComparing(Obj::getPrice,Comparator.nullsFirst(BigDecimal::compareTo)).reversed();
		
		List<Obj> result = list.stream().filter(new Predicate<Obj>() {
			    @Override
			    public boolean test(Obj obj) {
			    	if(obj.getName() == null && obj.getPrice() ==null){
			    		return false;
			    	}
			    	return true;
			    }
		    }).sorted(finalByPrice).collect(Collectors.toList());*/
		
		List<Obj> result = list.stream().sorted(
        	 //先按照name排序(模擬需求的a屬性排序)
				Comparator.comparing(Obj::getName,(x,y)->{
					if(x == null && y != null){
						return 1;
					}else if(x !=null && y == null){
						return -1;
					}else if(x == null && y == null){
						return -1;
					}else if("PPD".equals(x) || "PPD".equals(y)){
						if(x.equals(y)){
							return 0;
						}else if("PPD".equals(x)){
							return -1;
						}else{
							return 1;
						}
					}else
					if("合作".equals(x) || "合作".equals(y)){
						if(x.equals(y)){
							return 0;
						}else if("合作".equals(x)){
							return -1;
						}else{
							return 1;
						}
					}else
					if("政府".equals(x) || "政府".equals(y)){
						if(x.equals(y)){
							return 0;
						}else if("政府".equals(x)){
							return -1;
						}else{
							return 1;
						}
					}
					return 0;	})
          //再按照其他欄位排序,要考慮null(模擬需求b和c欄位排序)
          .thenComparing(Comparator.comparing(Obj::getPrice,
          Comparator.nullsFirst(BigDecimal::compareTo)).reversed()
   )).collect(Collectors.toList());
		System.out.println(result);
		System.out.println(result.size());
	}
	}


方案二的缺點就是硬編碼,使用者改排序就得改原始碼。對第二種方案的改進:  
package TestSort;

import java.math.BigDecimal;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;



public class Sort {
	public static void main(String[] args) {
		List<Obj> list =  Arrays.asList(
				new Obj("政府", null),
				new Obj("政府", new BigDecimal("1216.23")),
				new Obj("商業", new BigDecimal("123.23")),
				new Obj("PPD", new BigDecimal("123.23")),
				new Obj("合作", new BigDecimal("127.23")),
				new Obj("合作", new BigDecimal("125.23")),
				new Obj("諮詢", null),
				new Obj(null, null)
				);
		/*Comparator<Obj> byName =  Comparator.comparing(Obj::getName).reversed();
		Comparator<Obj> finalByPrice= byName.thenComparing(Obj::getPrice,Comparator.nullsFirst(BigDecimal::compareTo)).reversed();
		List<Obj> result = list.stream().filter(new Predicate<Obj>() {
			    @Override
			    public boolean test(Obj obj) {
			    	if(obj.getName() == null && obj.getPrice() ==null){
			    		return false;
			    	}
			    	return true;
			    }
		    }).sorted(finalByPrice).collect(Collectors.toList());*/
		//此處模擬從資料讀取配置到list
		List<String> sortList =  Arrays.asList("PPD","政府","合作");
		list.stream().sorted(
				Comparator.comparing(Obj::getName,(x,y)->{
					if(x == null && y != null){
						return 1;
					}else if(x !=null && y == null){
						return -1;
					}else if(x == null && y == null){
						return -1;
					}else{
						//按照讀取的list順序排序
						for(String sort : sortList){
							if(sort.equals(x) || sort.equals(y)){
								if(x.equals(y)){
									return 0;
								}else if(sort.equals(x)){
									return -1;
								}else{
									return 1;
								}
							}
						}
						return 0;
					}
				}).thenComparing(Comparator.comparing(Obj::getPrice,Comparator.nullsFirst(BigDecimal::compareTo)).reversed())
				).collect(Collectors.toList()).forEach(System.out::println);;
	}
	}