org.apache.beam.sdk.io.gcp.spanner

Class SpannerIO.Write

java.lang.Object

org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<com.google.cloud.spanner.Mutation>,SpannerWriteResult>

org.apache.beam.sdk.io.gcp.spanner.SpannerIO.Write

All Implemented Interfaces:

java.io.Serializable, org.apache.beam.sdk.transforms.display.HasDisplayData

Enclosing class:

SpannerIO

@Experimental(value=SOURCE_SINK)
public abstract static class SpannerIO.Write
extends org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<com.google.cloud.spanner.Mutation>,SpannerWriteResult>

A PTransform that writes Mutation objects to Google Cloud Spanner.

See Also:

SpannerIO, Serialized Form

Field Summary

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

name

Constructor Summary

Constructors

Constructor and Description
Write()

Method Summary

Modifier and Type	Method and Description
SpannerWriteResult	expand(org.apache.beam.sdk.values.PCollection<com.google.cloud.spanner.Mutation> input)
SpannerIO.WriteGrouped	grouped() Same transform but can be applied to PCollection of MutationGroup.
void	populateDisplayData(org.apache.beam.sdk.transforms.display.DisplayData.Builder builder)
SpannerIO.Write	withBatchSizeBytes(long batchSizeBytes) Specifies the batch size limit (max number of bytes mutated per batch).
SpannerIO.Write	withDatabaseId(java.lang.String databaseId) Specifies the Cloud Spanner database.
SpannerIO.Write	withDatabaseId(org.apache.beam.sdk.options.ValueProvider<java.lang.String> databaseId) Specifies the Cloud Spanner database.
SpannerIO.Write	withFailureMode(SpannerIO.FailureMode failureMode) Specifies failure mode.
SpannerIO.Write	withGroupingFactor(int groupingFactor) Specifies the multiple of max mutation (in terms of both bytes per batch and cells per batch) that is used to select a set of mutations to sort by key for batching.
SpannerIO.Write	withHost(java.lang.String host) Specifies the Cloud Spanner host.
SpannerIO.Write	withHost(org.apache.beam.sdk.options.ValueProvider<java.lang.String> host) Specifies the Cloud Spanner host.
SpannerIO.Write	withInstanceId(java.lang.String instanceId) Specifies the Cloud Spanner instance.
SpannerIO.Write	withInstanceId(org.apache.beam.sdk.options.ValueProvider<java.lang.String> instanceId) Specifies the Cloud Spanner instance.
SpannerIO.Write	withMaxNumMutations(long maxNumMutations) Specifies the cell mutation limit (maximum number of mutated cells per batch).
SpannerIO.Write	withProjectId(java.lang.String projectId) Specifies the Cloud Spanner project.
SpannerIO.Write	withProjectId(org.apache.beam.sdk.options.ValueProvider<java.lang.String> projectId) Specifies the Cloud Spanner project.
SpannerIO.Write	withSchemaReadySignal(org.apache.beam.sdk.values.PCollection signal) Specifies an optional input PCollection that can be used as the signal for Wait.OnSignal to indicate when the database schema is ready to be read.
SpannerIO.Write	withSpannerConfig(SpannerConfig spannerConfig) Specifies the Cloud Spanner configuration.

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

compose, getAdditionalInputs, getDefaultOutputCoder, getDefaultOutputCoder, getDefaultOutputCoder, getKindString, getName, toString, validate

Methods inherited from class java.lang.Object

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

Constructor Detail

Write

public Write()

Method Detail

withSpannerConfig

public SpannerIO.Write withSpannerConfig(SpannerConfig spannerConfig)

Specifies the Cloud Spanner configuration.

withProjectId

public SpannerIO.Write withProjectId(java.lang.String projectId)

Specifies the Cloud Spanner project.

withProjectId

public SpannerIO.Write withProjectId(org.apache.beam.sdk.options.ValueProvider<java.lang.String> projectId)

Specifies the Cloud Spanner project.

withInstanceId

public SpannerIO.Write withInstanceId(java.lang.String instanceId)

Specifies the Cloud Spanner instance.

withInstanceId

public SpannerIO.Write withInstanceId(org.apache.beam.sdk.options.ValueProvider<java.lang.String> instanceId)

Specifies the Cloud Spanner instance.

withDatabaseId

public SpannerIO.Write withDatabaseId(java.lang.String databaseId)

Specifies the Cloud Spanner database.

withDatabaseId

public SpannerIO.Write withDatabaseId(org.apache.beam.sdk.options.ValueProvider<java.lang.String> databaseId)

Specifies the Cloud Spanner database.

withHost

public SpannerIO.Write withHost(org.apache.beam.sdk.options.ValueProvider<java.lang.String> host)

Specifies the Cloud Spanner host.

withHost

public SpannerIO.Write withHost(java.lang.String host)

Specifies the Cloud Spanner host.

grouped

public SpannerIO.WriteGrouped grouped()

Same transform but can be applied to PCollection of MutationGroup.

withBatchSizeBytes

public SpannerIO.Write withBatchSizeBytes(long batchSizeBytes)

Specifies the batch size limit (max number of bytes mutated per batch). Default value is 1MB

withFailureMode

public SpannerIO.Write withFailureMode(SpannerIO.FailureMode failureMode)

Specifies failure mode. SpannerIO.FailureMode.FAIL_FAST mode is selected by default.

withMaxNumMutations

public SpannerIO.Write withMaxNumMutations(long maxNumMutations)

Specifies the cell mutation limit (maximum number of mutated cells per batch). Default value is 5000

withSchemaReadySignal

public SpannerIO.Write withSchemaReadySignal(org.apache.beam.sdk.values.PCollection signal)

Specifies an optional input PCollection that can be used as the signal for Wait.OnSignal to indicate when the database schema is ready to be read.

To be used when the database schema is created by another section of the pipeline, this causes this transform to wait until the signal PCollection has been closed before reading the schema from the database.

See Also:

Wait.OnSignal

withGroupingFactor

public SpannerIO.Write withGroupingFactor(int groupingFactor)

Specifies the multiple of max mutation (in terms of both bytes per batch and cells per batch) that is used to select a set of mutations to sort by key for batching. This sort uses local memory on the workers, so using large values can cause out of memory errors. Default value is 1000.

expand

public SpannerWriteResult expand(org.apache.beam.sdk.values.PCollection<com.google.cloud.spanner.Mutation> input)

Specified by:

expand in class org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<com.google.cloud.spanner.Mutation>,SpannerWriteResult>

populateDisplayData

public void populateDisplayData(org.apache.beam.sdk.transforms.display.DisplayData.Builder builder)

Specified by:

populateDisplayData in interface org.apache.beam.sdk.transforms.display.HasDisplayData

Overrides:

populateDisplayData in class org.apache.beam.sdk.transforms.PTransform<org.apache.beam.sdk.values.PCollection<com.google.cloud.spanner.Mutation>,SpannerWriteResult>