Examples of GenericRecordBuilder

Examples of org.apache.avro.generic.GenericRecordBuilder

79
80
81
82
83
84
85
86
  protected GenericData.Record newStringRecord(String text) {
    return new GenericRecordBuilder(STRING_SCHEMA).set("text", text).build();
  }

  protected GenericData.Record newStatsRecord(int count, String name) {
    return new GenericRecordBuilder(STATS_SCHEMA).set("count", count)
        .set("name", name).build();
  }
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

52
53
54
55
56
57
58
59
60
61
62
    AvroParquetWriter<GenericRecord> writer =
        new AvroParquetWriter<GenericRecord>(file, schema);

    // Write a record with a map with Utf8 keys.
    GenericData.Record record = new GenericRecordBuilder(schema)
        .set("mymap", new HashMap(ImmutableMap.of(utf8("a"), 1, utf8("b"), 2)))
        .build();
    writer.write(record);
    writer.close();
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
    Path file = new Path(tmp.getPath());
   
    AvroParquetWriter<GenericRecord> writer = new
        AvroParquetWriter<GenericRecord>(file, schema);

    GenericData.Record nestedRecord = new GenericRecordBuilder(
        schema.getField("mynestedrecord").schema())
            .set("mynestedint", 1).build();

    List<Integer> integerArray = Arrays.asList(1, 2, 3);
    GenericData.Array<Integer> genericIntegerArray = new GenericData.Array<Integer>(
        Schema.createArray(Schema.create(Schema.Type.INT)), integerArray);

    GenericFixed genericFixed = new GenericData.Fixed(
        Schema.createFixed("fixed", null, null, 1), new byte[] { (byte) 65 });

    List<Integer> emptyArray = new ArrayList<Integer>();
    ImmutableMap emptyMap = new ImmutableMap.Builder<String, Integer>().build();

    GenericData.Record record = new GenericRecordBuilder(schema)
        .set("mynull", null)
        .set("myboolean", true)
        .set("myint", 1)
        .set("mylong", 2L)
        .set("myfloat", 3.1f)
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

79
80
81
82
83
84
85
86
87
88
89
    writer.close();
  }

  @Test
  public void testWriter() throws IOException {
    Record record = new GenericRecordBuilder(USER_SCHEMA)
        .set("username", "test1").set("email", "a@example.com").build();
    try {
      writer.initialize();
      writer.write(record);
      writer.flush();
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

93
94
95
96
97
98
99
100
101
102
103
    }
  }

  @Test(expected = IllegalStateException.class)
  public void testWriteToClosedWriterFails() throws IOException {
    Record record = new GenericRecordBuilder(USER_SCHEMA)
        .set("username", "test1").set("email", "a@example.com").build();
    writer.initialize();
    writer.close();
    writer.write(record);
  }
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
    Path datasetPath = new Path("file:" + testDirectory + "/provided/users");

    final Dataset<GenericRecord> users = Datasets.create(
        "dataset:" + datasetPath, descriptor);

    final GenericRecord u1 = new GenericRecordBuilder(user)
        .set("username", "test1")
        .set("email", "a@example.com")
        .build();
    GenericRecord u2 = new GenericRecordBuilder(user)
        .set("username", "test2")
        .set("email", "b@example.com")
        .build();

    TestHelpers.assertThrows("Should reject write with unknown version",
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

152
153
154
155
156
157
158
159
160
161
162
163
164
165
  private void writeRecord(Dataset<GenericRecord> dataset, int partition) {
    PartitionKey key = new PartitionKey(partition);
    DatasetWriter<GenericRecord> writer =
        ((PartitionedDataset<GenericRecord>) dataset).getPartition(key, true).newWriter();
    try {
      GenericRecordBuilder recordBuilder = new GenericRecordBuilder(
          dataset.getDescriptor().getSchema())
          .set("username", partition + "").set("email", partition + "@example.com");
      writer.write(recordBuilder.build());
    } finally {
      writer.close();
    }

  }
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
  public static void writeTestUsers(Dataset<GenericData.Record> ds, int count, int start, String... fields) {
    DatasetWriter<GenericData.Record> writer = null;
    try {
      writer = ds.newWriter();
      for (int i = start; i < count + start; i++) {
        GenericRecordBuilder recordBuilder = new GenericRecordBuilder(ds.getDescriptor
            ().getSchema()).set("username", "test-" + i);
        for (String field : fields) {
          recordBuilder.set(field, field + "-" + i);
        }
        writer.write(recordBuilder.build());
      }
      writer.flush();
    } finally {
      if (writer != null) {
        writer.close();
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
    writer.append("2,user,user@example.com\n");
    writer.close();

    TestUtil.run("csv-schema", sample, "-o", avsc, "--class", "User");

    GenericRecordBuilder builder = new GenericRecordBuilder(
        new Schema.Parser().parse(new File(avsc)));
    builder.set("id", 1l);
    builder.set("username", "test");
    builder.set("email", "test@example.com");
    expected.add(builder.build());
    builder.set("id", 2l);
    builder.set("username", "user");
    builder.set("email", "user@example.com");
    expected.add(builder.build());
  }
View Full Code Here

Examples of org.apache.avro.generic.GenericRecordBuilder

68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
    writer.append("2,user,user@example.com\n");
    writer.close();

    TestUtil.run("-v", "csv-schema", sample, "-o", avsc, "--class", "User");

    GenericRecordBuilder builder = new GenericRecordBuilder(
        new Schema.Parser().parse(new File(avsc)));
    builder.set("id", 1l);
    builder.set("username", "test");
    builder.set("email", "test@example.com");
    expected.add(builder.build());
    builder.set("id", 2l);
    builder.set("username", "user");
    builder.set("email", "user@example.com");
    expected.add(builder.build());
  }
View Full Code Here
TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.