software.amazon.jdbc.plugin.bluegreen

Class BlueGreenStatusMonitor

java.lang.Object

software.amazon.jdbc.plugin.bluegreen.BlueGreenStatusMonitor

public class BlueGreenStatusMonitor
extends java.lang.Object

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	allStartTopologyEndpointsRemoved
protected boolean	allStartTopologyIpChanged
protected boolean	allTopologyChanged
protected static java.lang.String	BG_CLUSTER_ID
protected java.lang.String	bgdId
protected BlueGreenDialect	blueGreenDialect
protected java.util.concurrent.atomic.AtomicReference<java.sql.PreparedStatement>	checkStatusStatement
protected java.util.concurrent.atomic.AtomicBoolean	collectIpAddresses
protected java.util.concurrent.atomic.AtomicBoolean	collectTopology
protected java.util.concurrent.atomic.AtomicReference<java.lang.String>	connectedIpAddress
protected AtomicConnection	connection
protected java.util.concurrent.atomic.AtomicReference<HostSpec>	connectionHostSpec
protected java.util.concurrent.atomic.AtomicBoolean	connectionHostSpecCorrect
protected ConnectionUrlParser	connectionUrlParser
protected java.util.Map<java.lang.String,java.util.Optional<java.lang.String>>	currentIpAddressesByHostMap
protected @Nullable BlueGreenPhase	currentPhase
protected java.util.concurrent.atomic.AtomicReference<java.util.List<HostSpec>>	currentTopology
protected static long	DEFAULT_CHECK_INTERVAL_MS
protected java.util.concurrent.ExecutorService	executorService
protected @Nullable HostListProvider	hostListProvider
protected java.util.Set<java.lang.String>	hostNames
protected HostSpecBuilder	hostSpecBuilder
protected HostSpec	initialHostSpec
protected java.util.concurrent.atomic.AtomicReference<BlueGreenIntervalRate>	intervalRate
protected static java.util.Set<java.lang.String>	knownVersions
protected static java.lang.String	latestKnownVersion
protected @Nullable OnBlueGreenStatusChange	onBlueGreenStatusChangeFunc
protected java.util.concurrent.ExecutorService	openConnectionExecutorService
protected @Nullable java.util.concurrent.Future<java.lang.Void>	openConnectionFuture
protected java.util.concurrent.atomic.AtomicBoolean	panicMode
protected PluginService	pluginService
protected int	port
protected java.util.Properties	props
protected RdsUtils	rdsUtils
protected BlueGreenRole	role
protected FullServicesContainer	servicesContainer
protected java.lang.Object	sleepWaitObj
protected java.util.Map<java.lang.String,java.util.Optional<java.lang.String>>	startIpAddressesByHostMap
protected java.util.List<HostSpec>	startTopology
protected java.util.Map<BlueGreenIntervalRate,java.lang.Long>	statusCheckIntervalMap
protected java.util.concurrent.atomic.AtomicBoolean	stop
protected java.util.concurrent.atomic.AtomicBoolean	useIpAddress
protected java.lang.String	version

Constructor Summary

Constructors

Constructor and Description
BlueGreenStatusMonitor(@NonNull BlueGreenRole role, @NonNull java.lang.String bgdId, @NonNull HostSpec initialHostSpec, @NonNull FullServicesContainer servicesContainer, @NonNull java.util.Properties props, @NonNull java.util.Map<BlueGreenIntervalRate,java.lang.Long> statusCheckIntervalMap, @Nullable OnBlueGreenStatusChange onBlueGreenStatusChangeFunc)

Method Summary

