使用Lambda函数来消耗供应商提供的所有对象

Question

寻找如何使用Java lambda函数,以便Consumer可以处理供应商提供的所有对象,并摆脱显式while循环和null检查.

我有一个数据库的String键供应商,我想使用Consumer来处理每个键.

Supplier<String> keyGen = new SimpleKeySupplier(keyPrefix, numKeys);
Consumer<String> consumer = (String key) -> System.out.println("key="+key);

我想consumer处理由提供的每个密钥,keyGen并尝试以下.它有效,但我确信必须有一种更简洁的方法来使用lambda函数来简化这一过程.

    // Test that the correct keys have been populated.
    Supplier<String> keyGen = new SimpleKeySupplier(keyPrefix, NumKeys);
    String k = keyGen.get();
    while(k != null) {
        consumer.accept(k);
        k = keyGen.get();
    }

SimpleKeySupplier工作,简化版本如下:

import java.util.function.Supplier;

public class SimpleKeySupplier implements Supplier<String> {
    private final String keyPrefix;
    private final int numToGenerate;
    private       int numGenerated;

    public SimpleKeySupplier(String keyPrefix, int numRecs) {
        this.keyPrefix = keyPrefix;
        numToGenerate  = numRecs;
        numGenerated   = 0;
    }
    @Override
    public String get() {
        if (numGenerated >= numToGenerate) 
            return null; 
        else   
            return (keyPrefix + numGenerated++);
    }
}

此示例中的Consumer大大简化了在StackOverflow上发布.

Answer 1

您可以这样做,并在Java9中添加新功能.

Stream.generate(keyGen).takeWhile(Objects::nonNull).forEach(consumer);

Answer 2

你可以使用Stream:

Stream.generate(keyGen).limit(NumKeys).forEach(consumer);

Stream.generate将创建一个持续呼叫get所提供的供应商的流.

limit使流有限,将其限制为x元素.在这种情况下,我可以看到keyGen只生成NumKeys密钥,所以我从那一点"切断"流.

Answer 3

试试这个

Supplier<String> keyGen = new SimpleKeySupplier(keyPrefix, numKeys);
Consumer<String> consumer = (String key) -> System.out.println("key="+key);
Stream.generate(keyGen).filter(s -> s !=null).limit(NumKeys).forEach(consumer);