io.nosqlbench.virtdata.userlibs.streams.fillers.ByteBufferFillable Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of driver-cql-shaded Show documentation

A Shaded CQL ActivityType driver for http://nosqlbench.io/

There is a newer version: 4.15.102

package io.nosqlbench.virtdata.userlibs.streams.fillers;

import java.nio.ByteBuffer;
import java.util.Iterator;

/**
 * If a type implements this interface, then it wants to be provided with a data source in the form of a series of byte
 * buffers. The source must provide additional byte buffers as they are requested. These byte buffers aren't required to
 * be any particular size. They aren't required to have distinct or special data. All that is required is that they
 * never run out.
 */
public interface ByteBufferFillable extends Fillable {
    void fill(Iterable source);

    static void fillByteBuffer(ByteBuffer target, Iterable source) {
        Iterator iterator = source.iterator();
        target.clear();

        while (target.remaining() > 0 && iterator.hasNext()) {
            ByteBuffer next = iterator.next();
            if (next.remaining() > target.remaining()) {
                byte[] bytes = new byte[target.remaining()];
                next.get(bytes, 0, target.remaining());
                target.put(bytes);
            } else {
                target.put(next);
            }
            if (target.remaining() == 0) {
                break;
            }
        }
        target.flip();
    }

}