SizedUniformByteGeneratorGenerator.java

package com.ldbc.driver.generator;

import java.util.Iterator;

public class SizedUniformByteGeneratorGenerator extends Generator<Iterator<Byte>> {
    private final Iterator<Long> lengths;
    private final GeneratorFactory gf;

    SizedUniformByteGeneratorGenerator(Iterator<Long> lengths, GeneratorFactory gf) {
        this.lengths = lengths;
        this.gf = gf;
    }

    @Override
    protected Iterator<Byte> doNext() throws GeneratorException {
        return gf.limit(gf.uniformBytes(), lengths.next());
    }

}