두 스트림을 병합

Question

값을 Comparator을 기준으로 두 개의 Stream에 값을 병합하는 방법을 구현하려고합니다.

나는 스트림을 반복하고 Stream.Builder에 값을 삽입하는 방법이 있었지만 게으른 평가 버전을 만드는 방법을 알지 못했습니다 (많은 스트림 작업이), 그래서 그것은 또한 무한한 시내를 다룰 수 있습니다.

나는 그것이 입력 데이터에 단일 병합 패스를 수행되고 싶지 모든, 하지 종류의 스트림 (사실,이 스트림이 무질서가 될 가능성이 높습니다,이 장애는 보존되어야한다) .

static Stream<E> merge(Stream<E> first, Stream<E> second, Comparator<E> c) 

이렇게 두 스트림을 병합 할 수 있습니까?

나는 두 Queue 입력으로의 출력과 같은 일부 Consumer와 함께이 일을한다면, 그것은 매우 간단 같습니다

void merge(Queue<E> first, Queue<E> second, Consumer<E> out, Comparator<E> c){ 
    while(!first.isEmpty() && !second.isEmpty() 
     if(c.compare(first.peek(), second.peek()) <= 0) 
      out.accept(first.remove()); 
     else 
      out.accept(second.remove()); 
    for(E e:first) 
     out.accept(e); 
    for(E e:second) 
     out.accept(e); 
} 

하지만 게으른 평가와 함께이 작업을 수행해야하고, 스트리밍합니다.

예 1 :

1, 2, 2, 2, 3, 3, 1, 2, 2, 2, 2, 3 

예 :

merge(
    Stream.of(1, 2, 3, 1, 2, 3), 
    Stream.of(2, 2, 3, 2, 2, 2), 
    Comparator.naturalOrder() 
); 

이 시퀀스를 생성 할 스트림을 반환

은 코멘트를 해결하기 위해, 몇 가지 예를 들어 입력 및 결과는 2 :

merge(
    Stream.iterate(5, i->i-1), 
    Stream.iterate(1, i->i+1), 
    Comparator.naturalOrder() 
); 

(A)을 할 수없는 종류의 무한 스트리밍하기 때문에,이 단지 concat(first,second).sort()하지 않습니다 당신이 볼 수 있듯이

1, 2, 3, 4, 5, 5, 4, 3, 2, 1, 0, -1 ... 

:

는 (잘 INT_MAX + 5 항목) 시퀀스를 생성 할 스트림을 무한을 반환 그리고 (b) 스트림을 정렬 할 수있는 경우에도 원하는 결과를 얻지 못합니다.

Answer 1

Stream.Builder을 통과하는 대신 Spliterator을 구현해야합니다. 이를 위해 상당히 순차적 인 작업이기 때문에 Iterator을 통과 할 수도 있습니다. 구아바 사용하기 가볍게,

return StreamSupport.stream(Spliterators.spliteratorUnknownSize(
    Iterators.mergeSorted(
     Arrays.asList(stream1.iterator(), stream2.iterator()), 
     comparator), 
    Spliterator.ORDERED), 
    false /* not parallel */);