Modifier and Type	Method and Description
protected void	clearCheckStatusStatement()
protected void	collectHostIpAddresses()
protected void	collectStatus()
protected void	collectTopology()
protected void	delay(long delayMs)
protected java.sql.ResultSet	executeCheckStatusStatement(java.sql.Connection conn)
protected java.util.Optional<java.lang.String>	getIpAddress(java.lang.String host)
protected void	initHostListProvider()
protected boolean	isConnectionClosed(@Nullable java.sql.Connection conn)
protected void	notifyChanges()
protected void	openConnection()
void	resetCollectedData()
protected void	runMonitoringLoop()
void	setCollectIpAddresses(boolean collectIpAddresses)
void	setCollectTopology(boolean collectTopology)
void	setIntervalRate(@NonNull BlueGreenIntervalRate blueGreenIntervalRate)
void	setStop(boolean stop)
void	setUseIpAddress(boolean useIpAddress)
void	start()
protected @Nullable java.sql.Connection	tryConnectToAllInstances(HostSpec connectionHostSpecCopy) Attempts to connect to all known instance IPs in parallel and returns the first successful connection.
protected @Nullable java.sql.Connection	tryConnectToSingleIp(java.lang.String ip, HostSpec connectionHostSpecCopy) Attempts a single IP connection.
protected void	updateIpAddressFlags()

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_CHECK_INTERVAL_MS

protected static final long DEFAULT_CHECK_INTERVAL_MS

BG_CLUSTER_ID

protected static final java.lang.String BG_CLUSTER_ID

See Also:

Constant Field Values

latestKnownVersion

protected static final java.lang.String latestKnownVersion

See Also:

Constant Field Values

knownVersions

protected static final java.util.Set<java.lang.String> knownVersions

blueGreenDialect

protected final BlueGreenDialect blueGreenDialect

servicesContainer

protected final FullServicesContainer servicesContainer

pluginService

protected final PluginService pluginService

bgdId

protected final java.lang.String bgdId

props

protected final java.util.Properties props

role

protected final BlueGreenRole role

onBlueGreenStatusChangeFunc

protected final @Nullable OnBlueGreenStatusChange onBlueGreenStatusChangeFunc

statusCheckIntervalMap

protected final java.util.Map<BlueGreenIntervalRate,java.lang.Long> statusCheckIntervalMap

initialHostSpec

protected final HostSpec initialHostSpec

executorService

protected final java.util.concurrent.ExecutorService executorService

openConnectionExecutorService

protected final java.util.concurrent.ExecutorService openConnectionExecutorService

rdsUtils

protected final RdsUtils rdsUtils

connectionUrlParser

protected final ConnectionUrlParser connectionUrlParser

hostSpecBuilder

protected final HostSpecBuilder hostSpecBuilder

collectIpAddresses

protected final java.util.concurrent.atomic.AtomicBoolean collectIpAddresses

collectTopology

protected final java.util.concurrent.atomic.AtomicBoolean collectTopology

intervalRate

protected final java.util.concurrent.atomic.AtomicReference<BlueGreenIntervalRate> intervalRate

stop

protected final java.util.concurrent.atomic.AtomicBoolean stop

useIpAddress

protected final java.util.concurrent.atomic.AtomicBoolean useIpAddress

sleepWaitObj

protected final java.lang.Object sleepWaitObj

hostListProvider

protected @Nullable HostListProvider hostListProvider

startTopology

protected java.util.List<HostSpec> startTopology

currentTopology

protected final java.util.concurrent.atomic.AtomicReference<java.util.List<HostSpec>> currentTopology

startIpAddressesByHostMap

protected java.util.Map<java.lang.String,java.util.Optional<java.lang.String>> startIpAddressesByHostMap

currentIpAddressesByHostMap

protected java.util.Map<java.lang.String,java.util.Optional<java.lang.String>> currentIpAddressesByHostMap

allStartTopologyIpChanged

protected boolean allStartTopologyIpChanged

allStartTopologyEndpointsRemoved

protected boolean allStartTopologyEndpointsRemoved

allTopologyChanged

protected boolean allTopologyChanged

currentPhase

protected @Nullable BlueGreenPhase currentPhase

hostNames

