通过Google Cloud Dataflow将PubSub消息插入BigQuery

时间:2015-09-17 14:15:52

标签: google-bigquery google-cloud-dataflow google-cloud-pubsub

我想使用Google Cloud Dataflow将来自主题的PubSub消息数据插入到BigQuery表中。 一切都很好,但在BigQuery表中我可以看到像“߈ ”这样难以理解的字符串。 这是我的管道:

p.apply(PubsubIO.Read.named("ReadFromPubsub").topic("projects/project-name/topics/topic-name"))
.apply(ParDo.named("Transformation").of(new StringToRowConverter()))
.apply(BigQueryIO.Write.named("Write into BigQuery").to("project-name:dataset-name.table")
     .withSchema(schema)
     .withCreateDisposition(BigQueryIO.Write.CreateDisposition.CREATE_IF_NEEDED))

我的简单StringToRowConverter函数是:

class StringToRowConverter extends DoFn<String, TableRow> {
private static final long serialVersionUID = 0;

@Override
public void processElement(ProcessContext c) {
    for (String word : c.element().split(",")) {
      if (!word.isEmpty()) {
          System.out.println(word);
        c.output(new TableRow().set("data", word));
      }
    }
}
}

这是我通过POST请求发送的消息:

POST https://pubsub.googleapis.com/v1/projects/project-name/topics/topic-name:publish
{
 "messages": [
  {
   "attributes":{
"key": "tablet, smartphone, desktop",
"value": "eng"
   },
   "data": "34gf5ert"
  }
 ]
}

我错过了什么? 谢谢!

2 个答案:

答案 0 :(得分:7)

根据https://cloud.google.com/pubsub/reference/rest/v1/PubsubMessage,pubsub消息的JSON有效负载是base64编码的。默认情况下,Dataflow中的PubsubIO使用String UTF8编码器。你提供“34gf5ert”的示例字符串,当base64解码然后解释为UTF-8字符串时,恰好给出“߈ ”。

答案 1 :(得分:2)

这就是我解压我的pubsub消息的方法:

@Override
public void processElement(ProcessContext c) {

    String json = c.element();

    HashMap<String,String> items = new Gson().fromJson(json, new TypeToken<HashMap<String, String>>(){}.getType());
    String unpacked = items.get("JsonKey");

希望它对你有用。