Package com.linkedin.davinci.stats.ingestion.heartbeat

Class HeartbeatVersionedStats

java.lang.Object

com.linkedin.davinci.stats.AbstractVeniceAggVersionedStats<HeartbeatStat,HeartbeatStatReporter>

com.linkedin.davinci.stats.ingestion.heartbeat.HeartbeatVersionedStats

All Implemented Interfaces:

StoreDataChangedListener

public class HeartbeatVersionedStats
extends AbstractVeniceAggVersionedStats<HeartbeatStat,HeartbeatStatReporter>

Manages Tehuti and OTel heartbeat/record-level delay stats per store.

OTel stats lifecycle: OTel stats are created lazily on first metric recording via getOrCreateHeartbeatOtelStats(java.lang.String) and getOrCreateRecordLevelDelayOtelStats(java.lang.String). Version info updates from handleStoreChanged(com.linkedin.venice.meta.Store) propagate to existing OTel stats via onVersionInfoUpdated(java.lang.String, int, int) (computeIfPresent). See getOrCreateHeartbeatOtelStats(java.lang.String) for why version info is fetched outside the lambda.

Field Summary

Fields inherited from class com.linkedin.davinci.stats.AbstractVeniceAggVersionedStats

metadataRepository

Constructor Summary

Constructors

Constructor	Description
HeartbeatVersionedStats(io.tehuti.metrics.MetricsRepository metricsRepository, ReadOnlyStoreRepository metadataRepository, Supplier<HeartbeatStat> statsInitiator, StatsSupplier<HeartbeatStatReporter> reporterSupplier, Map<HeartbeatKey,IngestionTimestampEntry> leaderMonitors, Map<HeartbeatKey,IngestionTimestampEntry> followerMonitors, String clusterName)

Method Summary

Modifier and Type	Method	Description
void	emitPerRecordFollowerOtelMetric(String storeName, int version, String region, long delay, boolean isReadyToServe)	Emits a per-record OTel metric for follower record delay (called per record, not aggregated).
void	emitPerRecordLeaderOtelMetric(String storeName, int version, String region, long delay)	Emits a per-record OTel metric for leader record delay (called per record, not aggregated).
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()	No-op: heartbeat stats are loaded lazily when the first heartbeat/record arrives.
protected void	onVersionInfoUpdated(String storeName, int currentVersion, int futureVersion)	Updates version info for existing OTel stats.
void	recordFollowerLag(String storeName, int version, String region, long heartbeatTs, boolean isReadyToServe)
void	recordFollowerRecordLag(String storeName, int version, String region, long recordTs, boolean isReadyToServe)
void	recordLeaderLag(String storeName, int version, String region, long heartbeatTs)
void	recordLeaderRecordLag(String storeName, int version, String region, long recordTs)

Methods inherited from class com.linkedin.davinci.stats.AbstractVeniceAggVersionedStats

addStore, cleanupVersionResources, getCurrentVersion, getFutureVersion, getMetricsRepository, getStats, getTotalStats, recordVersionedAndTotalStat, registerConditionalStats, updateStatsVersionInfo, updateTotalStats

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

Constructor Details

HeartbeatVersionedStats

public HeartbeatVersionedStats(io.tehuti.metrics.MetricsRepository metricsRepository,
 ReadOnlyStoreRepository metadataRepository,
 Supplier<HeartbeatStat> statsInitiator,
 StatsSupplier<HeartbeatStatReporter> reporterSupplier,
 Map<HeartbeatKey,IngestionTimestampEntry> leaderMonitors,
 Map<HeartbeatKey,IngestionTimestampEntry> followerMonitors,
 String clusterName)

Method Details

recordLeaderLag

public void recordLeaderLag(String storeName,
 int version,
 String region,
 long heartbeatTs)

recordFollowerLag

public void recordFollowerLag(String storeName,
 int version,
 String region,
 long heartbeatTs,
 boolean isReadyToServe)

recordLeaderRecordLag

public void recordLeaderRecordLag(String storeName,
 int version,
 String region,
 long recordTs)

recordFollowerRecordLag

public void recordFollowerRecordLag(String storeName,
 int version,
 String region,
 long recordTs,
 boolean isReadyToServe)

loadAllStats

public void loadAllStats()

No-op: heartbeat stats are loaded lazily when the first heartbeat/record arrives.

Overrides:

loadAllStats in class AbstractVeniceAggVersionedStats<HeartbeatStat,HeartbeatStatReporter>

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

Overrides:

handleStoreCreated in class AbstractVeniceAggVersionedStats<HeartbeatStat,HeartbeatStatReporter>

handleStoreDeleted

public void handleStoreDeleted(String storeName)

Specified by:

handleStoreDeleted in interface StoreDataChangedListener

Overrides:

handleStoreDeleted in class AbstractVeniceAggVersionedStats<HeartbeatStat,HeartbeatStatReporter>

handleStoreChanged

public void handleStoreChanged(Store store)

Specified by:

handleStoreChanged in interface StoreDataChangedListener

Overrides:

handleStoreChanged in class AbstractVeniceAggVersionedStats<HeartbeatStat,HeartbeatStatReporter>

onVersionInfoUpdated

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

Updates version info for existing OTel stats. No null guard needed: lazy loading means this is never called from the super() constructor.

Overrides:

onVersionInfoUpdated in class AbstractVeniceAggVersionedStats<HeartbeatStat,HeartbeatStatReporter>

emitPerRecordLeaderOtelMetric

public void emitPerRecordLeaderOtelMetric(String storeName,
 int version,
 String region,
 long delay)

Emits a per-record OTel metric for leader record delay (called per record, not aggregated). Uses get() instead of getOrCreate to avoid synchronization on this hot path. Returns early if stats haven't been initialized yet for this store.

emitPerRecordFollowerOtelMetric

public void emitPerRecordFollowerOtelMetric(String storeName,
 int version,
 String region,
 long delay,
 boolean isReadyToServe)

Emits a per-record OTel metric for follower record delay (called per record, not aggregated). Uses get() instead of getOrCreate to avoid synchronization on this hot path. Returns early if stats haven't been initialized yet for this store.