如何在groupingBy之后进行排序并选择排序列表中的第一个元素

时间:2018-05-20 11:40:12

标签: lambda java-8

问题陈述:假设我列出了PriceRow(productCode,key,sector)对象

        List<PriceRow> priceRowList = new ArrayList<>();
        priceRowList.add(new PriceRow("10kgbag","", "SECTOR"));
        priceRowList.add(new PriceRow("10kgbag","12345", ""));
        priceRowList.add(new PriceRow("10kgbag","", ""));
        priceRowList.add(new PriceRow("20kgbag","", "SECTOR"));
        priceRowList.add(new PriceRow("20kgbag","12345", ""));
        priceRowList.add(new PriceRow("20kgbag","", ""));
        priceRowList.add(new PriceRow("30kgbag","", "SECTOR"));
        priceRowList.add(new PriceRow("30kgbag","", ""));
        priceRowList.add(new PriceRow("40kgbag","", ""));
        priceRowList.add(new PriceRow("50kgbag","", ""));

现在,我需要按 productCode 对其进行分组,然后根据第一个然后扇区对其进行排序,如果两行都行不可用然后用(键=空白)和(扇区=空白)取行,并从排序列表中取出第一行来创建一个地图

因此最终的断言看起来应该是

assertEquals("12345",map.get("10kgbag").getFlightKey());
assertEquals("12345",map.get("20kgbag").getFlightKey());
assertEquals("SECTOR",map.get("30kgbag").getSector());  
assertEquals("",map.get("40kgbag").getFlightKey());     
assertEquals("",map.get("50kgbag").getFlightKey());

我提出的解决方案是

import org.apache.commons.lang.StringUtils;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class Example {
    public Map<String,PriceRow> evaluate(List<PriceRow> priceRowList) {
        Map<String,PriceRow>  map =  priceRowList.stream()
                .collect(Collectors.groupingBy(priceRow -> priceRow.getProductCode(),
                        Collectors.collectingAndThen(Collectors.toList(), value -> getMostEligibleValue(value))));
        return map;
    }


    private PriceRow getMostEligibleValue(List<PriceRow> priceRowList){
        for(PriceRow priceRowWithKey : priceRowList)
            if(StringUtils.isNotBlank(priceRowWithKey.getKey()))
                return priceRowWithKey;

        for(PriceRow priceRowWithSector : priceRowList)
            if(StringUtils.isNotBlank(priceRowWithSector.getSector()))
                return priceRowWithSector;

        return priceRowList.stream().findFirst().get();
    }
}

希望我能够解释问题陈述。如果有更好的解决方案可以解决这个问题,请告诉我。在此先感谢您的帮助。

0 个答案:

没有答案