Examples with BinaryRowWriter org.apache.flink.table.data.writer.BinaryRowWriter used on opensource projects

Example 46 with BinaryRowWriter

use of org.apache.flink.table.data.writer.BinaryRowWriter in project flink by apache.

the class RowDataSerializer method toBinaryRow.

/**
 * Convert {@link RowData} into {@link BinaryRowData}. TODO modify it to code gen.
 */
@Override
public BinaryRowData toBinaryRow(RowData row) {
    if (row instanceof BinaryRowData) {
        return (BinaryRowData) row;
    }
    if (reuseRow == null) {
        reuseRow = new BinaryRowData(types.length);
        reuseWriter = new BinaryRowWriter(reuseRow);
    }
    reuseWriter.reset();
    reuseWriter.writeRowKind(row.getRowKind());
    for (int i = 0; i < types.length; i++) {
        if (row.isNullAt(i)) {
            reuseWriter.setNullAt(i);
        } else {
            BinaryWriter.write(reuseWriter, i, fieldGetters[i].getFieldOrNull(row), types[i], fieldSerializers[i]);
        }
    }
    reuseWriter.complete();
    return reuseRow;
}

Example 47 with BinaryRowWriter

use of org.apache.flink.table.data.writer.BinaryRowWriter in project flink by apache.

the class BytesMultiMapTestBase method genValues.

private BinaryRowData[] genValues(int num) {
    BinaryRowData[] values = new BinaryRowData[num];
    final Random rnd = new Random(RANDOM_SEED);
    for (int i = 0; i < num; i++) {
        values[i] = new BinaryRowData(2);
        BinaryRowWriter writer = new BinaryRowWriter(values[i]);
        writer.writeString(0, StringData.fromString("string" + rnd.nextInt()));
        writer.writeInt(1, rnd.nextInt());
        writer.complete();
    }
    return values;
}

Example 48 with BinaryRowWriter

use of org.apache.flink.table.data.writer.BinaryRowWriter in project flink by apache.

the class StreamRecordUtils method binaryrow.

/**
 * Receives a object array, generates a BinaryRowData based on the array.
 *
 * @param fields input object array
 * @return generated BinaryRowData.
 */
public static BinaryRowData binaryrow(Object... fields) {
    BinaryRowData row = new BinaryRowData(fields.length);
    BinaryRowWriter writer = new BinaryRowWriter(row);
    for (int j = 0; j < fields.length; j++) {
        Object value = fields[j];
        if (value == null) {
            writer.setNullAt(j);
        } else if (value instanceof Byte) {
            writer.writeByte(j, (Byte) value);
        } else if (value instanceof Short) {
            writer.writeShort(j, (Short) value);
        } else if (value instanceof Integer) {
            writer.writeInt(j, (Integer) value);
        } else if (value instanceof String) {
            writer.writeString(j, StringData.fromString((String) value));
        } else if (value instanceof Double) {
            writer.writeDouble(j, (Double) value);
        } else if (value instanceof Float) {
            writer.writeFloat(j, (Float) value);
        } else if (value instanceof Long) {
            writer.writeLong(j, (Long) value);
        } else if (value instanceof Boolean) {
            writer.writeBoolean(j, (Boolean) value);
        } else if (value instanceof byte[]) {
            writer.writeBinary(j, (byte[]) value);
        } else if (value instanceof DecimalData) {
            DecimalData decimal = (DecimalData) value;
            writer.writeDecimal(j, decimal, decimal.precision());
        } else if (value instanceof TimestampData) {
            TimestampData timestamp = (TimestampData) value;
            writer.writeTimestamp(j, timestamp, 3);
        } else if (value instanceof Tuple2 && ((Tuple2) value).f0 instanceof TimestampData) {
            TimestampData timestamp = (TimestampData) ((Tuple2) value).f0;
            writer.writeTimestamp(j, timestamp, (int) ((Tuple2) value).f1);
        } else if (value instanceof Tuple2 && ((Tuple2) value).f0 instanceof ArrayData) {
            ArrayData array = (ArrayData) ((Tuple2) value).f0;
            ArrayDataSerializer serializer = (ArrayDataSerializer) ((Tuple2) value).f1;
            writer.writeArray(j, array, serializer);
        } else if (value instanceof Tuple2 && ((Tuple2) value).f0 instanceof RowData) {
            RowData rowData = ((RowData) ((Tuple2) value).f0);
            RowDataSerializer serializer = (RowDataSerializer) ((Tuple2) value).f1;
            writer.writeRow(j, rowData, serializer);
        } else {
            throw new RuntimeException("Not support yet!");
        }
    }
    writer.complete();
    return row;
}

Example 49 with BinaryRowWriter

use of org.apache.flink.table.data.writer.BinaryRowWriter in project flink by apache.

the class SliceAssignerTestBase method row.

protected static RowData row(long timestamp) {
    BinaryRowData binaryRowData = new BinaryRowData(1);
    BinaryRowWriter writer = new BinaryRowWriter(binaryRowData);
    writer.writeTimestamp(0, TimestampData.fromEpochMillis(timestamp), 3);
    writer.complete();
    return binaryRowData;
}

Example 50 with BinaryRowWriter

use of org.apache.flink.table.data.writer.BinaryRowWriter in project flink by apache.

the class BinaryRowSerializerTest method createRow.

private static BinaryRowData createRow(String f0, int f1) {
    BinaryRowData row = new BinaryRowData(2);
    BinaryRowWriter writer = new BinaryRowWriter(row);
    writer.writeString(0, StringData.fromString(f0));
    writer.writeInt(1, f1);
    writer.complete();
    return row;
}