org.apache.beam.sdk.io.gcp.bigquery

Class StorageApiWriteUnshardedRecords<DestinationT,ElementT>

java.lang.Object

org.apache.beam.sdk.transforms.PTransform<PCollection<KV<DestinationT,StorageApiWritePayload>>,PCollectionTuple>

org.apache.beam.sdk.io.gcp.bigquery.StorageApiWriteUnshardedRecords<DestinationT,ElementT>

All Implemented Interfaces:

java.io.Serializable, HasDisplayData

public class StorageApiWriteUnshardedRecords<DestinationT,ElementT>
extends PTransform<PCollection<KV<DestinationT,StorageApiWritePayload>>,PCollectionTuple>

Write records to the Storage API using a standard batch approach. PENDING streams are used, which do not become visible until they are finalized and committed. Each input bundle to the DoFn creates a stream per output table, appends all records in the bundle to the stream, and schedules a finalize/commit operation at the end.

See Also:

Serialized Form

Field Summary

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

annotations, displayData, name, resourceHints

Constructor Summary

Constructors

Constructor and Description

StorageApiWriteUnshardedRecords(org.apache.beam.sdk.io.gcp.bigquery.StorageApiDynamicDestinations<ElementT,DestinationT> dynamicDestinations, BigQueryServices bqServices, TupleTag<BigQueryStorageApiInsertError> failedRowsTag, @Nullable TupleTag<TableRow> successfulRowsTag, java.util.function.Predicate<java.lang.String> successfulRowsPredicate, Coder<BigQueryStorageApiInsertError> failedRowsCoder, Coder<TableRow> successfulRowsCoder, boolean autoUpdateSchema, boolean ignoreUnknownValues, BigQueryIO.Write.CreateDisposition createDisposition, @Nullable java.lang.String kmsKey, boolean usesCdc, com.google.cloud.bigquery.storage.v1.AppendRowsRequest.MissingValueInterpretation defaultMissingValueInterpretation, @Nullable java.util.Map<java.lang.String,java.lang.String> bigLakeConfiguration)

Method Summary

Modifier and Type	Method and Description
PCollectionTuple	expand(PCollection<KV<DestinationT,StorageApiWritePayload>> input) Override this method to specify how this PTransform should be expanded on the given InputT.

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

addAnnotation, compose, compose, getAdditionalInputs, getAnnotations, getDefaultOutputCoder, getDefaultOutputCoder, getDefaultOutputCoder, getKindString, getName, getResourceHints, populateDisplayData, setDisplayData, setResourceHints, toString, validate, validate

Methods inherited from class java.lang.Object

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

Constructor Detail

StorageApiWriteUnshardedRecords

public StorageApiWriteUnshardedRecords(org.apache.beam.sdk.io.gcp.bigquery.StorageApiDynamicDestinations<ElementT,DestinationT> dynamicDestinations,
                                       BigQueryServices bqServices,
                                       TupleTag<BigQueryStorageApiInsertError> failedRowsTag,
                                       @Nullable TupleTag<TableRow> successfulRowsTag,
                                       java.util.function.Predicate<java.lang.String> successfulRowsPredicate,
                                       Coder<BigQueryStorageApiInsertError> failedRowsCoder,
                                       Coder<TableRow> successfulRowsCoder,
                                       boolean autoUpdateSchema,
                                       boolean ignoreUnknownValues,
                                       BigQueryIO.Write.CreateDisposition createDisposition,
                                       @Nullable java.lang.String kmsKey,
                                       boolean usesCdc,
                                       com.google.cloud.bigquery.storage.v1.AppendRowsRequest.MissingValueInterpretation defaultMissingValueInterpretation,
                                       @Nullable java.util.Map<java.lang.String,java.lang.String> bigLakeConfiguration)

Method Detail

expand

public PCollectionTuple expand(PCollection<KV<DestinationT,StorageApiWritePayload>> input)

Description copied from class: PTransform

Override this method to specify how this PTransform should be expanded on the given InputT.

NOTE: This method should not be called directly. Instead apply the PTransform should be applied to the InputT using the apply method.

Composite transforms, which are defined in terms of other transforms, should return the output of one of the composed transforms. Non-composite transforms, which do not apply any transforms internally, should return a new unbound output and register evaluators (via backend-specific registration methods).

Specified by:

expand in class PTransform<PCollection<KV<DestinationT,StorageApiWritePayload>>,PCollectionTuple>