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

Class PartitionRestrictionTracker

java.lang.Object

org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<PartitionRestriction,PartitionPosition>

org.apache.beam.sdk.io.gcp.spanner.changestreams.restriction.PartitionRestrictionTracker

All Implemented Interfaces:

org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.HasProgress

public class PartitionRestrictionTracker
extends org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<PartitionRestriction,PartitionPosition>
implements org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.HasProgress

Nested Class Summary

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

org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.HasProgress, org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.IsBounded, org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.Progress, org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.TruncateResult<RestrictionT>

Field Summary

Fields

Modifier and Type	Field and Description
protected PartitionRestriction	restriction
protected java.util.function.Supplier<com.google.cloud.Timestamp>	timeSupplier

Constructor Summary

Constructors

Constructor and Description
PartitionRestrictionTracker(PartitionRestriction restriction)

Method Summary

Modifier and Type	Method and Description
void	checkDone()
PartitionRestriction	currentRestriction()
org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.Progress	getProgress()
org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.IsBounded	isBounded()
void	setTimeSupplier(java.util.function.Supplier<com.google.cloud.Timestamp> timeSupplier)
boolean	tryClaim(PartitionPosition position)
@Nullable org.apache.beam.sdk.transforms.splittabledofn.SplitResult<PartitionRestriction>	trySplit(double fractionOfRemainder)

Methods inherited from class java.lang.Object

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

Field Detail

restriction

protected PartitionRestriction restriction

timeSupplier

protected java.util.function.Supplier<com.google.cloud.Timestamp> timeSupplier

Constructor Detail

PartitionRestrictionTracker

public PartitionRestrictionTracker(PartitionRestriction restriction)

Method Detail

setTimeSupplier

public void setTimeSupplier(java.util.function.Supplier<com.google.cloud.Timestamp> timeSupplier)

trySplit

public @Nullable org.apache.beam.sdk.transforms.splittabledofn.SplitResult<PartitionRestriction> trySplit(double fractionOfRemainder)

Specified by:

trySplit in class org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<PartitionRestriction,PartitionPosition>

tryClaim

public boolean tryClaim(PartitionPosition position)

Specified by:

tryClaim in class org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<PartitionRestriction,PartitionPosition>

getProgress

public org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.Progress getProgress()

Specified by:

getProgress in interface org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.HasProgress

currentRestriction

public PartitionRestriction currentRestriction()

Specified by:

currentRestriction in class org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<PartitionRestriction,PartitionPosition>

checkDone

public void checkDone()
               throws java.lang.IllegalStateException

Specified by:

checkDone in class org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<PartitionRestriction,PartitionPosition>

Throws:

java.lang.IllegalStateException

isBounded

public org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker.IsBounded isBounded()

Specified by:

isBounded in class org.apache.beam.sdk.transforms.splittabledofn.RestrictionTracker<PartitionRestriction,PartitionPosition>