org.apache.beam.it.gcp.artifacts.utils

Class ParquetTestUtil

java.lang.Object

org.apache.beam.it.gcp.artifacts.utils.ParquetTestUtil

public class ParquetTestUtil
extends java.lang.Object

The ParquetTestUtil class provides common utilities used for executing tests that involve Parquet.

Constructor Summary

Constructors

Constructor and Description
ParquetTestUtil()

Method Summary

Modifier and Type	Method and Description
static byte[]	createParquetFile(Schema schema, java.util.List<GenericRecord> records) Create Parquet file for the given schema and records list.
static java.util.List<GenericRecord>	readRecords(byte[] contents) Read Parquet records to a list of GenericRecord.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ParquetTestUtil

public ParquetTestUtil()

Method Detail

createParquetFile

public static byte[] createParquetFile(Schema schema,
                                       java.util.List<GenericRecord> records)
                                throws java.io.IOException

Create Parquet file for the given schema and records list.

Parameters:

schema - Schema to use.

records - Records to write to the sink.

Returns:

Byte array that represents the content.

Throws:

java.io.IOException

readRecords

public static java.util.List<GenericRecord> readRecords(byte[] contents)
                                                 throws java.io.IOException

Read Parquet records to a list of GenericRecord.

Parameters:

contents - Byte array with contents to read.

Returns:

A list with all records.

Throws:

java.io.IOException