java8 stream sort自定義複雜排序
阿新 • • 發佈:2018-12-14
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);; } }