org.apache.beam.sdk.io.gcp.bigtable.changestreams.action

Class ReadChangeStreamPartitionAction

java.lang.Object

org.apache.beam.sdk.io.gcp.bigtable.changestreams.action.ReadChangeStreamPartitionAction

@Internal
public class ReadChangeStreamPartitionAction
extends java.lang.Object

This class is part of ReadChangeStreamPartitionDoFn SDF.

Constructor Summary

Constructors

Constructor and Description
ReadChangeStreamPartitionAction(MetadataTableDao metadataTableDao, ChangeStreamDao changeStreamDao, ChangeStreamMetrics metrics, ChangeStreamAction changeStreamAction, org.joda.time.Duration heartbeatDuration)

Method Summary

Modifier and Type

Method and Description

org.apache.beam.sdk.transforms.DoFn.ProcessContinuation

run(PartitionRecord partitionRecord, org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<StreamProgress,StreamProgress> tracker, org.apache.beam.sdk.transforms.DoFn.OutputReceiver<org.apache.beam.sdk.values.KV<com.google.protobuf.ByteString,com.google.cloud.bigtable.data.v2.models.ChangeStreamMutation>> receiver, org.apache.beam.sdk.transforms.splittabledofn.ManualWatermarkEstimator<org.joda.time.Instant> watermarkEstimator) Streams changes from a specific partition.

Methods inherited from class java.lang.Object

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

Constructor Detail

ReadChangeStreamPartitionAction

public ReadChangeStreamPartitionAction(MetadataTableDao metadataTableDao,
                                       ChangeStreamDao changeStreamDao,
                                       ChangeStreamMetrics metrics,
                                       ChangeStreamAction changeStreamAction,
                                       org.joda.time.Duration heartbeatDuration)

Method Detail

run

public org.apache.beam.sdk.transforms.DoFn.ProcessContinuation run(PartitionRecord partitionRecord,
                                                                   org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<StreamProgress,StreamProgress> tracker,
                                                                   org.apache.beam.sdk.transforms.DoFn.OutputReceiver<org.apache.beam.sdk.values.KV<com.google.protobuf.ByteString,com.google.cloud.bigtable.data.v2.models.ChangeStreamMutation>> receiver,
                                                                   org.apache.beam.sdk.transforms.splittabledofn.ManualWatermarkEstimator<org.joda.time.Instant> watermarkEstimator)
                                                            throws java.io.IOException

Streams changes from a specific partition. This function is responsible to maintaining the lifecycle of streaming the partition. We delegate to ChangeStreamAction to process individual response from the change stream.

Before we send a request to Cloud Bigtable to stream the partition, we need to perform a few things.

1. Lock the partition. Due to the design of the change streams connector, it is possible that multiple DoFn are started trying to stream the same partition. However, only 1 DoFn should be streaming a partition. So we solve this by using the metadata table as a distributed lock. We attempt to lock the partition for this DoFn's UUID. If it is successful, it means this DoFn is the only one that can stream the partition and continue. Otherwise, terminate this DoFn because another DoFn is streaming this partition already.
2. Process CloseStream if it exists. In order to solve a possible inconsistent state problem, we do not process CloseStream after receiving it. We claim the CloseStream in the RestrictionTracker so it persists after a checkpoint. We checkpoint to flush all the DataChanges. Then on resume, we process the CloseStream. There is only 1 expected Status for CloseStream: Out of Range. Out of Range is returned when the partition has either been split into more partitions or merged into a larger partition. In this case, we write to the metadata table the new partitions' information so that DetectNewPartitionsDoFn can read and output those new partitions to be streamed. We also need to ensure we clean up this partition's metadata to release the lock.
3. Update the metadata table with the watermark and additional debugging info.
4. Stream the partition.

Parameters:

partitionRecord - partition information used to identify this stream

tracker - restriction tracker of ReadChangeStreamPartitionDoFn

receiver - output receiver for ReadChangeStreamPartitionDoFn

watermarkEstimator - watermark estimator ReadChangeStreamPartitionDoFn

Returns:

DoFn.ProcessContinuation.stop() if a checkpoint is required or the stream has completed. Or DoFn.ProcessContinuation.resume() if a checkpoint is required.

Throws:

java.io.IOException - when stream fails.