Comment vérifier si deux flux sont disjoints?
Je voudrais comparer aux flux, et vérifier s'ils ont 1 ou plusieurs éléments en commun (en trouver 1 suffit pour arrêter d'en chercher plus). Je veux pouvoir l'appliquer aux Streams contenant une classe créée sur mesure.
Par exemple, disons que j'ai une classe qui ressemble à:
public class Point {
public final int row;
public final int col;
public Point(int row, int col) {
this.row = row;
this.col = col;
}
@Override
public boolean equals(Object obj) {
if (obj == null) return false;
if (obj.getClass() != this.getClass()) return false;
final Point other = (Point) obj;
return this.row == other.row && this.col == other.col;
}
@Override
public int hashCode() {
return Objects.hash(row, col);
}
}
Et puis j'ai deux jolis streams qui ressemblent à:
Stream<Point> streamA = Stream.of(new Point(2, 5), new Point(3, 1));
Stream<Point> streamB = Stream.of(new Point(7, 3), new Point(3, 1));
Étant donné que ces flux ont 1 point en commun (à savoir Point(3, 1)
), je voudrais que le résultat final soit vrai.
La fonctionnalité souhaitée peut être représentée comme suit:
public static boolean haveSomethingInCommon(Stream<Point> a, Stream<Point> b){
//Code that compares a and b and returns true if they have at least 1 element in common
}
Réponses
Tout d'abord, vous devez convertir vos Streams en Set ou List pour ne pas obtenir la fameuse erreur:
java.lang.IllegalStateException: stream has already been operated upon or closed
Et puis vous pouvez utiliser anyMatch
comme ceci:
public static boolean haveSomethingInCommon(Stream<Coord> a, Stream<Coord> b) {
Set<Coord> setA = a.collect(Collectors.toSet());
Set<Coord> setB = b.collect(Collectors.toSet());
return setA.stream().anyMatch(setB::contains);
}
Ou vous pouvez convertir uniquement le b
flux en un ensemble et utiliser:
public static boolean haveSomethingInCommon(Stream<Coord> a, Stream<Coord> b) {
Set<Coord> setB = b.collect(Collectors.toSet());
return a.anyMatch(setB::contains);
}
Je recommanderais au Set<Coord>
lieu de Stream<Coord>
comme paramètre dans votre méthode.
public static boolean haveSomethingInCommon(Set<Coord> a, Set<Coord> b) {
return a.stream().anyMatch(b::contains);
}
Sans collecter les deux flux indépendamment, vous pouvez regrouper et identifier si plusieurs valeurs sont mappées à une clé.
public static boolean haveSomethingInCommon(Stream<Coord> a, Stream<Coord> b) {
return Stream.concat(a, b)
.collect(Collectors.groupingBy(Function.identity()))
.values().stream()
.anyMatch(l -> l.size() > 1);
}
Si le même flux peut avoir le même élément deux fois ou plus , vous pouvez modifier le code à utiliser -
Stream.concat(a.distinct(), b.distinct())
il y a une fonction disjoint
dans Collections
:
public static boolean haveSomethingInCommon( Stream<Coord> a, Stream<Coord> b ) {
return( ! Collections.disjoint( a.collect( toList() ), b.collect( toList() ) ) );
}