org.apache.beam.sdk.io.gcp.datastore

Class DatastoreV1.Read

java.lang.Object

org.apache.beam.sdk.transforms.PTransform<PBegin,PCollection<com.google.datastore.v1.Entity>>

org.apache.beam.sdk.io.gcp.datastore.DatastoreV1.Read

All Implemented Interfaces:

Serializable, HasDisplayData

Enclosing class:

DatastoreV1

public abstract static class DatastoreV1.Read
extends PTransform<PBegin,PCollection<com.google.datastore.v1.Entity>>

A PTransform that reads the result rows of a Cloud Datastore query as Entity objects.

See Also:

DatastoreIO, Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static int	NUM_QUERY_SPLITS_MAX An upper bound on the number of splits for a query.

Fields inherited from class org.apache.beam.sdk.transforms.PTransform

name

Constructor Summary

Constructors

Constructor and Description
Read()

Method Summary

Modifier and Type	Method and Description
PCollection<com.google.datastore.v1.Entity>	expand(PBegin input)
abstract ValueProvider<String>	getLiteralGqlQuery()
abstract String	getLocalhost()
abstract ValueProvider<String>	getNamespace()
abstract int	getNumQuerySplits()
abstract ValueProvider<String>	getProjectId()
abstract com.google.datastore.v1.Query	getQuery()
void	populateDisplayData(DisplayData.Builder builder)
abstract String	toString()
DatastoreV1.Read	withLiteralGqlQuery(String gqlQuery) Returns a new DatastoreV1.Read that reads the results of the specified GQL query.
DatastoreV1.Read	withLiteralGqlQuery(ValueProvider<String> gqlQuery) Same as withLiteralGqlQuery(String) but with a ValueProvider.
DatastoreV1.Read	withLocalhost(String localhost) Returns a new DatastoreV1.Read that reads from a Datastore Emulator running at the given localhost address.
DatastoreV1.Read	withNamespace(String namespace) Returns a new DatastoreV1.Read that reads from the given namespace.
DatastoreV1.Read	withNamespace(ValueProvider<String> namespace) Same as withNamespace(String) but with a ValueProvider.
DatastoreV1.Read	withNumQuerySplits(int numQuerySplits) Returns a new DatastoreV1.Read that reads by splitting the given query into numQuerySplits.
DatastoreV1.Read	withProjectId(String projectId) Returns a new DatastoreV1.Read that reads from the Cloud Datastore for the specified project.
DatastoreV1.Read	withProjectId(ValueProvider<String> projectId) Same as withProjectId(String) but with a ValueProvider.
DatastoreV1.Read	withQuery(com.google.datastore.v1.Query query) Returns a new DatastoreV1.Read that reads the results of the specified query.

Methods inherited from class org.apache.beam.sdk.transforms.PTransform

getAdditionalInputs, getDefaultOutputCoder, getDefaultOutputCoder, getDefaultOutputCoder, getKindString, getName, validate

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Field Detail

NUM_QUERY_SPLITS_MAX

public static final int NUM_QUERY_SPLITS_MAX

An upper bound on the number of splits for a query.

See Also:

Constant Field Values

Constructor Detail

Read

public Read()

Method Detail

getProjectId

@Nullable
public abstract ValueProvider<String> getProjectId()

getQuery

@Nullable
public abstract com.google.datastore.v1.Query getQuery()

getLiteralGqlQuery

@Nullable
public abstract ValueProvider<String> getLiteralGqlQuery()

getNamespace

@Nullable
public abstract ValueProvider<String> getNamespace()

getNumQuerySplits

public abstract int getNumQuerySplits()

getLocalhost

@Nullable
public abstract String getLocalhost()

toString

public abstract String toString()

Overrides:

toString in class PTransform<PBegin,PCollection<com.google.datastore.v1.Entity>>

withProjectId

public DatastoreV1.Read withProjectId(String projectId)

Returns a new DatastoreV1.Read that reads from the Cloud Datastore for the specified project.

withProjectId

public DatastoreV1.Read withProjectId(ValueProvider<String> projectId)

Same as withProjectId(String) but with a ValueProvider.

withQuery

public DatastoreV1.Read withQuery(com.google.datastore.v1.Query query)

Returns a new DatastoreV1.Read that reads the results of the specified query.

Note: Normally, DatastoreIO will read from Cloud Datastore in parallel across many workers. However, when the Query is configured with a limit using Query.Builder#setLimit, then all results will be read by a single worker in order to ensure correct results.

withLiteralGqlQuery

@Experimental(value=SOURCE_SINK)
public DatastoreV1.Read withLiteralGqlQuery(String gqlQuery)

Returns a new DatastoreV1.Read that reads the results of the specified GQL query. See GQL Reference to know more about GQL grammar.

Note: This query is executed with literals allowed, so the users should ensure that the query is originated from trusted sources to avoid any security vulnerabilities via SQL Injection.

Experimental: Cloud Datastore does not a provide a clean way to translate a gql query string to Query, so we end up making a query to the service for translation but this may read the actual data, although it will be a small amount. It needs more validation through production use cases before marking it as stable.

withLiteralGqlQuery

@Experimental(value=SOURCE_SINK)
public DatastoreV1.Read withLiteralGqlQuery(ValueProvider<String> gqlQuery)

Same as withLiteralGqlQuery(String) but with a ValueProvider.

withNamespace

public DatastoreV1.Read withNamespace(String namespace)

Returns a new DatastoreV1.Read that reads from the given namespace.

withNamespace

public DatastoreV1.Read withNamespace(ValueProvider<String> namespace)

Same as withNamespace(String) but with a ValueProvider.

withNumQuerySplits

public DatastoreV1.Read withNumQuerySplits(int numQuerySplits)

Returns a new DatastoreV1.Read that reads by splitting the given query into numQuerySplits.

The semantics for the query splitting is defined below:

• Any value less than or equal to 0 will be ignored, and the number of splits will be chosen dynamically at runtime based on the query data size.
• Any value greater than NUM_QUERY_SPLITS_MAX will be capped at NUM_QUERY_SPLITS_MAX.
• If the query has a user limit set, then numQuerySplits will be ignored and no split will be performed.
• Under certain cases Cloud Datastore is unable to split query to the requested number of splits. In such cases we just use whatever the Cloud Datastore returns.

withLocalhost

public DatastoreV1.Read withLocalhost(String localhost)

Returns a new DatastoreV1.Read that reads from a Datastore Emulator running at the given localhost address.

expand

public PCollection<com.google.datastore.v1.Entity> expand(PBegin input)

Specified by:

expand in class PTransform<PBegin,PCollection<com.google.datastore.v1.Entity>>

populateDisplayData

public void populateDisplayData(DisplayData.Builder builder)

Specified by:

populateDisplayData in interface HasDisplayData

Overrides:

populateDisplayData in class PTransform<PBegin,PCollection<com.google.datastore.v1.Entity>>