จัดกลุ่มรายการตามองค์ประกอบของรายการที่ซ้อนกัน [ซ้ำ]

Jan 25 2021

ฉันมีวัตถุรถยนต์ที่มีพารามิเตอร์เช่น:

private String model;
private BigDecimal price;
private Color color;
private BigDecimal milleage;
private List<String> components;

ฉันสร้างรายการวัตถุรถยนต์:

var cars = List.of(
    Car.create("FORD", BigDecimal.valueOf(120000), Color.RED, BigDecimal.valueOf(150000),
            List.of("AIR CONDITIONING", "VOICE SERVICE")),
    Car.create("FORD", BigDecimal.valueOf(160000), Color.RED, BigDecimal.valueOf(150000),
            List.of("AIR CONDITIONING", "VOICE SERVICE")),
    Car.create("AUDI", BigDecimal.valueOf(200000), Color.BLACK, BigDecimal.valueOf(195000),
            List.of("NAVIGATION", "AUTOMATIC GEARBOX")),
    Car.create("FIAT", BigDecimal.valueOf(70000), Color.BLUE, BigDecimal.valueOf(85000),
            List.of("AIR CONDITIONING", "MANUAL GEARBOX")));

ตอนนี้ฉันต้องการสร้างMap<String, List<Car>>โดยที่สตริงเป็นองค์ประกอบของรายการส่วนประกอบและList<Car>>เป็นรายการของCarวัตถุที่มีส่วนประกอบนี้

ฉันลองทำสิ่งนี้จากปัญหาที่คล้ายกัน แต่ไม่รู้วิธีแก้ปัญหานี้จริงๆ:

static Map<String, List<Car>> carsThatGotComponent(List<Car> cars) {
    return cars.stream()
               .flatMap(car -> car.getComponents()
                       .stream()
                       .map(component -> new AbstractMap.SimpleEntry<>(car, component)))
               .collect(Collectors.groupingBy(
                        Map.Entry::getValue,
                        Collectors.mapping(Map.Entry::getKey, Map.Entry::getValue)));
}

คำตอบ

2 NikolasCharalambidis Jan 25 2021 at 19:15

Collectors#mappingต้องการเป็นพารามิเตอร์ที่สองเป็นดาวน์สตรีCollectorมไม่ใช่ฟังก์ชันการแมป

public static <T,U,A,R> Collector<T,?,R> mapping(
    Function<? super T,? extends U> mapper, 
    Collector<? super U,A,R> downstream)

คุณต้องการใช้Collectors.toList()แทน:

return cars.stream()
    .flatMap(car -> car.getComponents()
                       .stream()
                       .map(component -> new AbstractMap.SimpleEntry<>(car, component)))
    .collect(Collectors.groupingBy(
            AbstractMap.SimpleEntry::getValue,
            Collectors.mapping(AbstractMap.SimpleEntry::getKey, Collectors.toList())));

ตราบใดที่คุณใช้Java-10หรือหลังจากนั้นคุณสามารถใช้ลดความซับซ้อนทั้งสตรีมเป็นหนึ่งในนักสะสมที่ใช้Collectors#flatMappingเป็นของJava-9 :

return cars.stream()
    .collect(Collectors.flatMapping(
             car -> car.getComponents()
                       .stream()
                       .map(component -> new AbstractMap.SimpleEntry<>(car, component)),
             Collectors.groupingBy(AbstractMap.SimpleEntry::getValue,
                     Collectors.mapping(AbstractMap.SimpleEntry::getKey, 
                             Collectors.toList()))));