private Datastore getDatastore(PipelineOptions pipelineOptions) {
DatastoreOptions.Builder builder =
new DatastoreOptions.Builder()
.host(host)
.dataset(datasetId)
.initializer(new RetryHttpRequestInitializer());
Credential credential = pipelineOptions.as(GcpOptions.class).getGcpCredential();
if (credential != null) {
builder.credential(credential);
}
return DatastoreFactory.get().create(builder.build());
}
@Override
public DatastoreWriter createWriter(PipelineOptions options) throws Exception {
DatastoreOptions.Builder builder =
new DatastoreOptions.Builder()
.host(sink.host)
.dataset(sink.datasetId)
.initializer(new RetryHttpRequestInitializer());
Credential credential = options.as(GcpOptions.class).getGcpCredential();
if (credential != null) {
builder.credential(credential);
}
Datastore datastore = DatastoreFactory.get().create(builder.build());
return new DatastoreWriter(this, datastore);
}
