Package org.apache.pinot.segment.local.upsert

Class ConcurrentMapPartitionUpsertMetadataManager

java.lang.Object

org.apache.pinot.segment.local.upsert.ConcurrentMapPartitionUpsertMetadataManager

All Implemented Interfaces:

Closeable, AutoCloseable, PartitionUpsertMetadataManager

@ThreadSafe
public class ConcurrentMapPartitionUpsertMetadataManager
extends Object
implements PartitionUpsertMetadataManager

Implementation of PartitionUpsertMetadataManager that is backed by a ConcurrentHashMap.

Constructor Summary

Constructors

Constructor	Description
ConcurrentMapPartitionUpsertMetadataManager(String tableNameWithType, int partitionId, List<String> primaryKeyColumns, String comparisonColumn, HashFunction hashFunction, PartialUpsertHandler partialUpsertHandler, ServerMetrics serverMetrics)

Method Summary

Modifier and Type	Method	Description
void	addRecord(MutableSegment segment, RecordInfo recordInfo)	Updates the upsert metadata for a new consumed record in the given consuming segment.
void	addSegment(ImmutableSegment segment)	Initializes the upsert metadata for the given immutable segment.
void	close()
List<String>	getPrimaryKeyColumns()	Returns the primary key columns.
void	removeSegment(IndexSegment segment)	Removes the upsert metadata for the given segment.
void	replaceSegment(ImmutableSegment segment, IndexSegment oldSegment)	Replaces the upsert metadata for the old segment with the new immutable segment.
GenericRow	updateRecord(GenericRow record, RecordInfo recordInfo)	Returns the merged record when partial-upsert is enabled.

Methods inherited from class java.lang.Object

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

Constructor Detail

ConcurrentMapPartitionUpsertMetadataManager

public ConcurrentMapPartitionUpsertMetadataManager(String tableNameWithType,
                                                   int partitionId,
                                                   List<String> primaryKeyColumns,
                                                   String comparisonColumn,
                                                   HashFunction hashFunction,
                                                   @Nullable
                                                   PartialUpsertHandler partialUpsertHandler,
                                                   ServerMetrics serverMetrics)

Method Detail

getPrimaryKeyColumns

public List<String> getPrimaryKeyColumns()

Description copied from interface: PartitionUpsertMetadataManager

Returns the primary key columns.

Specified by:

getPrimaryKeyColumns in interface PartitionUpsertMetadataManager

addSegment

public void addSegment(ImmutableSegment segment)

Description copied from interface: PartitionUpsertMetadataManager

Initializes the upsert metadata for the given immutable segment.

Specified by:

addSegment in interface PartitionUpsertMetadataManager

addRecord

public void addRecord(MutableSegment segment,
                      RecordInfo recordInfo)

Description copied from interface: PartitionUpsertMetadataManager

Updates the upsert metadata for a new consumed record in the given consuming segment.

Specified by:

addRecord in interface PartitionUpsertMetadataManager

replaceSegment

public void replaceSegment(ImmutableSegment segment,
                           IndexSegment oldSegment)

Description copied from interface: PartitionUpsertMetadataManager

Replaces the upsert metadata for the old segment with the new immutable segment.

Specified by:

replaceSegment in interface PartitionUpsertMetadataManager

removeSegment

public void removeSegment(IndexSegment segment)

Description copied from interface: PartitionUpsertMetadataManager

Removes the upsert metadata for the given segment.

Specified by:

removeSegment in interface PartitionUpsertMetadataManager

updateRecord

public GenericRow updateRecord(GenericRow record,
                               RecordInfo recordInfo)

Description copied from interface: PartitionUpsertMetadataManager

Returns the merged record when partial-upsert is enabled.

Specified by:

updateRecord in interface PartitionUpsertMetadataManager

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Closeable