วิธีแปลง iterator เป็นสตรีม


468

ฉันกำลังมองหาวิธีที่กระชับในการแปลงIteratorเป็นStreamหรือมากกว่าโดยเฉพาะเพื่อ "ดู" ตัววนซ้ำเป็นสตรีม

เพื่อเหตุผลด้านประสิทธิภาพฉันต้องการหลีกเลี่ยงสำเนาตัววนซ้ำในรายการใหม่:

Iterator<String> sourceIterator = Arrays.asList("A", "B", "C").iterator();
Collection<String> copyList = new ArrayList<String>();
sourceIterator.forEachRemaining(copyList::add);
Stream<String> targetStream = copyList.stream();

จากคำแนะนำในความคิดเห็นฉันได้ลองใช้Stream.generate:

public static void main(String[] args) throws Exception {
    Iterator<String> sourceIterator = Arrays.asList("A", "B", "C").iterator();
    Stream<String> targetStream = Stream.generate(sourceIterator::next);
    targetStream.forEach(System.out::println);
}

อย่างไรก็ตามฉันได้รับNoSuchElementException(เนื่องจากไม่มีการร้องขอhasNext)

Exception in thread "main" java.util.NoSuchElementException
    at java.util.AbstractList$Itr.next(AbstractList.java:364)
    at Main$$Lambda$1/1175962212.get(Unknown Source)
    at java.util.stream.StreamSpliterators$InfiniteSupplyingSpliterator$OfRef.tryAdvance(StreamSpliterators.java:1351)
    at java.util.Spliterator.forEachRemaining(Spliterator.java:326)
    at java.util.stream.ReferencePipeline$Head.forEach(ReferencePipeline.java:580)
    at Main.main(Main.java:20)

ฉันดูStreamSupportแล้วCollectionsแต่ไม่พบอะไรเลย


5
มีความเป็นไปได้ที่ซ้ำกันของวิธีสร้างสตรีมไม่สิ้นสุด <E> จาก Iterator <E>
Dmitry Ginzburg

3
@DmitryGinzburg euh ฉันไม่ต้องการที่จะสร้างกระแส "ไม่มีที่สิ้นสุด"
gontard

1
@DmitryGinzburg Stream.generate(iterator::next)ทำงานอย่างไร
gontard

1
@DmitryGinzburg นั่นใช้ไม่ได้กับ iterator ที่ จำกัด
assylias

คำตอบ:


543

วิธีหนึ่งคือการสร้าง Spliterator จาก Iterator และใช้เป็นพื้นฐานสำหรับสตรีมของคุณ:

Iterator<String> sourceIterator = Arrays.asList("A", "B", "C").iterator();
Stream<String> targetStream = StreamSupport.stream(
          Spliterators.spliteratorUnknownSize(sourceIterator, Spliterator.ORDERED),
          false);

อีกทางเลือกหนึ่งซึ่งอาจจะอ่านได้มากกว่าคือการใช้ Iterable - และการสร้าง Iterable จาก Iterator นั้นง่ายมากกับ lambdas เพราะ Iterable เป็นอินเตอร์เฟสที่ใช้งานได้:

Iterator<String> sourceIterator = Arrays.asList("A", "B", "C").iterator();

Iterable<String> iterable = () -> sourceIterator;
Stream<String> targetStream = StreamSupport.stream(iterable.spliterator(), false);

26
สตรีมขี้เกียจ: รหัสเชื่อมโยงสตรีมไปยัง Iterator เท่านั้น แต่การทำซ้ำตามจริงจะไม่เกิดขึ้นจนกว่าจะมีการใช้งานเทอร์มินัล หากคุณใช้ตัววนซ้ำในระหว่างนี้คุณจะไม่ได้รับผลลัพธ์ตามที่คาดหวัง ตัวอย่างเช่นคุณสามารถแนะนำsourceIterator.next()ก่อนที่จะใช้สตรีมและคุณจะเห็นผลกระทบ (รายการแรกจะไม่เห็นโดยสตรีม)
assylias

9
@assylias ใช่มันดีจริงๆ! บางทีคุณอาจจะอธิบายให้ผู้อ่านในอนาคตเห็นว่าเป็นสายเวทย์มนตร์Iterable<String> iterable = () -> sourceIterator;นี้ ฉันต้องยอมรับว่าต้องใช้เวลาพอสมควรในการทำความเข้าใจ
gontard

7
ฉันควรจะบอกสิ่งที่ฉันพบ Iterable<T>เป็นที่มีเพียงหนึ่งวิธีนามธรรมFunctionalInterface iterator()ดังนั้น() -> sourceIteratorการแสดงออกแลมบ์ดาเป็นIterableตัวอย่างอินสแตนซ์เป็นการใช้งานที่ไม่ระบุชื่อ
Jin Kwon

13
อีกครั้ง() -> sourceIterator;เป็นรูปแบบที่สั้นลงของnew Iterable<>() { @Override public Iterator<String> iterator() { return sourceIterator; } }
จินค

