Package io.kroxylicious.proxy.internal

Class VirtualClusterLifecycle

java.lang.Object

io.kroxylicious.proxy.internal.VirtualClusterLifecycle

public class VirtualClusterLifecycle
extends Object

Manages the lifecycle state of a single virtual cluster.

Thread-safe: state transitions and connection registration share the same monitor, so the state transition → connection snapshot sequence in startDraining() is atomic with respect to registerConnection(ClientConnectionStateMachine). This closes the TOCTOU window where a connection could be registered after the drain snapshot is taken and therefore missed by graceful drain.

Constructor Summary

Constructors

Constructor	Description
VirtualClusterLifecycle(String clusterName, Duration drainTimeout)

Method Summary

Modifier and Type	Method	Description
Set<ClientConnectionStateMachine>	activeConnections()
String	clusterName()
void	deregisterConnection(ClientConnectionStateMachine ccsm)
void	drainComplete()	Transitions from VirtualClusterLifecycleState.Draining to VirtualClusterLifecycleState.Stopped.
CompletableFuture<Void>	drainFuture()	Returns the future that completes when all connections have drained.
void	initializationFailed(Throwable cause)	Transitions from VirtualClusterLifecycleState.Initializing to VirtualClusterLifecycleState.Failed, recording the cause.
void	initializationSucceeded()	Transitions from VirtualClusterLifecycleState.Initializing to VirtualClusterLifecycleState.Serving.
boolean	registerConnection(ClientConnectionStateMachine ccsm)
CompletableFuture<Void>	startDraining()	Transitions from VirtualClusterLifecycleState.Serving to VirtualClusterLifecycleState.Draining and initiates close on all active connections.
VirtualClusterLifecycleState	state()
void	stop()	Transitions to VirtualClusterLifecycleState.Stopped from VirtualClusterLifecycleState.Failed or VirtualClusterLifecycleState.Initializing.

Methods inherited from class java.lang.Object

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

Constructor Details

VirtualClusterLifecycle

public VirtualClusterLifecycle(String clusterName,
 Duration drainTimeout)

Method Details

initializationSucceeded

public void initializationSucceeded()

Transitions from VirtualClusterLifecycleState.Initializing to VirtualClusterLifecycleState.Serving.

initializationFailed

public void initializationFailed(Throwable cause)

Transitions from VirtualClusterLifecycleState.Initializing to VirtualClusterLifecycleState.Failed, recording the cause.

startDraining

public CompletableFuture<Void> startDraining()

Transitions from VirtualClusterLifecycleState.Serving to VirtualClusterLifecycleState.Draining and initiates close on all active connections.

Returns:

future that completes when all connections have closed

drainFuture

public CompletableFuture<Void> drainFuture()

Returns the future that completes when all connections have drained. Only valid to call when the cluster is in VirtualClusterLifecycleState.Draining state.

drainComplete

public void drainComplete()

Transitions from VirtualClusterLifecycleState.Draining to VirtualClusterLifecycleState.Stopped.

stop

public void stop()

Transitions to VirtualClusterLifecycleState.Stopped from VirtualClusterLifecycleState.Failed or VirtualClusterLifecycleState.Initializing. Idempotent: a call when already VirtualClusterLifecycleState.Stopped is a silent no-op, so two concurrent shutdown paths that both observe a non-terminal state and race to call stop() cannot fail — the second caller's transition simply finds the state already terminal.

state

public VirtualClusterLifecycleState state()

clusterName

public String clusterName()

registerConnection

public boolean registerConnection(ClientConnectionStateMachine ccsm)

deregisterConnection

public void deregisterConnection(ClientConnectionStateMachine ccsm)

activeConnections

public Set<ClientConnectionStateMachine> activeConnections()