driver_infrastructure

type AuroraLimitlessDialect interface {
	GetLimitlessRouterEndpointQuery() string
	DatabaseDialect
}

type AuroraMySQLDatabaseDialect struct {
	MySQLTopologyAwareDatabaseDialect
}

type AuroraPgDatabaseDialect struct {
	PgTopologyAwareDatabaseDialect
}

type BlockingHostListProvider interface {
	HostListProvider
	ForceRefreshHostListWithTimeout(shouldVerifyWriter bool, timeoutMs int) ([]*host_info_util.HostInfo, error)
}

type CanReleaseResources interface {
	ReleaseResources()
}

type ClusterTopologyMonitor interface {
	SetClusterId(clusterId string)
	ForceRefreshVerifyWriter(writerImportant bool, timeoutMs int) ([]*host_info_util.HostInfo, error)
	ForceRefreshUsingConn(conn driver.Conn, timeoutMs int) ([]*host_info_util.HostInfo, error)
	Close()
	Start(wg *sync.WaitGroup)
}

type ClusterTopologyMonitorImpl struct {
	// contains filtered or unexported fields
}

type ConnectFunc func(props map[string]string) (driver.Conn, error)

type ConnectionContainer struct {
	Conn driver.Conn
}

type ConnectionPlugin interface {
	GetSubscribedMethods() []string
	Execute(
		connInvokedOn driver.Conn,
		methodName string,
		executeFunc ExecuteFunc,
		methodArgs ...any) (wrappedReturnValue any, wrappedReturnValue2 any, wrappedOk bool, wrappedErr error)
	Connect(hostInfo *host_info_util.HostInfo, props map[string]string, isInitialConnection bool, connectFunc ConnectFunc) (driver.Conn, error)
	ForceConnect(hostInfo *host_info_util.HostInfo, props map[string]string, isInitialConnection bool, connectFunc ConnectFunc) (driver.Conn, error)
	AcceptsStrategy(strategy string) bool
	GetHostInfoByStrategy(role host_info_util.HostRole, strategy string, hosts []*host_info_util.HostInfo) (*host_info_util.HostInfo, error)
	GetHostSelectorStrategy(strategy string) (HostSelector, error)
	NotifyConnectionChanged(changes map[HostChangeOptions]bool) OldConnectionSuggestedAction
	NotifyHostListChanged(changes map[string]map[HostChangeOptions]bool)
	InitHostProvider(initialUrl string, props map[string]string, hostListProviderService HostListProviderService, initHostProviderFunc func() error) error
}

type ConnectionPluginFactory interface {
	GetInstance(pluginService PluginService, props map[string]string) (ConnectionPlugin, error)
	ClearCaches()
}

type ConnectionProvider interface {
	AcceptsUrl(hostInfo host_info_util.HostInfo, props map[string]string) bool
	AcceptsStrategy(strategy string) bool
	GetHostInfoByStrategy(hosts []*host_info_util.HostInfo, role host_info_util.HostRole, strategy string, props map[string]string) (*host_info_util.HostInfo, error)
	GetHostSelectorStrategy(strategy string) (HostSelector, error)
	Connect(hostInfo *host_info_util.HostInfo, props map[string]string, pluginService PluginService) (driver.Conn, error)
}

type ConnectionProviderManager struct {
	DefaultProvider   ConnectionProvider
	EffectiveProvider ConnectionProvider
}

type DatabaseDialect interface {
	GetDefaultPort() int
	GetHostAliasQuery() string
	GetServerVersionQuery() string
	GetDialectUpdateCandidates() []string
	IsDialect(conn driver.Conn) bool
	GetHostListProvider(props map[string]string, initialDsn string, hostListProviderService HostListProviderService, pluginService PluginService) HostListProvider
	DoesStatementSetAutoCommit(statement string) (bool, bool)
	DoesStatementSetReadOnly(statement string) (bool, bool)
	DoesStatementSetCatalog(statement string) (string, bool)
	DoesStatementSetSchema(statement string) (string, bool)
	DoesStatementSetTransactionIsolation(statement string) (TransactionIsolationLevel, bool)
	GetSetAutoCommitQuery(autoCommit bool) (string, error)
	GetSetReadOnlyQuery(readOnly bool) (string, error)
	GetSetCatalogQuery(catalog string) (string, error)
	GetSetSchemaQuery(schema string) (string, error)
	GetSetTransactionIsolationQuery(level TransactionIsolationLevel) (string, error)
}

type DatabaseEngine string

type DialectCode string

type DialectManager struct {
	FindRegisteredDriver func(dialectCode string) bool
	// contains filtered or unexported fields
}

type DialectProvider interface {
	GetDialect(dsn string, props map[string]string) (DatabaseDialect, error)
	GetDialectForUpdate(conn driver.Conn, originalHost string, newHost string) DatabaseDialect
}

type DriverConnectionProvider struct {
	// contains filtered or unexported fields
}

