Prometheus Proxy/io.prometheus.proxy/AgentContextManager

AgentContextManager

internal class AgentContextManager(isTestMode: Boolean)(source)

Registry for connected agents and in-flight chunked scrape contexts.

Manages two concurrent maps: one mapping agent IDs to their AgentContext instances, and another mapping scrape IDs to ChunkedContext instances for chunked transfers. Provides lookup, addition, removal, stale-agent detection, and bulk invalidation of agent contexts.

Parameters

isTestMode

when true, suppresses verbose logging during tests

See also

AgentContextCleanupService

Constructors

AgentContextManager

constructor(isTestMode: Boolean)

Types

object Companion

Properties

agentContextEntries

val agentContextEntries: Set<Map.Entry<String, AgentContext>>

agentContextSize

val agentContextSize: Int

chunkedContextMapView

val chunkedContextMapView: Map<Long, ChunkedContext>

chunkedContextSize

val chunkedContextSize: Int

totalAgentScrapeRequestBacklogSize

val totalAgentScrapeRequestBacklogSize: Int

Functions

addAgentContext

fun addAgentContext(agentContext: AgentContext): AgentContext?

findStaleAgents

fun findStaleAgents(maxInactivity: Duration): List<Pair<String, AgentContext>>

getAgentContext

fun getAgentContext(agentId: String): AgentContext?

getChunkedContext

fun getChunkedContext(scrapeId: Long): ChunkedContext?

invalidateAllAgentContexts

fun invalidateAllAgentContexts()

putAgentContext

internal fun putAgentContext(agentId: String, context: AgentContext)

putChunkedContext

fun putChunkedContext(scrapeId: Long, context: ChunkedContext)

removeChunkedContext

fun removeChunkedContext(scrapeId: Long): ChunkedContext?

removeFromContextManager

fun removeFromContextManager(agentId: String, reason: String): AgentContext?