Package com.linkedin.davinci.stats

Class AbstractVeniceAggVersionedStats<STATS,STATS_REPORTER extends AbstractVeniceStatsReporter<STATS>>

java.lang.Object

com.linkedin.davinci.stats.AbstractVeniceAggVersionedStats<STATS,STATS_REPORTER>

All Implemented Interfaces:

StoreDataChangedListener

Direct Known Subclasses:

AggVersionedBlobTransferStats, AggVersionedDaVinciRecordTransformerStats, AggVersionedDIVStats, AggVersionedIngestionStats, AggVersionedStorageEngineStats, HeartbeatVersionedStats

public abstract class AbstractVeniceAggVersionedStats<STATS,STATS_REPORTER extends AbstractVeniceStatsReporter<STATS>>
extends Object
implements StoreDataChangedListener

Field Summary

Fields

Modifier and Type	Field	Description
protected final ReadOnlyStoreRepository	metadataRepository

Constructor Summary

Constructors

Constructor	Description
AbstractVeniceAggVersionedStats(io.tehuti.metrics.MetricsRepository metricsRepository, ReadOnlyStoreRepository metadataRepository, Supplier<STATS> statsInitiator, StatsSupplier<STATS_REPORTER> reporterSupplier, boolean unregisterMetricForDeletedStoreEnabled)

Method Summary

Modifier and Type	Method	Description
protected VeniceVersionedStats<STATS,STATS_REPORTER>	addStore(Store store)	Adds a store and initializes its version info.
protected void	cleanupVersionResources(String storeName, int version)	Hook method for subclasses to clean up version-specific resources (e.g., OTel stats) when a version is removed.
protected int	getCurrentVersion(String storeName)	return Store.NON_EXISTING_VERSION if current version doesn't exist.
protected int	getFutureVersion(String storeName)	return Store.NON_EXISTING_VERSION if future version doesn't exist.
protected io.tehuti.metrics.MetricsRepository	getMetricsRepository()
protected STATS	getStats(String storeName, int version)
protected STATS	getTotalStats(String storeName)
void	handleStoreChanged(Store store)
void	handleStoreCreated(Store store)	Do NOT try to acquire the lock of store repository again in the implementation, otherwise a dead lock issue will happen.
void	handleStoreDeleted(String storeName)
void	loadAllStats()
protected void	onVersionInfoUpdated(String storeName, int currentVersion, int futureVersion)	Hook method called when version info is updated for a store.
protected void	recordVersionedAndTotalStat(String storeName, int version, Consumer<STATS> function)
protected void	registerConditionalStats(String storeName)
protected void	updateStatsVersionInfo(String storeName, List<Version> existingVersions, int newCurrentVersion)
protected void	updateTotalStats(String storeName)	Some versioned stats might always increasing; in this case, the value in the total stats should be updated with the aggregated values across the new version list.

Methods inherited from class java.lang.Object

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

Methods inherited from interface com.linkedin.venice.meta.StoreDataChangedListener

handleStoreDeleted

Field Details

metadataRepository

protected final ReadOnlyStoreRepository metadataRepository

Constructor Details

AbstractVeniceAggVersionedStats

public AbstractVeniceAggVersionedStats(io.tehuti.metrics.MetricsRepository metricsRepository,
 ReadOnlyStoreRepository metadataRepository,
 Supplier<STATS> statsInitiator,
 StatsSupplier<STATS_REPORTER> reporterSupplier,
 boolean unregisterMetricForDeletedStoreEnabled)

Method Details

getMetricsRepository

protected io.tehuti.metrics.MetricsRepository getMetricsRepository()

loadAllStats

public void loadAllStats()

recordVersionedAndTotalStat

protected void recordVersionedAndTotalStat(String storeName,
 int version,
 Consumer<STATS> function)

getTotalStats

protected STATS getTotalStats(String storeName)

getStats

protected STATS getStats(String storeName,
 int version)

registerConditionalStats

protected void registerConditionalStats(String storeName)

addStore

protected VeniceVersionedStats<STATS,STATS_REPORTER> addStore(Store store)

Adds a store and initializes its version info. Uses computeIfAbsent for thread-safety, ensuring version info is always initialized exactly once when the store is first added.

updateStatsVersionInfo

protected void updateStatsVersionInfo(String storeName,
 List<Version> existingVersions,
 int newCurrentVersion)

handleStoreCreated

public void handleStoreCreated(Store store)

Description copied from interface: StoreDataChangedListener

Do NOT try to acquire the lock of store repository again in the implementation, otherwise a dead lock issue will happen.

Specified by:

handleStoreCreated in interface StoreDataChangedListener

handleStoreDeleted

public void handleStoreDeleted(String storeName)

Specified by:

handleStoreDeleted in interface StoreDataChangedListener

handleStoreChanged

public void handleStoreChanged(Store store)

Specified by:

handleStoreChanged in interface StoreDataChangedListener

getFutureVersion

protected int getFutureVersion(String storeName)

return Store.NON_EXISTING_VERSION if future version doesn't exist.

getCurrentVersion

protected int getCurrentVersion(String storeName)

return Store.NON_EXISTING_VERSION if current version doesn't exist.

updateTotalStats

protected void updateTotalStats(String storeName)

Some versioned stats might always increasing; in this case, the value in the total stats should be updated with the aggregated values across the new version list.

onVersionInfoUpdated

protected void onVersionInfoUpdated(String storeName,
 int currentVersion,
 int futureVersion)

Hook method called when version info is updated for a store. Subclasses can override this to update existing OTel stats with new version info.

WARNING: This method may be called from within aggStats.computeIfAbsent in addStore(Store). Implementations MUST NOT call getCurrentVersion(java.lang.String), getFutureVersion(java.lang.String), or getVersionedStats(java.lang.String) — these re-enter aggStats and cause a deadlock or IllegalStateException. Use computeIfPresent on the subclass's own OTel stats map to update existing entries only.

cleanupVersionResources

protected void cleanupVersionResources(String storeName,
 int version)

Hook method for subclasses to clean up version-specific resources (e.g., OTel stats) when a version is removed. Same re-entrance warning as onVersionInfoUpdated(java.lang.String, int, int).