type DriverDialect interface {
	IsDialect(driver driver.Driver) bool
	GetAllowedOnConnectionMethodNames() []string
	PrepareDsn(properties map[string]string, info *host_info_util.HostInfo) string
	IsNetworkError(err error) bool
	IsLoginError(err error) bool
	IsClosed(conn driver.Conn) bool
	IsDriverRegistered(drivers map[string]driver.Driver) bool
	RegisterDriver()
}

type DsnHostListProvider struct {
	// contains filtered or unexported fields
}

type ExecuteFunc func() (any, any, bool, error)

type HighestWeightHostSelector struct{}

type HostChangeOptions int

type HostListProvider interface {
	Refresh(conn driver.Conn) ([]*host_info_util.HostInfo, error)
	ForceRefresh(conn driver.Conn) ([]*host_info_util.HostInfo, error)
	GetHostRole(conn driver.Conn) host_info_util.HostRole
	IdentifyConnection(conn driver.Conn) (*host_info_util.HostInfo, error)
	GetClusterId() (string, error)
	IsStaticHostListProvider() bool
	CreateHost(hostName string, role host_info_util.HostRole, lag float64, cpu float64, lastUpdateTime time.Time) *host_info_util.HostInfo
}

type HostListProviderService interface {
	IsStaticHostListProvider() bool
	CreateHostListProvider(props map[string]string, dsn string) HostListProvider
	GetHostListProvider() HostListProvider
	SetHostListProvider(hostListProvider HostListProvider)
	SetInitialConnectionHostInfo(info *host_info_util.HostInfo)
	GetDialect() DatabaseDialect
	GetCurrentConnection() driver.Conn
}

type HostMonitoringRoutine struct {
	// contains filtered or unexported fields
}

type HostSelector interface {
	GetHost(hosts []*host_info_util.HostInfo, role host_info_util.HostRole, props map[string]string) (*host_info_util.HostInfo, error)
}

type MonitoringRdsHostListProvider struct {
	*RdsHostListProvider
	// contains filtered or unexported fields
}

type MySQLDatabaseDialect struct {
}

type MySQLTopologyAwareDatabaseDialect struct {
	MySQLDatabaseDialect
}

type NumberRange struct {
	Start int
	End   int
}

type OldConnectionSuggestedAction string

type PgDatabaseDialect struct {
}

type PgTopologyAwareDatabaseDialect struct {
	PgDatabaseDialect
}

type PluginConnectFunc func(plugin ConnectionPlugin, props map[string]string, targetFunc func(props map[string]string) (driver.Conn, error)) (driver.Conn, error)

type PluginExecFunc func(plugin ConnectionPlugin, targetFunc func() (any, any, bool, error)) (any, any, bool, error)

type PluginManager interface {
	Init(pluginService PluginService, plugins []ConnectionPlugin) error
	InitHostProvider(initialUrl string, props map[string]string, hostListProviderService HostListProviderService) error
	Connect(hostInfo *host_info_util.HostInfo, props map[string]string, isInitialConnection bool, pluginToSkip ConnectionPlugin) (driver.Conn, error)
	ForceConnect(hostInfo *host_info_util.HostInfo, props map[string]string, isInitialConnection bool) (driver.Conn, error)
	Execute(connInvokedOn driver.Conn, name string, methodFunc ExecuteFunc, methodArgs ...any) (
		wrappedReturnValue any,
		wrappedReturnValue2 any,
		wrappedOk bool,
		wrappedErr error)
	AcceptsStrategy(strategy string) bool
	NotifyHostListChanged(changes map[string]map[HostChangeOptions]bool)
	NotifyConnectionChanged(
		changes map[HostChangeOptions]bool, skipNotificationForThisPlugin ConnectionPlugin) map[OldConnectionSuggestedAction]bool
	NotifySubscribedPlugins(methodName string, pluginFunc PluginExecFunc, skipNotificationForThisPlugin ConnectionPlugin) error
	GetHostInfoByStrategy(role host_info_util.HostRole, strategy string, hosts []*host_info_util.HostInfo) (*host_info_util.HostInfo, error)
	GetHostSelectorStrategy(strategy string) (hostSelector HostSelector, err error)
	GetDefaultConnectionProvider() ConnectionProvider
	GetEffectiveConnectionProvider() ConnectionProvider
	GetConnectionProviderManager() ConnectionProviderManager
	GetTelemetryContext() context.Context
	GetTelemetryFactory() telemetry.TelemetryFactory
	SetTelemetryContext(ctx context.Context)
	ReleaseResources()
}

type PluginManagerProvider func(
	targetDriver driver.Driver,
	props map[string]string,
	connProviderManager ConnectionProviderManager,
	telemetryFactory telemetry.TelemetryFactory) PluginManager

