com.google.api.services.bigquery.model.TableRow.set()方法的使用及代码示例

x33g5p2x  于2022-01-30 转载在 其他  
字(9.4k)|赞(0)|评价(0)|浏览(130)

本文整理了Java中com.google.api.services.bigquery.model.TableRow.set()方法的一些代码示例,展示了TableRow.set()的具体用法。这些代码示例主要来源于Github/Stackoverflow/Maven等平台,是从一些精选项目中提取出来的代码,具有较强的参考意义,能在一定程度帮忙到你。TableRow.set()方法的具体详情如下:
包路径:com.google.api.services.bigquery.model.TableRow
类名称:TableRow
方法名:set

TableRow.set介绍

[英]Represents a single row in the result set, consisting of one or more fields.
[中]表示结果集中的一行,由一个或多个字段组成。

代码示例

代码示例来源:origin: GoogleCloudPlatform/DataflowTemplates

@Override
 public TableRow mapRow(ResultSet resultSet) throws Exception {
  ResultSetMetaData metaData = resultSet.getMetaData();
  TableRow outputTableRow = new TableRow();
  for (int i = 1; i <= metaData.getColumnCount(); i++) {
   outputTableRow.set(metaData.getColumnName(i), resultSet.getObject(i));
  }
  return outputTableRow;
 }
}

代码示例来源:origin: org.apache.beam/beam-sdks-java-io-google-cloud-platform

