1. ホーム
  2. java

[解決済み] ストリームから条件に合う最初の要素を取得する

2022-03-06 09:16:21

質問

ストリームで条件に一致する最初の要素を取得するには?私はこれを試しましたが、動作しません。

this.stops.stream().filter(Stop s-> s.getStation().getName().equals(name));

その基準は、Stopとは別のクラスでfilterメソッドが呼び出されているため、機能しません。

public class Train {

private final String name;
private final SortedSet<Stop> stops;

public Train(String name) {
    this.name = name;
    this.stops = new TreeSet<Stop>();
}

public void addStop(Stop stop) {
    this.stops.add(stop);
}

public Stop getFirstStation() {
    return this.getStops().first();
}

public Stop getLastStation() {
    return this.getStops().last();
}

public SortedSet<Stop> getStops() {
    return stops;
}

public SortedSet<Stop> getStopsAfter(String name) {


    // return this.stops.subSet(, toElement);
    return null;
}
}


import java.util.ArrayList;
import java.util.List;

public class Station {
private final String name;
private final List<Stop> stops;

public Station(String name) {
    this.name = name;
    this.stops = new ArrayList<Stop>();

}

public String getName() {
    return name;
}

}

解決方法は?

これは、あなたが探しているものであるかもしれません。

yourStream
    .filter(/* your criteria */)
    .findFirst()
    .get();

さらに良いことに、どの要素にもマッチしない可能性がある場合、その場合は get() はNPEを投げます。だから使ってください。

yourStream
    .filter(/* your criteria */)
    .findFirst()
    .orElse(null); /* You could also create a default object here */


一例です。
public static void main(String[] args) {
    class Stop {
        private final String stationName;
        private final int    passengerCount;

        Stop(final String stationName, final int passengerCount) {
            this.stationName    = stationName;
            this.passengerCount = passengerCount;
        }
    }

    List<Stop> stops = new LinkedList<>();

    stops.add(new Stop("Station1", 250));
    stops.add(new Stop("Station2", 275));
    stops.add(new Stop("Station3", 390));
    stops.add(new Stop("Station2", 210));
    stops.add(new Stop("Station1", 190));

    Stop firstStopAtStation1 = stops.stream()
            .filter(e -> e.stationName.equals("Station1"))
            .findFirst()
            .orElse(null);

    System.out.printf("At the first stop at Station1 there were %d passengers in the train.", firstStopAtStation1.passengerCount);
}

出力は

At the first stop at Station1 there were 250 passengers in the train.