Search in sources :

Example 91 with Record

use of org.apache.avro.generic.GenericData.Record in project beam by apache.

the class BigQueryIOStorageQueryTest method createRecord.

private static GenericRecord createRecord(String name, long number, Schema schema) {
    GenericRecord genericRecord = new Record(schema);
    genericRecord.put("name", name);
    genericRecord.put("number", number);
    return genericRecord;
}
Also used : Record(org.apache.avro.generic.GenericData.Record) GenericRecord(org.apache.avro.generic.GenericRecord) GenericRecord(org.apache.avro.generic.GenericRecord)

Example 92 with Record

use of org.apache.avro.generic.GenericData.Record in project voldemort by voldemort.

the class AvroBackwardsCompatibilityTest method testAvroSchemaEvolution.

/*
     * This tests if a client tries to deserialize an object created using an
     * old schema is successful or not
     */
@Test
public void testAvroSchemaEvolution() throws IOException {
    String versionZero = "{\"type\": \"record\", \"name\": \"myrec\",\"fields\": [{ \"name\": \"original\", \"type\": \"string\" }]}";
    String versionOne = "{\"type\": \"record\", \"name\": \"myrec\",\"fields\": [{ \"name\": \"original\", \"type\": \"string\" } ," + "{ \"name\": \"new-field\", \"type\": \"string\", \"default\":\"\" }]}";
    Schema s0 = Schema.parse(versionZero);
    Schema s1 = Schema.parse(versionOne);
    Map<Integer, String> versions = new HashMap<Integer, String>();
    versions.put(0, versionZero);
    versions.put(1, versionOne);
    byte[] versionZeroBytes = writeVersion0(s0);
    GenericData.Record record = (Record) readVersion0(versions, versionZeroBytes);
}
Also used : Record(org.apache.avro.generic.GenericData.Record) HashMap(java.util.HashMap) Schema(org.apache.avro.Schema) Record(org.apache.avro.generic.GenericData.Record) GenericData(org.apache.avro.generic.GenericData) Test(org.junit.Test)

Example 93 with Record

use of org.apache.avro.generic.GenericData.Record in project voldemort by voldemort.

the class AvroBackwardsCompatibilityTest method writeVersion0.

private static byte[] writeVersion0(Schema s0) {
    GenericData.Record record = new GenericData.Record(s0);
    record.put("original", new Utf8("Abhinay"));
    AvroVersionedGenericSerializer serializer = new AvroVersionedGenericSerializer(s0.toString());
    return serializer.toBytes(record);
}
Also used : Record(org.apache.avro.generic.GenericData.Record) Utf8(org.apache.avro.util.Utf8) Record(org.apache.avro.generic.GenericData.Record) GenericData(org.apache.avro.generic.GenericData)

Example 94 with Record

use of org.apache.avro.generic.GenericData.Record in project voldemort by voldemort.

the class AvroGenericSerializerTest method testRoundtripAvroWithGenericRecord.

public void testRoundtripAvroWithGenericRecord() throws Exception {
    String jsonSchema = "{\"name\": \"Compact Disk\", \"type\": \"record\", " + "\"fields\": [" + "{\"name\": \"name\", \"type\": \"string\", \"order\": \"ascending\"}" + "]}";
    AvroGenericSerializer serializer = new AvroGenericSerializer(jsonSchema);
    Record record = new Record(Schema.parse(jsonSchema));
    // we need to use a Utf8 instance to map to a String.
    record.put("name", new Utf8("Hello"));
    byte[] bytes = serializer.toBytes(record);
    assertTrue(serializer.toObject(bytes).equals(record));
}
Also used : Utf8(org.apache.avro.util.Utf8) Record(org.apache.avro.generic.GenericData.Record)

Example 95 with Record

use of org.apache.avro.generic.GenericData.Record in project alluxio by Alluxio.

the class ParquetReader method create.

/**
 * Creates a parquet reader.
 *
 * @param uri the URI to the input
 * @return the reader
 * @throws IOException when failed to create the reader
 */
public static ParquetReader create(AlluxioURI uri) throws IOException {
    Path inputPath = new JobPath(uri.getScheme(), uri.getAuthority().toString(), uri.getPath());
    Configuration conf = ReadWriterUtils.readNoCacheConf();
    InputFile inputFile = HadoopInputFile.fromPath(inputPath, conf);
    org.apache.parquet.hadoop.ParquetReader<Record> reader = AvroParquetReader.<Record>builder(inputFile).disableCompatibility().withDataModel(GenericData.get()).withConf(conf).build();
    Schema schema;
    ParquetMetadata footer;
    try (ParquetFileReader r = new ParquetFileReader(inputFile, ParquetReadOptions.builder().build())) {
        footer = r.getFooter();
        schema = new AvroSchemaConverter().convert(footer.getFileMetaData().getSchema());
    }
    return new ParquetReader(reader, schema, footer);
}
Also used : JobPath(alluxio.job.plan.transform.format.JobPath) Path(org.apache.hadoop.fs.Path) AvroSchemaConverter(org.apache.parquet.avro.AvroSchemaConverter) JobPath(alluxio.job.plan.transform.format.JobPath) Configuration(org.apache.hadoop.conf.Configuration) ParquetMetadata(org.apache.parquet.hadoop.metadata.ParquetMetadata) Schema(org.apache.avro.Schema) TableSchema(alluxio.job.plan.transform.format.TableSchema) ParquetFileReader(org.apache.parquet.hadoop.ParquetFileReader) AvroParquetReader(org.apache.parquet.avro.AvroParquetReader) InputFile(org.apache.parquet.io.InputFile) HadoopInputFile(org.apache.parquet.hadoop.util.HadoopInputFile) Record(org.apache.avro.generic.GenericData.Record)

Aggregations

Record (org.apache.avro.generic.GenericData.Record)96 Test (org.junit.Test)44 IndexedRecord (org.apache.avro.generic.IndexedRecord)43 Schema (org.apache.avro.Schema)33 ArrayList (java.util.ArrayList)24 GenericRecord (org.apache.avro.generic.GenericRecord)14 Field (org.apache.avro.Schema.Field)11 List (java.util.List)10 GenericData (org.apache.avro.generic.GenericData)10 TestRunner (org.apache.nifi.util.TestRunner)8 GenericRecordBuilder (org.apache.avro.generic.GenericRecordBuilder)7 JsonObject (com.google.gson.JsonObject)6 DataFileStream (org.apache.avro.file.DataFileStream)6 DataFileWriter (org.apache.avro.file.DataFileWriter)6 GenericDatumReader (org.apache.avro.generic.GenericDatumReader)6 Utf8 (org.apache.avro.util.Utf8)6 TMarketoOutputProperties (org.talend.components.marketo.tmarketooutput.TMarketoOutputProperties)6 ActivityRecord (com.marketo.mktows.ActivityRecord)5 ArrayOfLeadRecord (com.marketo.mktows.ArrayOfLeadRecord)5 LeadChangeRecord (com.marketo.mktows.LeadChangeRecord)5