Package org.apache.iceberg

Class BaseReplacePartitions

java.lang.Object
org.apache.iceberg.BaseReplacePartitions
All Implemented Interfaces:
PendingUpdate<Snapshot>, ReplacePartitions, SnapshotUpdate<ReplacePartitions>
public class BaseReplacePartitions extends Object implements ReplacePartitions

  • Method Details

    • self

      protected ReplacePartitions self()

    • operation

      protected String operation()
      A string that describes the action that produced the new snapshot.
      Returns:
      a string operation

    • addFile

      public ReplacePartitions addFile(DataFile file)
      Description copied from interface: ReplacePartitions
      Add a DataFile to the table.
      Specified by:
      addFile in interface ReplacePartitions
      Parameters:
      file - a data file
      Returns:
      this for method chaining

    • validateAppendOnly

      public ReplacePartitions validateAppendOnly()
      Description copied from interface: ReplacePartitions
      Validate that no partitions will be replaced and the operation is append-only.
      Specified by:
      validateAppendOnly in interface ReplacePartitions
      Returns:
      this for method chaining

    • validateFromSnapshot

      public ReplacePartitions validateFromSnapshot(long newStartingSnapshotId)
      Description copied from interface: ReplacePartitions
      Set the snapshot ID used in validations for this operation.

      All validations will check changes after this snapshot ID. If this is not called, validation will occur from the beginning of the table's history.

      This method should be called before this operation is committed. If a concurrent operation committed a data or delta file or removed a data file after the given snapshot ID that might contain rows matching a partition marked for deletion, validation will detect this and fail.

      Specified by:
      validateFromSnapshot in interface ReplacePartitions
      Parameters:
      newStartingSnapshotId - a snapshot ID, it should be set to when this operation started to read the table.
      Returns:
      this for method chaining

    • validateNoConflictingDeletes

      public ReplacePartitions validateNoConflictingDeletes()
      Description copied from interface: ReplacePartitions
      Enables validation that deletes that happened concurrently do not conflict with this commit's operation.

      Validating concurrent deletes is required during non-idempotent replace partition operations. This will check if a concurrent operation deletes data in any of the partitions being overwritten, as the replace partition must be aborted to avoid undeleting rows that were removed concurrently.

      Specified by:
      validateNoConflictingDeletes in interface ReplacePartitions
      Returns:
      this for method chaining

    • validateNoConflictingData

      public ReplacePartitions validateNoConflictingData()
      Description copied from interface: ReplacePartitions
      Enables validation that data added concurrently does not conflict with this commit's operation.

      Validating concurrent data files is required during non-idempotent replace partition operations. This will check if a concurrent operation inserts data in any of the partitions being overwritten, as the replace partition must be aborted to avoid removing rows added concurrently.

      Specified by:
      validateNoConflictingData in interface ReplacePartitions
      Returns:
      this for method chaining

    • toBranch

      public BaseReplacePartitions toBranch(String branch)
      Description copied from interface: SnapshotUpdate
      Perform operations on a particular branch
      Specified by:
      toBranch in interface SnapshotUpdate<ReplacePartitions>
      Parameters:
      branch - which is name of SnapshotRef of type branch.

    • validate

      public void validate(TableMetadata currentMetadata, Snapshot parent)
      Validate the current metadata.

      Child operations can override this to add custom validation.

      Parameters:
      currentMetadata - current table metadata to validate
      parent - ending snapshot on the lineage which is being validated

    • apply

      public List<ManifestFile> apply(TableMetadata base, Snapshot snapshot)
      Apply the update's changes to the given metadata and snapshot. Return the new manifest list.
      Parameters:
      base - the base table metadata to apply changes to
      snapshot - snapshot to apply the changes to
      Returns:
      a manifest list for the new snapshot.

    • set

      public ReplacePartitions set(String property, String value)
      Description copied from interface: SnapshotUpdate
      Set a summary property in the snapshot produced by this update.
      Parameters:
      property - a String property name
      value - a String property value
      Returns:
      this for method chaining

    • caseSensitive

      public ReplacePartitions caseSensitive(boolean isCaseSensitive)

    • isCaseSensitive

      protected boolean isCaseSensitive()

    • dataSpec

      protected PartitionSpec dataSpec()

    • rowFilter

      protected Expression rowFilter()

    • addedDataFiles

      protected List<DataFile> addedDataFiles()

    • failAnyDelete

      protected void failAnyDelete()

    • failMissingDeletePaths

      protected void failMissingDeletePaths()

    • deleteByRowFilter

      protected void deleteByRowFilter(Expression expr)
      Add a filter to match files to delete. A file will be deleted if all of the rows it contains match this or any other filter passed to this method.
      Parameters:
      expr - an expression to match rows.

    • dropPartition

      protected void dropPartition(int specId, StructLike partition)
      Add a partition tuple to drop from the table during the delete phase.

    • delete

      protected void delete(DataFile file)
      Add a specific data file to be deleted in the new snapshot.

    • delete

      protected void delete(DeleteFile file)
      Add a specific delete file to be deleted in the new snapshot.

    • delete

      protected void delete(CharSequence path)
      Add a specific data path to be deleted in the new snapshot.

    • deletesDataFiles

      protected boolean deletesDataFiles()

    • deletesDeleteFiles

      protected boolean deletesDeleteFiles()

    • addsDataFiles

      protected boolean addsDataFiles()

    • addsDeleteFiles

      protected boolean addsDeleteFiles()

    • add

      protected void add(DataFile file)
      Add a data file to the new snapshot.

    • add

      protected void add(DeleteFile file)
      Add a delete file to the new snapshot.

    • add

      protected void add(DeleteFile file, long dataSequenceNumber)
      Add a delete file to the new snapshot.

    • add

      protected void add(ManifestFile manifest)
      Add all files in a manifest to the new snapshot.

    • validateAddedDataFiles

      protected void validateAddedDataFiles(TableMetadata base, Long startingSnapshotId, PartitionSet partitionSet, Snapshot parent)
      Validates that no files matching given partitions have been added to the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      partitionSet - a set of partitions to filter new conflicting data files
      parent - ending snapshot on the lineage being validated

    • validateAddedDataFiles

      protected void validateAddedDataFiles(TableMetadata base, Long startingSnapshotId, Expression conflictDetectionFilter, Snapshot parent)
      Validates that no files matching a filter have been added to the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      conflictDetectionFilter - an expression used to find new conflicting data files

    • validateNoNewDeletesForDataFiles

      protected void validateNoNewDeletesForDataFiles(TableMetadata base, Long startingSnapshotId, Iterable<DataFile> dataFiles, Snapshot parent)
      Validates that no new delete files that must be applied to the given data files have been added to the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      dataFiles - data files to validate have no new row deletes
      parent - ending snapshot on the branch being validated

    • validateNoNewDeletesForDataFiles

      protected void validateNoNewDeletesForDataFiles(TableMetadata base, Long startingSnapshotId, Expression dataFilter, Iterable<DataFile> dataFiles, Snapshot parent)
      Validates that no new delete files that must be applied to the given data files have been added to the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      dataFilter - a data filter
      dataFiles - data files to validate have no new row deletes
      parent - ending snapshot on the branch being validated

    • validateNoNewDeleteFiles

      protected void validateNoNewDeleteFiles(TableMetadata base, Long startingSnapshotId, Expression dataFilter, Snapshot parent)
      Validates that no delete files matching a filter have been added to the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      dataFilter - an expression used to find new conflicting delete files
      parent - ending snapshot on the branch being validated

    • validateNoNewDeleteFiles

      protected void validateNoNewDeleteFiles(TableMetadata base, Long startingSnapshotId, PartitionSet partitionSet, Snapshot parent)
      Validates that no delete files matching a partition set have been added to the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      partitionSet - a partition set used to find new conflicting delete files
      parent - ending snapshot on the branch being validated

    • addedDeleteFiles

      protected org.apache.iceberg.DeleteFileIndex addedDeleteFiles(TableMetadata base, Long startingSnapshotId, Expression dataFilter, PartitionSet partitionSet, Snapshot parent)
      Returns matching delete files have been added to the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      dataFilter - an expression used to find delete files
      partitionSet - a partition set used to find delete files
      parent - parent snapshot of the branch

    • validateDeletedDataFiles

      protected void validateDeletedDataFiles(TableMetadata base, Long startingSnapshotId, Expression dataFilter, Snapshot parent)
      Validates that no files matching a filter have been deleted from the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      dataFilter - an expression used to find deleted data files
      parent - ending snapshot on the branch being validated

    • validateDeletedDataFiles

      protected void validateDeletedDataFiles(TableMetadata base, Long startingSnapshotId, PartitionSet partitionSet, Snapshot parent)
      Validates that no files matching a filter have been deleted from the table since a starting snapshot.
      Parameters:
      base - table metadata to validate
      startingSnapshotId - id of the snapshot current at the start of the operation
      partitionSet - a partition set used to find deleted data files
      parent - ending snapshot on the branch being validated

    • setNewDataFilesDataSequenceNumber

      protected void setNewDataFilesDataSequenceNumber(long sequenceNumber)

    • validateDataFilesExist

      protected void validateDataFilesExist(TableMetadata base, Long startingSnapshotId, CharSequenceSet requiredDataFiles, boolean skipDeletes, Expression conflictDetectionFilter, Snapshot parent)

    • summary

      protected Map<String,String> summary()

    • updateEvent

      public Object updateEvent()
      Description copied from interface: PendingUpdate
      Generates update event to notify about metadata changes
      Returns:
      the generated event

    • cleanUncommitted

      protected void cleanUncommitted(Set<ManifestFile> committed)
      Clean up any uncommitted manifests that were created.

      Manifests may not be committed if apply is called more because a commit conflict has occurred. Implementations may keep around manifests because the same changes will be made by both apply calls. This method instructs the implementation to clean up those manifests and passes the paths of the manifests that were actually committed.

      Parameters:
      committed - a set of manifest paths that were actually committed

    • stageOnly

      public ReplacePartitions stageOnly()
      Description copied from interface: SnapshotUpdate
      Called to stage a snapshot in table metadata, but not update the current snapshot id.
      Specified by:
      stageOnly in interface SnapshotUpdate<ThisT>
      Returns:
      this for method chaining

    • scanManifestsWith

      public ReplacePartitions scanManifestsWith(ExecutorService executorService)
      Description copied from interface: SnapshotUpdate
      Use a particular executor to scan manifests. The default worker pool will be used by default.
      Specified by:
      scanManifestsWith in interface SnapshotUpdate<ThisT>
      Parameters:
      executorService - the provided executor
      Returns:
      this for method chaining

    • commitMetrics

      protected CommitMetrics commitMetrics()

    • reportWith

      protected ReplacePartitions reportWith(MetricsReporter newReporter)

    • targetBranch

      protected void targetBranch(String branch)
      A setter for the target branch on which snapshot producer operation should be performed
      Parameters:
      branch - to set as target branch

    • targetBranch

      protected String targetBranch()

    • workerPool

      protected ExecutorService workerPool()

    • deleteWith

      public ReplacePartitions deleteWith(Consumer<String> deleteCallback)
      Description copied from interface: SnapshotUpdate
      Set a callback to delete files instead of the table's default.
      Specified by:
      deleteWith in interface SnapshotUpdate<ThisT>
      Parameters:
      deleteCallback - a String consumer used to delete locations.
      Returns:
      this for method chaining

    • apply

      public Snapshot apply()
      Description copied from interface: PendingUpdate
      Apply the pending changes and return the uncommitted changes for validation.

      This does not result in a permanent update.

      Specified by:
      apply in interface PendingUpdate<ThisT>
      Returns:
      the uncommitted changes that would be committed by calling PendingUpdate.commit()

    • current

      protected TableMetadata current()

    • refresh

      protected TableMetadata refresh()

    • commit

      public void commit()
      Description copied from interface: PendingUpdate
      Apply the pending changes and commit.

      Changes are committed by calling the underlying table's commit method.

      Once the commit is successful, the updated table will be refreshed.

      Specified by:
      commit in interface PendingUpdate<ThisT>

    • cleanAll

      protected void cleanAll()

    • deleteFile

      protected void deleteFile(String path)

    • manifestListPath

      protected OutputFile manifestListPath()

    • newManifestOutputFile

      protected EncryptedOutputFile newManifestOutputFile()

    • newManifestWriter

      protected ManifestWriter<DataFile> newManifestWriter(PartitionSpec spec)

    • newDeleteManifestWriter

      protected ManifestWriter<DeleteFile> newDeleteManifestWriter(PartitionSpec spec)

    • newRollingManifestWriter

      protected RollingManifestWriter<DataFile> newRollingManifestWriter(PartitionSpec spec)

    • newRollingDeleteManifestWriter

      protected RollingManifestWriter<DeleteFile> newRollingDeleteManifestWriter(PartitionSpec spec)

    • newManifestReader

      protected ManifestReader<DataFile> newManifestReader(ManifestFile manifest)

    • newDeleteManifestReader

      protected ManifestReader<DeleteFile> newDeleteManifestReader(ManifestFile manifest)

    • snapshotId

      protected long snapshotId()

    • canInheritSnapshotId

      protected boolean canInheritSnapshotId()

    • cleanupAfterCommit

      protected boolean cleanupAfterCommit()

    • writeDataManifests

      protected List<ManifestFile> writeDataManifests(Collection<DataFile> files, PartitionSpec spec)

    • writeDataManifests

      protected List<ManifestFile> writeDataManifests(Collection<DataFile> files, Long dataSeq, PartitionSpec spec)

    • writeDeleteManifests

      protected List<ManifestFile> writeDeleteManifests(Collection<DeleteFile> files, PartitionSpec spec)