postgrestest

type Activity struct {
	PID                   int64
	Query                 string
	XactStart, QueryStart *time.Time
}

type ActivityKey struct{ Application, State, WaitEvent string }

type ActivityRecord struct {
	Database    string     `json:"database,omitempty"`
	Application string     `json:"application,omitempty"`
	State       string     `json:"state,omitempty"`
	WaitEvent   string     `json:"wait,omitempty"`
	PID         int64      `json:"pid"`
	QuerySHA256 string     `json:"query_sha256,omitempty"`
	XactStart   *time.Time `json:"xact_start,omitempty"`
	QueryStart  *time.Time `json:"query_start,omitempty"`
}

type ActivitySnapshot []ActivityRecord

type Clock interface{ Now() time.Time }

type ClockFunc func() time.Time

type Config struct {
	RunPrefix, Source string
	Interval          time.Duration

	Pooler, Postgres ConnFactory
	Clock            Clock
	// contains filtered or unexported fields
}

type ConfigSnapshot struct {
	Source, Identity, Version string
	ActivityRequired          bool `json:"activity_required"`
}

type ConnFactory func(context.Context) (QueryConn, error)

type Counter struct{ Xacts, Queries, Received, Sent, Wait, QueryTime, XactTime, Assignments, ClientParse, ServerParse, ClientBind int64 }

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

type DiagnosticReport struct {
	Identity    string           `json:"identity"`
	Status      DiagnosticStatus `json:"status"`
	Overhead    OverheadReport   `json:"overhead"`
	Complete    bool             `json:"complete"`
	Gaps        []string         `json:"gaps,omitempty"`
	Samples     []Sample         `json:"samples,omitempty"`
	PhaseEvents []PhaseEvent     `json:"phase_events,omitempty"`
	Truncated   bool             `json:"truncated,omitempty"`
}

type DiagnosticStatus string

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

type Overhead struct{ Off, On []time.Duration }

type OverheadReport struct {
	OffP50 time.Duration `json:"off_p50_ns"`
	OnP50  time.Duration `json:"on_p50_ns"`
	OffP95 time.Duration `json:"off_p95_ns"`
	OnP95  time.Duration `json:"on_p95_ns"`
	Pass   bool          `json:"pass"`
}

type PgBouncerSnapshot struct {
	Stats     map[string]Counter
	Pools     map[string]Pool
	Servers   map[string]Server
	Truncated bool `json:"truncated,omitempty"`
}

type Phase string

type PhaseEvent struct {
	Phase Phase     `json:"phase"`
	At    time.Time `json:"at"`
}

type Pool struct{ ClientActive, ClientWaiting, ServerActive, ServerIdle, ServerUsed, ServerTested, ServerLogin, MaxWait int64 }

type QueryConn interface {
	Query(context.Context, string, ...any) (Rows, error)
	Close(context.Context) error
}

type Rows interface {
	Next() bool
	Values() ([]any, error)
	Err() error
	Close()
	FieldDescriptions() []pgconn.FieldDescription
}

type Sample struct {
	Offset            time.Duration
	Wall              time.Time
	Sequence          uint64
	Gap               time.Duration
	Stats             PgBouncerSnapshot
	Activity          ActivitySnapshot
	Config            ConfigSnapshot
	Truncated         bool `json:"truncated,omitempty"`
	ActivityTruncated bool `json:"activity_truncated,omitempty"`
}

type Server struct {
	Type                                                                string
	Active, Idle, Used, Tested, Login, New, ActiveCancel, BeingCanceled int64
}