postgresutil

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

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

type AdminOptions struct {
	ConnectionTimeout int
	Username          string
	Password          string
}

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

type IAdmin interface {
	// Connections returns the connection map of all clients
	Connections() IAdminConnections

	// Close the admin connections
	Close()

	// GetReplicationStatus return the Nodes infos for all nodes
	GetReplicationStatus(ctx context.Context) *ReplicationInfo

	// ConfigureStandby set postgres standby primary connect info
	ConfigureStandby(ctx context.Context, addr, nodeName, sourceHost, sourcePort, replicationUser, replicationPwd string) error

	// ConfigureSyncMode set postgres replication to synchronous
	ConfigureSyncMode(ctx context.Context, addr, standbyList string) error

	// ConfigureAsyncMode set postgres replication to asynchronous
	ConfigureAsyncMode(ctx context.Context, addr string) error
}

type IAdminConnections interface {
	// AddAll connect to the given list of addresses and
	// register them in the map
	// fail silently
	AddAll(addrs []string)

	// Add connect to the given address and
	// register the client connection to the pool
	Add(addr string) error

	// Reconnect force a reconnection on the given address
	// if the adress is not part of the map, act like Add
	Reconnect(addr string) error

	// Remove disconnect and remove the client connection from the map
	Remove(addr string)

	// ReplaceAll clear the map and re-populate it with new connections
	// fail silently
	ReplaceAll(addrs []string)

	// Reset close all connections and clear the connection map
	Reset()

	// Get returns a client connection for the given address,
	// connects if the connection is not in the map yet
	Get(addr string) (IClient, error)

	// GetAll returns a map of all clients per address
	GetAll() map[string]IClient

	//GetSelected returns a map of clients based on the input addresses
	GetSelected(addrs []string) map[string]IClient
}

type IClient interface {
	// Close closes the connection.
	Close() error

	// Exec calls the given query like
	Exec(ctx context.Context, query string, args ...any) (sql.Result, error)
	Query(ctx context.Context, query string, args ...any) (*sql.Rows, error)
	QueryRow(ctx context.Context, input interface{}, query string, args ...any) error
}

type Node struct {
	Host            string
	Port            int
	Role            string
	ReplicationStat []string
	DataDir         string
	ReplicationMode string
	WalDiff         int
}

type Nodes []*Node

type ReplicationInfo struct {
	Nodes map[string]*Node
}