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); }