type PluginService interface {
	GetCurrentConnection() driver.Conn
	GetCurrentConnectionRef() *driver.Conn
	SetCurrentConnection(conn driver.Conn, hostInfo *host_info_util.HostInfo, skipNotificationForThisPlugin ConnectionPlugin) error
	GetInitialConnectionHostInfo() *host_info_util.HostInfo
	GetCurrentHostInfo() (*host_info_util.HostInfo, error)
	GetHosts() []*host_info_util.HostInfo
	AcceptsStrategy(strategy string) bool
	GetHostInfoByStrategy(role host_info_util.HostRole, strategy string, hosts []*host_info_util.HostInfo) (*host_info_util.HostInfo, error)
	GetHostSelectorStrategy(strategy string) (hostSelector HostSelector, err error)
	GetHostRole(driver.Conn) host_info_util.HostRole
	SetAvailability(hostAliases map[string]bool, availability host_info_util.HostAvailability)
	IsInTransaction() bool
	SetInTransaction(inTransaction bool)
	GetCurrentTx() driver.Tx
	SetCurrentTx(driver.Tx)
	CreateHostListProvider(props map[string]string, dsn string) HostListProvider
	SetHostListProvider(hostListProvider HostListProvider)
	SetInitialConnectionHostInfo(info *host_info_util.HostInfo)
	IsStaticHostListProvider() bool
	GetHostListProvider() HostListProvider
	RefreshHostList(conn driver.Conn) error
	ForceRefreshHostList(conn driver.Conn) error
	ForceRefreshHostListWithTimeout(shouldVerifyWriter bool, timeoutMs int) (bool, error)
	GetUpdatedHostListWithTimeout(shouldVerifyWriter bool, timeoutMs int) ([]*host_info_util.HostInfo, error)
	Connect(hostInfo *host_info_util.HostInfo, props map[string]string, pluginToSkip ConnectionPlugin) (driver.Conn, error)
	ForceConnect(hostInfo *host_info_util.HostInfo, props map[string]string) (driver.Conn, error)
	GetDialect() DatabaseDialect
	SetDialect(dialect DatabaseDialect)
	UpdateDialect(conn driver.Conn)
	GetTargetDriverDialect() DriverDialect
	IdentifyConnection(conn driver.Conn) (*host_info_util.HostInfo, error)
	FillAliases(conn driver.Conn, hostInfo *host_info_util.HostInfo)
	GetConnectionProvider() ConnectionProvider
	GetProperties() map[string]string
	IsNetworkError(err error) bool
	IsLoginError(err error) bool
	GetTelemetryContext() context.Context
	GetTelemetryFactory() telemetry.TelemetryFactory
	SetTelemetryContext(ctx context.Context)
	UpdateState(sql string, methodArgs ...any)
}

type PluginServiceProvider func(
	pluginManager PluginManager,
	driverDialect DriverDialect,
	props map[string]string,
	dsn string) (PluginService, error)

type RandomHostSelector struct{}

type RdsHostListProvider struct {
	IsPrimaryClusterId bool
	// contains filtered or unexported fields
}

type RdsMultiAzDbClusterMySQLDialect struct {
	MySQLTopologyAwareDatabaseDialect
}

type RdsMultiAzDbClusterPgDialect struct {
	PgTopologyAwareDatabaseDialect
}

type RdsMySQLDatabaseDialect struct {
	MySQLDatabaseDialect
}

type RdsPgDatabaseDialect struct {
	PgDatabaseDialect
}

type SessionState struct {
	AutoCommit           SessionStateField[bool]
	ReadOnly             SessionStateField[bool]
	Catalog              SessionStateField[string]
	Schema               SessionStateField[string]
	TransactionIsolation SessionStateField[TransactionIsolationLevel]
}

type SessionStateField[T comparable] struct {
	// contains filtered or unexported fields
}

type SessionStateService interface {
	GetAutoCommit() *bool
	SetAutoCommit(bool)
	SetupPristineAutoCommit()
	SetupPristineAutoCommitWithVal(bool)

	GetReadOnly() *bool
	SetReadOnly(bool)
	SetupPristineReadOnly()
	SetupPristineReadOnlyWithVal(bool)

	GetCatalog() *string
	SetCatalog(string)
	SetupPristineCatalog()
	SetupPristineCatalogWithVal(string)

	GetSchema() *string
	SetSchema(string)
	SetupPristineSchema()
	SetupPristineSchemaWithVal(string)

	GetTransactionIsolation() *TransactionIsolationLevel
	SetTransactionIsolation(level TransactionIsolationLevel)
	SetupPristineTransactionIsolation()
	SetupPristineTransactionIsolationWithVal(TransactionIsolationLevel)

	Begin() error
	Complete()
	Reset()

	ApplyCurrentSessionState(conn driver.Conn) error
	ApplyPristineSessionState(conn driver.Conn) error
}

type SessionStateServiceImpl struct {
	SessionState *SessionState
	// contains filtered or unexported fields
}

type TopologyAwareDialect interface {
	GetTopology(conn driver.Conn, provider HostListProvider) ([]*host_info_util.HostInfo, error)
	GetHostRole(conn driver.Conn) host_info_util.HostRole
	GetHostName(conn driver.Conn) string
	GetWriterHostName(conn driver.Conn) (string, error)
	DatabaseDialect
}

type TransactionIsolationLevel int

type WeightedHostSelector interface {
	SetHostWeights(hostWeightMap map[string]int)
	ClearHostWeights()
	HostSelector
}

type WeightedRandomHostSelector struct {
	// contains filtered or unexported fields
}