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

Class DetectNewPartitionsDoFn

java.lang.Object

org.apache.beam.sdk.transforms.DoFn<PartitionMetadata,PartitionMetadata>

org.apache.beam.sdk.io.gcp.spanner.changestreams.dofn.DetectNewPartitionsDoFn

All Implemented Interfaces:

java.io.Serializable, HasDisplayData

@DoFn.UnboundedPerElement
public class DetectNewPartitionsDoFn
extends DoFn<PartitionMetadata,PartitionMetadata>

A SplittableDoFn (SDF) that is responsible for scheduling partitions to be queried. This component will periodically scan the partition metadata table looking for partitions in the PartitionMetadata.State.CREATED, update their state to PartitionMetadata.State.SCHEDULED and output them to the next stage in the pipeline.

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.beam.sdk.transforms.DoFn

DoFn.AlwaysFetched, DoFn.BoundedPerElement, DoFn.BundleFinalizer, DoFn.Element, DoFn.FieldAccess, DoFn.FinishBundle, DoFn.FinishBundleContext, DoFn.GetInitialRestriction, DoFn.GetInitialWatermarkEstimatorState, DoFn.GetRestrictionCoder, DoFn.GetSize, DoFn.GetWatermarkEstimatorStateCoder, DoFn.Key, DoFn.MultiOutputReceiver, DoFn.NewTracker, DoFn.NewWatermarkEstimator, DoFn.OnTimer, DoFn.OnTimerContext, DoFn.OnTimerFamily, DoFn.OnWindowExpiration, DoFn.OnWindowExpirationContext, DoFn.OutputReceiver<T>, DoFn.ProcessContext, DoFn.ProcessContinuation, DoFn.ProcessElement, DoFn.RequiresStableInput, DoFn.RequiresTimeSortedInput, DoFn.Restriction, DoFn.Setup, DoFn.SideInput, DoFn.SplitRestriction, DoFn.StartBundle, DoFn.StartBundleContext, DoFn.StateId, DoFn.Teardown, DoFn.TimerFamily, DoFn.TimerId, DoFn.Timestamp, DoFn.TruncateRestriction, DoFn.UnboundedPerElement, DoFn.WatermarkEstimatorState, DoFn.WindowedContext

Constructor Summary

Constructors

Constructor and Description
DetectNewPartitionsDoFn(DaoFactory daoFactory, MapperFactory mapperFactory, ActionFactory actionFactory, ChangeStreamMetrics metrics) This class needs a DaoFactory to build DAOs to access the partition metadata tables.

Method Summary

Modifier and Type	Method and Description
Instant	getInitialWatermarkEstimatorState(PartitionMetadata partition)
double	getSize(TimestampRange restriction)
TimestampRange	initialRestriction(PartitionMetadata partition) Uses an TimestampRange with a max range.
DetectNewPartitionsRangeTracker	newTracker(TimestampRange restriction)
ManualWatermarkEstimator<Instant>	newWatermarkEstimator(Instant watermarkEstimatorState)
DoFn.ProcessContinuation	processElement(RestrictionTracker<TimestampRange,com.google.cloud.Timestamp> tracker, DoFn.OutputReceiver<PartitionMetadata> receiver, ManualWatermarkEstimator<Instant> watermarkEstimator) Main processing function for the DetectNewPartitionsDoFn function.
void	setAveragePartitionBytesSize(long averagePartitionBytesSize) Sets the average partition bytes size to estimate the backlog of this DoFn.
void	setup() Obtains the instance of DetectNewPartitionsAction.

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

getAllowedTimestampSkew, getInputTypeDescriptor, getOutputTypeDescriptor, populateDisplayData, prepareForProcessing

Methods inherited from class java.lang.Object

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

Constructor Detail

DetectNewPartitionsDoFn

public DetectNewPartitionsDoFn(DaoFactory daoFactory,
                               MapperFactory mapperFactory,
                               ActionFactory actionFactory,
                               ChangeStreamMetrics metrics)

This class needs a DaoFactory to build DAOs to access the partition metadata tables. It uses mappers to transform database rows into the PartitionMetadata model. It builds the delegating action class using the ActionFactory. It emits metrics for the partitions read using the ChangeStreamMetrics. It re-schedules the process element function to be executed according to the default resume interval as in DEFAULT_RESUME_DURATION (best effort).

Parameters:

daoFactory - the DaoFactory to construct PartitionMetadataDaos

mapperFactory - the MapperFactory to construct PartitionMetadataMappers

actionFactory - the ActionFactory to construct actions

metrics - the ChangeStreamMetrics to emit partition related metrics

Method Detail

getInitialWatermarkEstimatorState

@DoFn.GetInitialWatermarkEstimatorState
public Instant getInitialWatermarkEstimatorState(@DoFn.Element
                                                                                         PartitionMetadata partition)

newWatermarkEstimator

@DoFn.NewWatermarkEstimator
public ManualWatermarkEstimator<Instant> newWatermarkEstimator(@DoFn.WatermarkEstimatorState
                                                                                           Instant watermarkEstimatorState)

initialRestriction

@DoFn.GetInitialRestriction
public TimestampRange initialRestriction(@DoFn.Element
                                                                     PartitionMetadata partition)

Uses an TimestampRange with a max range. This is because it does not know beforehand how many partitions it will schedule.

Returns:

the timestamp range for the component

getSize

@DoFn.GetSize
public double getSize(@DoFn.Restriction
                                    TimestampRange restriction)

newTracker

@DoFn.NewTracker
public DetectNewPartitionsRangeTracker newTracker(@DoFn.Restriction
                                                                   TimestampRange restriction)

setup

@DoFn.Setup
public void setup()

Obtains the instance of DetectNewPartitionsAction.

processElement

@DoFn.ProcessElement
public DoFn.ProcessContinuation processElement(RestrictionTracker<TimestampRange,com.google.cloud.Timestamp> tracker,
                                                                    DoFn.OutputReceiver<PartitionMetadata> receiver,
                                                                    ManualWatermarkEstimator<Instant> watermarkEstimator)

Main processing function for the DetectNewPartitionsDoFn function. It will delegate to the DetectNewPartitionsAction class.

setAveragePartitionBytesSize

public void setAveragePartitionBytesSize(long averagePartitionBytesSize)

Sets the average partition bytes size to estimate the backlog of this DoFn. Must be called after the initialization of this DoFn.

Parameters:

averagePartitionBytesSize - the estimated average size of a partition record used in the backlog bytes calculation (DoFn.GetSize)