7
@ JinKwon มันไม่ใช่รูปแบบที่สั้นลงของคลาสนิรนาม (มีความแตกต่างเล็กน้อยเช่นขอบเขตและวิธีการคอมไพล์) แต่มันก็มีพฤติกรรมคล้ายกันในกรณีนี้
assylias

122

ห้องสมุด Guava ให้บริการตั้งแต่เวอร์ชัน 21 Streams.stream(iterator)

มันจะมีอะไร @ assylias 's คำตอบที่แสดงให้เห็นว่า



ดีกว่าที่จะใช้สิ่งนี้อย่างสม่ำเสมอจนกว่าจะถึงเวลาเช่น JDK รองรับสายการบินเดียว มันจะง่ายกว่าที่จะหา (ด้วยเหตุนี้ refactor) สิ่งนี้ในอนาคตกว่าโซลูชัน JDK บริสุทธิ์ที่แสดงที่อื่น
drekbour

นี่ยอดเยี่ยม แต่… Java มีตัววนซ้ำพื้นเมืองและลำธารอย่างไร… แต่ไม่มีวิธีที่ตรงไปตรงมาในตัวจากหนึ่งไปอีก! การละเว้นในความคิดของฉันค่อนข้าง
Dan Lenski

92

คำแนะนำยอดเยี่ยม! นี่ฉันใช้ซ้ำได้กับมัน:

public class StreamUtils {

    public static <T> Stream<T> asStream(Iterator<T> sourceIterator) {
        return asStream(sourceIterator, false);
    }

    public static <T> Stream<T> asStream(Iterator<T> sourceIterator, boolean parallel) {
        Iterable<T> iterable = () -> sourceIterator;
        return StreamSupport.stream(iterable.spliterator(), parallel);
    }
}

และการใช้งาน (อย่าลืมนำเข้า asStream แบบคงที่):

List<String> aPrefixedStrings = asStream(sourceIterator)
                .filter(t -> t.startsWith("A"))
                .collect(toList());

43

สิ่งนี้เป็นไปได้ใน Java 9

Stream.generate(() -> null)
    .takeWhile(x -> iterator.hasNext())
    .map(n -> iterator.next())
    .forEach(System.out::println);

1
ง่ายมีประสิทธิภาพและไม่ต้องใช้การ subclassing - นี่ควรเป็นคำตอบที่ยอมรับได้!
martyglaubitz

1
น่าเสียดายที่สิ่งเหล่านี้ดูเหมือนจะไม่ทำงานกับ.parallel()สตรีม นอกจากนี้ยังปรากฏว่าช้าลงกว่าจะSpliteratorใช้งานต่อเนื่องเล็กน้อย
โทมัส Ahle

นอกจากนี้วิธีแรกจะพ่นหากตัววนซ้ำว่างเปล่า วิธีที่สองใช้งานได้ในตอนนี้ แต่มันละเมิดข้อกำหนดของฟังก์ชั่นในแผนที่และใช้ในขณะที่ไร้สัญชาติดังนั้นฉันลังเลที่จะทำเช่นนั้นในรหัสการผลิต
Hans-Peter Störr

จริง ๆ แล้วนี่ควรเป็นคำตอบที่ได้รับการยอมรับ แม้ว่าparallelอาจจะขี้ขลาดความเรียบง่ายก็น่าทึ่ง
สเวน

11

สร้างSpliteratorจากการIteratorใช้Spliteratorsคลาสที่มีมากกว่าหนึ่งฟังก์ชั่นสำหรับการสร้าง spliterator เช่นที่นี่กำลังใช้spliteratorUnknownSizeซึ่งได้รับ iterator เป็นพารามิเตอร์แล้วสร้างกระแสโดยใช้StreamSupport

Spliterator<Model> spliterator = Spliterators.spliteratorUnknownSize(
        iterator, Spliterator.NONNULL);
Stream<Model> stream = StreamSupport.stream(spliterator, false);

1
import com.google.common.collect.Streams;

และใช้Streams.stream(iterator):

Streams.stream(iterator)
       .map(v-> function(v))
       .collect(Collectors.toList());

-4

ใช้ Collections.list(iterator).stream()...


6
ในขณะที่สั้น ๆ นี้มีประสิทธิภาพต่ำกว่ามาก
Olivier Grégoire

2
นี่จะเป็นการวนซ้ำตัววนซ้ำทั้งหมดเป็นออบเจ็กต์ java จากนั้นแปลงเป็นสตรีม ฉันไม่แนะนำมัน
iec2011007

3
นี่ดูเหมือนจะเป็นเพียงการแจกแจงเท่านั้นไม่ใช่ตัววนซ้ำ
john16384

1
ไม่ใช่คำตอบที่น่ากลัวโดยทั่วไปมีประโยชน์ในการเหน็บแนม แต่คำถามที่กล่าวถึงประสิทธิภาพและคำตอบนั้นไม่ใช่นักแสดง
เลื่อน
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.