protected java.util.Set<java.lang.String> hostNames

version

protected java.lang.String version

port

protected int port

connection

protected AtomicConnection connection

connectionHostSpec

protected final java.util.concurrent.atomic.AtomicReference<HostSpec> connectionHostSpec

connectedIpAddress

protected final java.util.concurrent.atomic.AtomicReference<java.lang.String> connectedIpAddress

connectionHostSpecCorrect

protected final java.util.concurrent.atomic.AtomicBoolean connectionHostSpecCorrect

panicMode

protected final java.util.concurrent.atomic.AtomicBoolean panicMode

openConnectionFuture

protected @Nullable java.util.concurrent.Future<java.lang.Void> openConnectionFuture

checkStatusStatement

protected final java.util.concurrent.atomic.AtomicReference<java.sql.PreparedStatement> checkStatusStatement

Constructor Detail

BlueGreenStatusMonitor

public BlueGreenStatusMonitor(@NonNull BlueGreenRole role,
                              @NonNull java.lang.String bgdId,
                              @NonNull HostSpec initialHostSpec,
                              @NonNull FullServicesContainer servicesContainer,
                              @NonNull java.util.Properties props,
                              @NonNull java.util.Map<BlueGreenIntervalRate,java.lang.Long> statusCheckIntervalMap,
                              @Nullable OnBlueGreenStatusChange onBlueGreenStatusChangeFunc)

Method Detail

start

public void start()

runMonitoringLoop

protected void runMonitoringLoop()

delay

protected void delay(long delayMs)
              throws java.lang.InterruptedException

Throws:

java.lang.InterruptedException

setIntervalRate

public void setIntervalRate(@NonNull BlueGreenIntervalRate blueGreenIntervalRate)

setCollectIpAddresses

public void setCollectIpAddresses(boolean collectIpAddresses)

setCollectTopology

public void setCollectTopology(boolean collectTopology)

setUseIpAddress

public void setUseIpAddress(boolean useIpAddress)

setStop

public void setStop(boolean stop)

resetCollectedData

public void resetCollectedData()

collectHostIpAddresses

protected void collectHostIpAddresses()

updateIpAddressFlags

protected void updateIpAddressFlags()

getIpAddress

protected java.util.Optional<java.lang.String> getIpAddress(java.lang.String host)

collectTopology

protected void collectTopology()
                        throws java.sql.SQLException

Throws:

java.sql.SQLException

clearCheckStatusStatement

protected void clearCheckStatusStatement()

executeCheckStatusStatement

protected java.sql.ResultSet executeCheckStatusStatement(java.sql.Connection conn)
                                                  throws java.sql.SQLException

Throws:

java.sql.SQLException

collectStatus

protected void collectStatus()

isConnectionClosed

protected boolean isConnectionClosed(@Nullable java.sql.Connection conn)

openConnection

protected void openConnection()

tryConnectToAllInstances

protected @Nullable java.sql.Connection tryConnectToAllInstances(HostSpec connectionHostSpecCopy)

Attempts to connect to all known instance IPs in parallel and returns the first successful connection. Each task races to set a shared AtomicConnection via compareAndSet — the first to succeed wins, and all others close their connections immediately. The AtomicConnection's GC-based cleanup provides a safety net for any edge case where a connection might otherwise leak.

IP connections are prioritized over endpoint connections to avoid stale DNS data.

Parameters:

connectionHostSpecCopy - the host spec to use as a template (for port, etc.)

Returns:

the first successful connection, or null if all attempts failed

tryConnectToSingleIp

protected @Nullable java.sql.Connection tryConnectToSingleIp(java.lang.String ip,
                                                             HostSpec connectionHostSpecCopy)

Attempts a single IP connection. Used as a fallback when only one IP is known.

notifyChanges

protected void notifyChanges()

initHostListProvider

protected void initHostListProvider()
                             throws java.sql.SQLException

Throws:

java.sql.SQLException