public TableRowBuilder set(String fieldName, Object value) {
 row.set(fieldName, value);
 return this;
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@ProcessElement
public void processElement(ProcessContext c) {
 List<TableRow> completions = new ArrayList<>();
 for (CompletionCandidate cc : c.element().getValue()) {
  completions.add(new TableRow().set("count", cc.getCount()).set("tag", cc.getValue()));
 }
 TableRow row = new TableRow().set("prefix", c.element().getKey()).set("tags", completions);
 c.output(row);
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@ProcessElement
public void processElement(ProcessContext c) {
 LaneInfo laneInfo = c.element().getValue();
 TableRow row =
   new TableRow()
     .set("station_id", c.element().getKey())
     .set("direction", laneInfo.getDirection())
     .set("freeway", laneInfo.getFreeway())
     .set("lane_max_flow", laneInfo.getLaneFlow())
     .set("lane", laneInfo.getLane())
     .set("avg_occ", laneInfo.getLaneAO())
     .set("avg_speed", laneInfo.getLaneAS())
     .set("total_flow", laneInfo.getTotalFlow())
     .set("recorded_timestamp", laneInfo.getRecordedTimestamp())
     .set("window_timestamp", c.timestamp().toString());
 c.output(row);
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@ProcessElement
public void processElement(ProcessContext c) {
 RouteInfo routeInfo = c.element().getValue();
 TableRow row =
   new TableRow()
     .set("avg_speed", routeInfo.getAvgSpeed())
     .set("slowdown_event", routeInfo.getSlowdownEvent())
     .set("route", c.element().getKey())
     .set("window_timestamp", c.timestamp().toString());
 c.output(row);
}

代码示例来源:origin: org.apache.beam/beam-sdks-java-io-google-cloud-platform

private static TableRow convertGenericRecordToTableRow(
  GenericRecord record, List<TableFieldSchema> fields) {
 TableRow row = new TableRow();
 for (TableFieldSchema subSchema : fields) {
  // Per https://cloud.google.com/bigquery/docs/reference/v2/tables#schema, the name field
  // is required, so it may not be null.
  Field field = record.getSchema().getField(subSchema.getName());
  Object convertedValue =
    getTypedCellValue(field.schema(), subSchema, record.get(field.name()));
  if (convertedValue != null) {
   // To match the JSON files exported by BigQuery, do not include null values in the output.
   row.set(field.name(), convertedValue);
  }
 }
 return row;
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@ProcessElement
 public void processElement(ProcessContext c) {
  TableRow row =
    new TableRow().set("word", c.element().getKey()).set("all_plays", c.element().getValue());
  c.output(row);
 }
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@ProcessElement
 public void processElement(ProcessContext c) {
  TableRow row =
    new TableRow()
      .set("month", c.element().getKey())
      .set("tornado_count", c.element().getValue());
  c.output(row);
 }
}

代码示例来源:origin: org.apache.beam/beam-examples-java

/** In this example, put the whole string into single BigQuery field. */
@ProcessElement
public void processElement(ProcessContext c) {
 c.output(new TableRow().set("string_field", c.element()));
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@ProcessElement
 public void processElement(ProcessContext c) {
  TableRow row =
    new TableRow()
      .set("month", c.element().getKey())
      .set("max_mean_temp", c.element().getValue());
  c.output(row);
 }
}

代码示例来源:origin: org.talend.components/bigquery-runtime

@Override
public TableRow convertToDatum(IndexedRecord indexedRecord) {
  // When BigQueryOutput do not specify schema, so read it from the incoming data
  if (schema == null) {
    schema = indexedRecord.getSchema();
    initFieldConverters();
  }
  TableRow row = new TableRow();
  for (Schema.Field field : schema.getFields()) {
    Object v = indexedRecord.get(field.pos());
    if (v != null) {
      row.set(field.name(), fieldConverters.get(field.name()).convertToDatum(v));
    }
  }
  return row;
}

代码示例来源:origin: Talend/components

@Override
public TableRow convertToDatum(IndexedRecord indexedRecord) {
  // When BigQueryOutput do not specify schema, so read it from the incoming data
  if (schema == null) {
    schema = indexedRecord.getSchema();
    initFieldConverters();
  }
  TableRow row = new TableRow();
  for (Schema.Field field : schema.getFields()) {
    Object v = indexedRecord.get(field.pos());
    if (v != null) {
      row.set(field.name(), fieldConverters.get(field.name()).convertToDatum(v));
    }
  }
  return row;
}

代码示例来源:origin: org.apache.beam/beam-sdks-java-io-google-cloud-platform

List<TableRow> convertStringsToLong(List<TableRow> toConvert) {
 // The numbers come back as String after JSON serialization. Change them back to
 // longs so that we can assert the output.
 List<TableRow> converted = Lists.newArrayList();
 for (TableRow entry : toConvert) {
  TableRow convertedEntry = entry.clone();
  convertedEntry.set("number", Long.parseLong((String) convertedEntry.get("number")));
  converted.add(convertedEntry);
 }
 return converted;
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@ProcessElement
 public void processElement(ProcessContext c) throws Exception {
  TableRow element = c.element();
  TableRow row =
    new TableRow()
      .set("trigger_type", element.get("trigger_type"))
      .set("freeway", element.get("freeway"))
      .set("total_flow", element.get("total_flow"))
      .set("number_of_records", element.get("number_of_records"))
      .set("isFirst", element.get("isFirst"))
      .set("isLast", element.get("isLast"))
      .set("timing", element.get("timing"))
      .set("window", element.get("window"));
  c.output(canonicalFormat(row));
 }
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@Test
 @Category(ValidatesRunner.class)
 public void testFormatCounts() {
  PCollection<KV<Integer, Long>> inputs =
    p.apply(Create.of(KV.of(3, 0L), KV.of(4, Long.MAX_VALUE), KV.of(5, Long.MIN_VALUE)));
  PCollection<TableRow> result = inputs.apply(ParDo.of(new FormatCountsFn()));
  PAssert.that(result)
    .containsInAnyOrder(
      new TableRow().set("month", 3).set("tornado_count", 0),
      new TableRow().set("month", 4).set("tornado_count", Long.MAX_VALUE),
      new TableRow().set("month", 5).set("tornado_count", Long.MIN_VALUE));
  p.run().waitUntilFinish();
 }
}

代码示例来源:origin: org.apache.beam/beam-sdks-java-io-google-cloud-platform

@Test
public void testWriteToTableDecorator() throws Exception {
 TableRow row1 = new TableRow().set("name", "a").set("number", "1");
 TableRow row2 = new TableRow().set("name", "b").set("number", "2");
 TableSchema schema =
   new TableSchema()
     .setFields(
       ImmutableList.of(new TableFieldSchema().setName("number").setType("INTEGER")));
 p.apply(Create.of(row1, row2))
   .apply(
     BigQueryIO.writeTableRows()
       .to("project-id:dataset-id.table-id$20171127")
       .withTestServices(fakeBqServices)
       .withMethod(BigQueryIO.Write.Method.STREAMING_INSERTS)
       .withSchema(schema)
       .withoutValidation());
 p.run();
}

代码示例来源:origin: org.apache.beam/beam-sdks-java-io-google-cloud-platform

@Test
public void testCoder_nullCell() throws CoderException {
 TableRow row = new TableRow();
 row.set("temperature", Data.nullOf(Object.class));
 row.set("max_temperature", Data.nullOf(Object.class));
 byte[] bytes = CoderUtils.encodeToByteArray(TableRowJsonCoder.of(), row);
 TableRow newRow = CoderUtils.decodeFromByteArray(TableRowJsonCoder.of(), bytes);
 byte[] newBytes = CoderUtils.encodeToByteArray(TableRowJsonCoder.of(), newRow);
 Assert.assertArrayEquals(bytes, newBytes);
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@Test
@Category(ValidatesRunner.class)
public void testNoTornadoes() {
 TableRow row = new TableRow().set("month", 6).set("tornado", false);
 PCollection<TableRow> inputs = p.apply(Create.of(ImmutableList.of(row)));
 PCollection<Integer> result = inputs.apply(ParDo.of(new ExtractTornadoesFn()));
 PAssert.that(result).empty();
 p.run().waitUntilFinish();
}

代码示例来源:origin: org.apache.beam/beam-examples-java

@Test
@Category(ValidatesRunner.class)
public void testExtractTornadoes() {
 TableRow row = new TableRow().set("month", "6").set("tornado", true);
 PCollection<TableRow> input = p.apply(Create.of(ImmutableList.of(row)));
 PCollection<Integer> result = input.apply(ParDo.of(new ExtractTornadoesFn()));
 PAssert.that(result).containsInAnyOrder(6);
 p.run().waitUntilFinish();
}

代码示例来源:origin: org.apache.beam/beam-sdks-java-io-google-cloud-platform

@Test
public void testWriteWithBrokenGetTable() throws Exception {
 p.apply(Create.<TableRow>of(new TableRow().set("foo", "bar")))
   .apply(
     BigQueryIO.writeTableRows()
       .to(input -> null)
       .withCreateDisposition(BigQueryIO.Write.CreateDisposition.CREATE_NEVER)
       .withTestServices(fakeBqServices)
       .withoutValidation());
 thrown.expectMessage("result of tableFunction can not be null");
 thrown.expectMessage("foo");
 p.run();
}

相关文章