aiven

type AivenPostgreSQLAdapter struct {
	agent.CommonAgent
	Config            Config
	Client            aivenclient.Client
	State             *State
	GuardrailSettings guardrails.Config
	PGDriver          *pgPool.Pool
	PGVersion         string
}

type Config 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"`
	MetricResolution 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 FetchMetricJSONData struct {
	Data struct {
		Cols []Schema `json:"cols" validate:"required"`
		Rows [][]any  `json:"rows" validate:"required"` // 1st element is Schema, rest are values with length of Cols
	} `json:"data" validate:"required"`
	MaxTimestamp int64 `json:"max_timestamp" validate:"required"` // seconds since epoch
	TotalSeries  int64 `json:"total_series"`                      // How many rows there are I guess?
	Hints        struct {
		Title string `json:"title"`
	} `json:"hints"` // Additional metadata like title
}

type FetchMetricsJSONScheme map[string]FetchMetricJSONData

type FetchedMetricsIn struct {
	ProjectName string
	ServiceName string
	Client      *aiven.Client
	Logger      *log.Logger
	Period      service.PeriodType
	Metrics     []MetricKnownName
}

type FetchedMetricsOut map[MetricKnownName]MaybeParsedMetric

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

type InitialServiceLevelParameters struct {
	InitialSharedBuffersPercentage float64
	InitialPGStatMonitorEnable     bool
}

type MaybeParsedMetric struct {
	ParsedMetric
	Error error
}

type MetricKnownName string

type ModifyLevel string

type ParsedMetric struct {
	Name      MetricKnownName
	Value     any
	MetricDef metrics.MetricDef
	Timestamp time.Time
}

type Schema struct {
	Label string `json:"label" validate:"required"`
	Type  string `json:"type"` // "number" or "date", could be others
}

type State struct {
	Hardware                       Hardware
	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
}