Class ArrayBucket

libs/dex/src/main/java/io/crate/data/ArrayBucket.java:27–85

import java.util.Iterator;

public class ArrayBucket implements Bucket {

    private final Object[][] rows;
    private final int numColumns;

    /**
     * Constructs a new ArrayBucket with rows of the given size, regardless of what length the row arrays is.
     *
     * @param rows       the backing array of this bucket
     * @param numColumns the size of rows emitted from this bucket
     */
    public ArrayBucket(Object[][] rows, int numColumns) {
        this.rows = rows;
        this.numColumns = numColumns;
    }

    public ArrayBucket(Object[][] rows) {
        this.rows = rows;
        if (rows.length > 0) {
            numColumns = rows[0].length;
        } else {
            numColumns = -1;
        }
    }

    @Override
    public int size() {
        return rows.length;
    }

    @Override
    public Iterator<Row> iterator() {
        return new Iterator<Row>() {
            int pos = 0;
            final RowN row = new RowN(numColumns);

            @Override
            public boolean hasNext() {
                return pos < rows.length;
            }

            @Override
            public Row next() {
                row.cells(rows[pos++]);
                return row;
            }

            @Override
            public void remove() {

            }
        };
    }

    @Override
    public String toString() {
        return "ArrayBucket{numRows=" + rows.length + ", numColumns=" + numColumns + '}';
    }