Warunkowo usuń pierwszy element (z indeksem zerowym) ze strumienia

Mam następujący kod:

Stream<String> lines = reader.lines();

Jeśli ciąg pięści jest równy "email", chcę usunąć pierwszy ciąg ze strumienia. W przypadku innych ciągów ze strumienia nie potrzebuję tego sprawdzania.

Jak mogę to osiągnąć?

PS

Jasne, że mogę przekształcić go w listę, a następnie użyć starej pętli dla starej szkoły, ale dalej potrzebuję strumienia ponownie.

Podczas gdy czytnik będzie w nieokreślonym stanie po zbudowaniu z niego strumienia linii, jest w dobrze zdefiniowanym stanie, zanim to zrobisz.

Więc możesz to zrobić

String firstLine = reader.readLine();
Stream<String> lines = reader.lines();
if(firstLine != null && !"email".equals(firstLine))
    lines = Stream.concat(Stream.of(firstLine), lines);

Co jest moim zdaniem najczystszym rozwiązaniem. Zauważ, że nie jest to to samo co Java 9 dropWhile, które spadłyby więcej niż jedną linię, jeśli się zgadzają.

Jeśli nie możesz mieć listy i musisz to zrobić tylko za pomocą a Stream, możesz to zrobić za pomocą zmiennej.

Chodzi o to, że można użyć zmiennej tylko wtedy, gdy jest ona „ostateczna” lub „efektywnie ostateczna”, więc nie można użyć literalnej wartości logicznej. Nadal możesz to zrobić za pomocą AtomicBoolean:

Stream<String> stream  = Arrays.asList("test", "email", "foo").stream();

AtomicBoolean first = new AtomicBoolean(true);
stream.filter(s -> {
    if (first.compareAndSet(true, false)) {
        return !s.equals("email");
    }
    return true;
})
// Then here, do whatever you need
.forEach(System.out::println);

Uwaga: Nie lubię używać „zmiennych zewnętrznych”, Streamponieważ skutki uboczne są złą praktyką w paradygmacie programowania funkcjonalnego. Lepsze opcje są mile widziane.

Aby uniknąć sprawdzania warunku w każdej linii pliku, po prostu czytam i sprawdzam pierwszy wiersz osobno, a następnie uruchamiam potok na pozostałych liniach bez sprawdzania warunku:

String first = reader.readLine();
Stream<String> firstLines = Optional.of(first)
        .filter(s -> !"email".equals(s))
        .map(s -> Stream.of(s))
        .orElseGet(() -> Stream.empty());

Stream<String> lines = Stream.concat(firstLines, reader.lines());

Prostsze w Javie 9+:

Stream<String> firstLines = Optional.of(first)
        .filter(s -> !"email".equals(s))
        .stream();

Stream<String> lines = Stream.concat(firstLines, reader.lines());

Odpowiedź @Arnouds jest prawidłowa. Możesz utworzyć jeden strumień dla pierwszej linii, a następnie porównać jak poniżej,

Stream<String> firstLineStream = reader.lines().limit(1).filter(line -> !line.startsWith("email"));;
Stream<String> remainingLinesStream = reader.lines().skip(1);
Stream.concat(firstLineStream, remainingLinesStream);

Aby filtrować elementy na podstawie ich indeksu, możesz użyć AtomicIntegerdo przechowywania i zwiększania indeksu podczas przetwarzania Stream:

private static void filter(Stream<String> stream) {
  AtomicInteger index = new AtomicInteger();
  List<String> result = stream
      .filter(el -> {
        int i = index.getAndIncrement();
        return i > 0 || (i == 0 && !"email".equals(el));
      })
      .collect(toList());
  System.out.println(result);
}

public static void main(String[] args) {
  filter(Stream.of("email", "test1", "test2", "test3")); 
  //[test1, test2, test3]

  filter(Stream.of("test1", "email", "test2", "test3")); 
  //[test1, email, test2, test3]

  filter(Stream.of("test1", "test2", "test3")); 
  //[test1, test2, test3]
}

Takie podejście pozwala filtrować elementy według dowolnego indeksu, nie tylko pierwszego.

Trochę bardziej skomplikowany, czerpiąc inspirację z tego fragmentu .

Możesz utworzyć plik, Stream<Integer>który będzie reprezentował indeksy i „spakować” go, Stream<String>aby utworzyć plikStream<Pair<String, Integer>>

Następnie filtruj za pomocą indeksu i zamapuj go z powrotem na Stream<String>

public static void main(String[] args) {
    Stream<String> s = reader.lines();
    Stream<Integer> indexes = Stream.iterate(0, i -> i + 1);

    zip(s, indexes)
        .filter(pair -> !(pair.getKey().equals("email") && pair.getValue() == 0))
        .map(Pair::getKey)
        .forEach(System.out::println);
}

private static Stream<Pair<String,Integer>> zip(Stream<String> stringStream, Stream<Integer> indexesStream){
    Iterable<Pair<String,Integer>> iterable = () -> new ZippedWithIndexIterator(stringStream.iterator(), indexesStream.iterator());
    return StreamSupport.stream(iterable.spliterator(), false);
}

static class ZippedWithIndexIterator implements Iterator<Pair<String, Integer>> {
    private final Iterator<String> stringIterator;
    private final Iterator<Integer> integerIterator;

    ZippedWithIndexIterator(Iterator<String> stringIterator, Iterator<Integer> integerIterator) {
        this.stringIterator = stringIterator;
        this.integerIterator = integerIterator;
    }
    @Override
    public Pair<String, Integer> next() {
        return new Pair<>(stringIterator.next(), integerIterator.next());
    }
    @Override
    public boolean hasNext() {
        return stringIterator.hasNext() && integerIterator.hasNext();
    }
}

Oto przykład z Collectors.reducing. Ale ostatecznie i tak tworzy listę.

Stream<String> lines = Arrays.asList("email", "aaa", "bbb", "ccc")
        .stream();

List reduceList = (List) lines
        .collect(Collectors.reducing( new ArrayList<String>(), (a, v) -> {
                    List list = (List) a;
                    if (!(list.isEmpty() && v.equals("email"))) {
                        list.add(v);
                    }
                    return a;
                }));

reduceList.forEach(System.out::println);

Spróbuj tego:

MutableBoolean isFirst = MutableBoolean.of(true);
lines..dropWhile(e -> isFirst.getAndSet(false) && "email".equals(e))