Bỏ qua phần tử x cuối cùng trong Dòng <T>

Question

Nếu tôi có số Stream<T>, tôi có thể dễ dàng sử dụng skip(long) để bỏ qua một vài thành phần đầu tiên của luồng. Tuy nhiên, dường như không có tương đương với việc bỏ qua một số phần tử đã cho ở cuối luồng.

Giải pháp rõ ràng nhất là sử dụng limit(originalLength - elementsToRemoveAtEnd), nhưng điều đó đòi hỏi phải biết trước về độ dài ban đầu, không phải lúc nào cũng như vậy.

Có cách nào để xóa vài phần tử cuối cùng của luồng có độ dài không xác định mà không phải thu thập nó vào một số Collection, đếm các phần tử và phát lại nó không?

Answer 1

Không có giải pháp lưu trữ chung miễn phí cho Stream s có thể có độ dài không xác định. Tuy nhiên, bạn không cần phải thu thập toàn bộ dòng, bạn chỉ cần một lưu trữ lớn như số lượng các yếu tố mà bạn muốn bỏ qua:

static <T> Stream<T> skipLastElements(Stream<T> s, int count) { 
    if(count<=0) { 
     if(count==0) return s; 
     throw new IllegalArgumentException(count+" < 0"); 
    } 
    ArrayDeque<T> pending=new ArrayDeque<T>(count+1); 
    Spliterator<T> src=s.spliterator(); 
    return StreamSupport.stream(new Spliterator<T>() { 
     public boolean tryAdvance(Consumer<? super T> action) { 
      while(pending.size()<=count && src.tryAdvance(pending::add)); 
      if(pending.size()>count) { 
       action.accept(pending.remove()); 
       return true; 
      } 
      return false; 
     } 
     public Spliterator<T> trySplit() { 
      return null; 
     } 
     public long estimateSize() { 
      return src.estimateSize()-count; 
     } 
     public int characteristics() { 
      return src.characteristics(); 
     } 
    }, false); 
} 
public static void main(String[] args) { 
    skipLastElements(Stream.of("foo", "bar", "baz", "hello", "world"), 2) 
    .forEach(System.out::println); 
} 

Answer 2

Các mã sau sử dụng ArrayDeque để đệm n yếu tố nơi n là số lượng các phần tử để bỏ qua ở cuối. Bí quyết là sử dụng skip(n). Điều này làm cho các thành phần n đầu tiên được thêm vào ArrayDeque. Sau đó, khi các phần tử n đã được lưu vào bộ đệm, luồng sẽ tiếp tục xử lý các phần tử nhưng bật các phần tử từ ArrayDeque. Khi kết thúc luồng, phần tử n cuối cùng bị kẹt trong ArrayDeque và bị hủy.

ArrayDeque không cho phép null yếu tố. Mã bên dưới bản đồ null thành NULL_VALUE trước khi thêm vào ArrayDeque và sau đó bản đồ NULL_VALUE quay lại null sau khi popping từ ArrayDeque.

private static final Object NULL_VALUE = new Object(); 

public static <T> Stream<T> skipLast(Stream<T> input, int n)     
{ 
    ArrayDeque<T> queue; 

    if (n <= 0) 
     return(input); 

    queue = new ArrayDeque<>(n + 1); 

    input = input. 
     map(item -> item != null ? item : NULL_VALUE). 
     peek(queue::add). 
     skip(n). 
     map(item -> queue.pop()). 
     map(item -> item != NULL_VALUE ? item : null); 

    return(input); 
}