adeptersinterfaces

type AivenConfig struct {
	APIToken                string        `mapstructure:"AIVEN_API_TOKEN" validate:"required"`
	ProjectName             string        `mapstructure:"AIVEN_PROJECT_NAME" validate:"required"`
	ServiceName             string        `mapstructure:"AIVEN_SERVICE_NAME" validate:"required"`
	MetricResolutionSeconds time.Duration `mapstructure:"metric_resolution_seconds" validate:"required"`
	// NOTE: If specified, we are able to use the
	// session refresh hack. Not documented.
	DatabaseName string `mapstructure:"database_name"`
}

type AivenHardwareState struct {
	TotalMemoryBytes int64
	NumCPUs          int
	LastChecked      time.Time
}

type AivenPostgreSQLAdapter interface {
	PostgreSQLAdapter
	GetAivenClient() *aiven.Client
	GetAivenConfig() *AivenConfig
	GetAivenState() *AivenState
}

type AivenState struct {
	Hardware                       *AivenHardwareState
	InitialSharedBuffersPercentage float64
	InitialWorkMem                 int64
	LastAppliedConfig              time.Time
	// HACK: Used to trigger restarts on ALTER DATABASE statements
	LastKnownPGStatMonitorEnable bool
	// Guardrails
	LastGuardrailCheck            time.Time
	LastMemoryAvailableTime       time.Time
	LastMemoryAvailablePercentage float64
	LastHardwareInfoTime          time.Time
}

type AuroraHardwareState struct {
	TotalMemoryBytes int64
	NumCPUs          int
	LastChecked      time.Time
}

type AuroraPerformanceInsightsState struct {
	Enabled     bool
	ResourceID  string
	LastChecked time.Time
}

type AuroraRDSAdapter interface {
	PostgreSQLAdapter
	GetPIClient() *pi.Client
	GetEC2Client() *ec2.Client
	GetRDSClient() *rds.Client
	GetCWClient() *cloudwatch.Client
	GetAuroraRDSConfig() *AuroraRDSConfig
	GetEC2InstanceTypeInfo() (*ec2types.InstanceTypeInfo, error)
	GetAuroraState() *AuroraRDSState
}

type AuroraRDSConfig struct {
	AWSAccessKey          string `mapstructure:"AWS_ACCESS_KEY_ID"`
	AWSSecretAccessKey    string `mapstructure:"AWS_SECRET_ACCESS_KEY"`
	AWSRegion             string `mapstructure:"AWS_REGION" validate:"required"`
	RDSDatabaseIdentifier string `mapstructure:"RDS_DATABASE_IDENTIFIER" validate:"required"`
	RDSParameterGroupName string `mapstructure:"RDS_PARAMETER_GROUP_NAME" validate:"required"`
}

type AuroraRDSState struct {
	LastAppliedConfig   time.Time
	PerformanceInsights *AuroraPerformanceInsightsState
	Hardware            *AuroraHardwareState
	LastGuardrailCheck  time.Time
}

type DockerAdapter interface {
	PostgreSQLAdapter
	GetContainerName() string
	GetDockerClient() *client.Client
}

type PostgreSQLAdapter interface {
	Logger() *logrus.Logger
	PGDriver() *pgPool.Pool
	APIClient() *retryablehttp.Client
}