使用 Apache Beam 将重复的字符串写入 BigQuery

Question

我有一个数据 stream 包含看起来像JSONArrays的Strings 。 我想解析这些字符串并使用 Apache Beam 写入 BigQuery 表，但在写入重复字符串时出现错误。

这是我将字符串转换为TableRow的方法：

    String dataString = "[{\"EMAIL\": [\"zog@yahoo.com\"]}]";

    JSONArray jsonArray = new JSONArray(dataString);
    TableRow tableRow = new TableRow();

    for (int i = 0; i < jsonArray.length(); i++) {
      JSONArray emailArray = new JSONArray(jsonArray.getJSONObject(i).get("EMAIL").toString());

      tableRow.set("EMAIL", emailArray); //Results in error
    }

这是我的 BigQuery 架构的样子：

[
  {
    "name": "EMAIL",
    "type": "STRING",
    "mode": "REPEATED"
  }
]

我已经设法使用 Python 将类似的重复字符串写入 BigQuery 表，但使用 Apache Beam 无法做到这一点。 我想我没有在TableRow中保存正确的键值对。 我现在得到的错误是：

java.io.IOException: Insert failed: [{"errors":[{"debugInfo":"","location":"email","message":"This field is not a record.","reason":"invalid"}],"index":0}]

我需要有关如何在不创建记录的情况下将类似的重复字符串保存到 BigQuery 的帮助，并且希望获得任何建议或建议。 提前致谢。

Answer 1

看来你想创建

1. 一行包含 email 地址的串联字符串，或
2. 每个 email 一行，或
3. 具有重复字段的一行。

请注意，您的ValidFrom字段似乎是STRING类型，而不是重复字段，除非它包含在分层架构中的重复字段中。

在您提供的示例代码中，您正在创建一个JSONArray并将其放入STRING字段，我认为这会导致问题，因为类型不兼容。 如果要将其保留为纯STRING字段，可以使用下面的解决方案 1。

还要确保 BigQuery 中的列名称与代码中的名称相匹配，我看到您同时使用ValidFrom和EMAIL （尽管您发布的代码中可能有错误）。

解决方案 1：一行字符串字段

如果您想在 BigQuery 中添加一行具有串联String字段的行，可以使用以下命令：

// Initialize your final row
TableRow tableRow = new TableRow();

// Find email addresses
String [] emails = ... // your extraction logic

// Build a concatenated string of emails
String allEmails = String.join(";", emails);

// Add the string field to the row
tableRow.set('EMAILS', allEmails);

解决方案 2：带有字符串字段的多行

如果您想插入多行，您可以创建多个表行：

// Find email addresses
String [] emails = ... // your extraction logic

// Build a row per email
for(String email: emails) {
    // Initialize your final row
    TableRow tableRow = new TableRow();
    tableRow.set('EMAIL', email);
    
    // TODO: do something with the row (add to list, or ...)
}

解决方案 3：一行包含 REPEATED 字段

如果您想在 BigQuery 中添加一行带有REPEATED STRING字段的行，您可以使用以下命令：

// Initialize your final row
TableRow tableRow = new TableRow();

// Find email addresses
String [] emails = ... // your extraction logic

// Build the repeated field
List<String> emailCells = new ArrayList<>();
for(String email: emails) {
    emailCells.add(email);
}

// Add the repeated field to the row
tableRow.set('EMAILS', emailCells);

如果这不是您的目标，请提供更多详细信息。