在一个流中过滤和设置（）

Question

所以我有一些我正在使用的示例代码来尝试找出下面的逻辑。它只是一个大的 main 方法和两个 POJO，但它会运行。我正在调试以获取终止点的值。

public class Main {

    public static void main(String[] args) {

        obj1 obj1 = new obj1();
        obj1 obj12 = new obj1();

        obj2 obj2 = new obj2();
        obj2 obj22 = new obj2();

        obj1.id = 123L;
        obj1.carId = 1234L;
        obj12.id = 123L;
        obj12.carId = 1234L;

        obj2.carId = 1234L;
        obj22.carId = 12345L;

        ArrayList<obj1> obj1Arr = new ArrayList<>();
        ArrayList<obj2> obj2Arr = new ArrayList<>();

        obj1Arr.add(obj1);
        obj1Arr.add(obj12);

        obj2Arr.add(obj2);
        obj2Arr.add(obj22);

        List<obj2> newCarList= obj2Arr.stream()
                .filter(anObjOf2 -> obj1Arr
                        .stream()
                        .anyMatch(carReg -> carReg.getCarId().equals(anObjOf2.getCarId())))
                .collect(Collectors.toList());
    }
}

POJO1：

public class obj1 {

    Long id = null;
    Long carId = null;

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public Long getCarId() {
        return carId;
    }

    public void setCarId(Long carId) {
        this.carId = carId;
    }
}

POJO2：

public class obj2 {

    Long id = null;
    Long carId = null;

    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public Long getCarId() {
        return carId;
    }

    public void setCarId(Long carId) {
        this.carId = carId;
    }
}

如您所见，我正在流式传输并过滤掉与carId 不匹配的任何objs。我的下一个目标是在匹配的对象上使用 set id（不是 carId）。

基本上，如果obj1.carId == obj2.carId THEN set obj1.id = obj2.id。

我的问题是我不知道如何在同一个流中执行此操作。是否可以？我不禁认为此时需要一个迭代器？

Answer 1

不要在一个流中执行这两个操作。这是一种非常糟糕的风格。

首先收集数据。然后变异元素。比如：

Map<obj2, Optional<obj1>> map = 
    obj2Arr.stream()
           .collect(toMap(identity(), o2 -> obj1Arr.stream()
                                                  .filter(o1 -> o1.getCarId().equals(o2.getCarId()))
                                                  .findAny()));

map.forEach((o1, optO2) -> optO2.ifPresent(o2 -> o1.setId(o2.id)));

否则，您可以只使用嵌套的 for-each 循环。

Answer 2

我建议首先为第一个列表中的每个元素创建一个 Map 映射 carId 到 id（我已经修改了你的类以使其更具可读性并删除重复的 ID，我假设是一个错误）：

@Data @AllArgsConstructor
class Car {
    Long id = null;
    Long carId = null;
}
List<Car> obj1Arr = Arrays.asList(new Car(1L, 123L), new Car(2L, 1234L));
List<Car> obj2Arr = Arrays.asList(new Car(0L, 1234L), new Car(0L, 12345L));

Map<Long, Long> carIdToId = obj1Arr.stream()
        .collect(Collectors.toMap(Car::getCarId, Car::getId));
// [Car(id=2, carId=1234), Car(id=0, carId=12345)]

然后您可以使用此Map 过滤具有相同carID 的汽车，而无需与其他汽车进行比较。然后，使用forEach只需在原始列表中设置Id即可。

obj2Arr.stream()
        .filter(anObjOf2 -> carIdToId.containsKey(anObjOf2.getCarId()))
        .forEach(anObjOf2 -> anObjOf2.setId(carIdToId.get(anObjOf2.getCarId())));
System.out.println(obj2Arr);
// [Car(id=2, carId=1234), Car(id=0, carId=12345)]

如果您想要只与carId 相等的项目，您可以在peek 中设置ID，然后在collect 中设置ID。不过，这仍会修改原始实例：

obj2Arr = obj2Arr.stream()
        .filter(anObjOf2 -> carIdToId.containsKey(anObjOf2.getCarId()))
        .peek(anObjOf2 -> anObjOf2.setId(carIdToId.get(anObjOf2.getCarId())))
        .collect(Collectors.toList());
System.out.println(obj2Arr);
// [Car(id=2, carId=1234)]

或者创建一个全新实例的列表：

obj2Arr = obj2Arr.stream()
        .map(Car::getCarId)
        .filter(carIdToId::containsKey)
        .map(carId -> new Car(carIdToId.get(carId), carId))
        .collect(Collectors.toList());
System.out.println(obj2Arr);
// [Car(id=2, carId=1234)]

Answer 3

如果 O(m*n) 解决方案适合您，那么更简单的实现是使用 for 循环：

// classes renamed for naming convention and relatiing to question at the same time 
List<ObjectTwo> newCarList = new ArrayList<>();
for(ObjectTwo objectTwo: objectTwoArr) {
    for (ObjectOne objectOne : objectOneArr) {
        if (objectTwo.getCarId().equals(objectOne.getCarId())) {
            objectOne.setId(objectTwo.getId());
            newCarList.add(objectTwo);
        }
    }
}

m 和 n 分别是列表的大小。

Answer 4

你可以在谓词anyMatch中做到这一点

List<obj2> newCarList= obj2Arr.stream()
            .filter(anObjOf2 -> obj1Arr
                    .stream()
                    .anyMatch(carReg ->  {
            if(carReg.getCarId().equals(anObjOf2.getCarId()))) {
                 carReg.setId(anObjOf2.getId());
                  return true;
               }
           return false;
     }).collect(Collectors.toList());

第二种方法

先过滤

List<obj2> newCarList= obj2Arr.stream()
            .filter(anObjOf2 -> obj1Arr
                    .stream()
                    .anyMatch(carReg -> carReg.getCarId().equals(anObjOf2.getCarId())))
            .collect(Collectors.toList());

现在设置id

newCarList.forEach(obj->{
           obj1Arr.forEach(o-> {
            if(o.getCarId().equals(obj.getCarId()) {
                  o.setId(obj.getId());
                }
           };
    });