使用Google数据存储区以流模式检查项目

Question

如果项目不存在，我想将项目插入Google Datastore。 我写了一个数据流流媒体作业。

作业

static class RawToObjectConverter extends DoFn<String, Entity> {
    @Override
    public void processElement(ProcessContext c) {      

        Query<Entity> query = Query.entityQueryBuilder().kind("Post").filter(PropertyFilter.eq("postid", rq.postid))
                .build();
        QueryResults<Entity> posts = datastore.run(query);

        if (posts == null || !posts.hasNext()) {
            Entity post = Entity.builder(datastore.newKeyFactory().newKey("Post"))                 
                     .set("postid", "1")
                    .set("title", "p1")
                    .build();
            c.output(post);
        }           
    }
}

问题

lines.apply(ParDo.of(new RawToObjectConverter()))
    .apply(DatastoreIO.v1().write().withProjectId(projectid));

apply(PTransform<? super PCollection<Entity>,OutputT>)类型中的方法PCollection<Entity>不适用于参数（DatastoreV1.Write）

我还应该使用com.google.cloud.datastore.Datastore还是com.google.datastore.v1.Entity？

Answer 1

您需要将 com.google.cloud.datastore.Entity 转换为 RawToObjectConverter com.google.datastore.v1.Entity >在使用 DatastoreIO.v1（）。write（）之前， 使用toPb方法

c.output(post.toPb());