Examples with ArrowSerializer org.apache.flink.table.runtime.arrow.serializers.ArrowSerializer used on opensource projects

Example 1 with ArrowSerializer

use of org.apache.flink.table.runtime.arrow.serializers.ArrowSerializer in project flink by apache.

the class ArrowPythonScalarFunctionOperator method open.

@Override
public void open() throws Exception {
    super.open();
    maxArrowBatchSize = Math.min(pythonConfig.getMaxArrowBatchSize(), maxBundleSize);
    arrowSerializer = new ArrowSerializer(udfInputType, udfOutputType);
    arrowSerializer.open(bais, baos);
    currentBatchCount = 0;
}

Example 2 with ArrowSerializer

use of org.apache.flink.table.runtime.arrow.serializers.ArrowSerializer in project flink by apache.

the class AbstractArrowPythonAggregateFunctionOperator method open.

@SuppressWarnings("unchecked")
@Override
public void open() throws Exception {
    super.open();
    rowDataWrapper = new StreamRecordRowDataWrappingCollector(output);
    reuseJoinedRow = new JoinedRowData();
    udafInputProjection = udafInputGeneratedProjection.newInstance(Thread.currentThread().getContextClassLoader());
    arrowSerializer = new ArrowSerializer(udfInputType, udfOutputType);
    arrowSerializer.open(bais, baos);
    currentBatchCount = 0;
}