postgresql

type AuthManager interface {
	// AddAuthenticator adds a new authenticator.
	AddAuthenticator(authenticator auth.Authenticator)
	// ClearAuthenticators clears all authenticators.
	ClearAuthenticators()
}

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

type BaseBulkExecutor struct {
}

type BaseErrorHandler struct {
}

type BaseExecutor struct {
	StartupHandler
	QueryExecutor
	QueryExtraExecutor
	TCLExecutor
	ExtendedQueryExecutor
	BulkExecutor
	ErrorHandler
	SystemQueryExecutor
}

type BaseExtendedQueryExecutor struct {
	*BaseExecutor
}

type BaseQueryExecutor struct {
}

type BaseStartupExecutor struct {
	*BaseAuthExecutor
	// contains filtered or unexported fields
}

type BaseSugarExecutor struct {
	*BaseExecutor
}

type BaseSystemQueryExecutor struct {
}

type BaseTransactionExecutor struct {
}

type BulkExecutor interface {
	// Copy handles a COPY query.
	Copy(Conn, query.Copy) (protocol.Responses, error)
	// CopyData handles a COPY data protocol.
	CopyData(Conn, query.Copy, *CopyStream) (protocol.Responses, error)
}

type Config interface {
	TLSConfig
	// SetAddress sets a listen address to the configuration.
	SetAddress(addr string)
	// SetPort sets a listen port to the configuration.
	SetPort(port int)
	// Address returns a listen address from the configuration.
	Address() string
	// Port returns a listen port from the configuration.
	Port() int
}

type Conn interface {
	net.Conn
	PreparedConn
	MessageConn
	TLSConn
}

type ConnID = net.ConnID

type CopyStream struct {
	*protocol.MessageReader
}

type DDOExecutor interface {
	// CreateDatabase handles a CREATE DATABASE query.
	CreateDatabase(Conn, query.CreateDatabase) (protocol.Responses, error)
	// CreateTable handles a CREATE TABLE query.
	CreateTable(Conn, query.CreateTable) (protocol.Responses, error)
	// AlterDatabase handles a ALTER DATABASE query.
	AlterDatabase(Conn, query.AlterDatabase) (protocol.Responses, error)
	// AlterTable handles a ALTER TABLE query.
	AlterTable(Conn, query.AlterTable) (protocol.Responses, error)
	// DropDatabase handles a DROP DATABASE query.
	DropDatabase(Conn, query.DropDatabase) (protocol.Responses, error)
	// DropIndex handles a DROP INDEX query.
	DropTable(Conn, query.DropTable) (protocol.Responses, error)
}

type DMOExecutor interface {
	// Insert handles a INSERT query.
	Insert(Conn, query.Insert) (protocol.Responses, error)
	// Select handles a SELECT query.
	Select(Conn, query.Select) (protocol.Responses, error)
	// Update handles a UPDATE query.
	Update(Conn, query.Update) (protocol.Responses, error)
	// Delete handles a DELETE query.
	Delete(Conn, query.Delete) (protocol.Responses, error)
}

type DMOExtraExecutor interface {
	// Vacuum handles a VACUUM query.
	Vacuum(Conn, query.Vacuum) (protocol.Responses, error)
	// Truncate handles a TRUNCATE query.
	Truncate(Conn, query.Truncate) (protocol.Responses, error)
}

type ErrorHandler interface {
	ParserError(Conn, string, error) (protocol.Responses, error)
}

type Executor interface {
	SystemExecutor
	UserExecutor
}

type ExtendedQueryExecutor interface {
	// Query handles a query protocol.
	Query(Conn, *protocol.Query) (protocol.Responses, error)
	// Prepare handles a parse protocol.
	Parse(Conn, *protocol.Parse) (protocol.Responses, error)
	// Bind handles a bind protocol.
	Bind(Conn, *protocol.Bind) (protocol.Responses, error)
	// Describe handles a describe protocol.
	Describe(Conn, *protocol.Describe) (protocol.Responses, error)
	// Execute handles a execute protocol.
	Execute(Conn, *protocol.Execute) (protocol.Responses, error)
	// Close handles a close protocol.
	Close(Conn, *protocol.Close) (protocol.Responses, error)
	// Sync handles a sync protocol.
	Sync(Conn, *protocol.Sync) (protocol.Responses, error)
	// Flush handles a flush protocol.
	Flush(Conn, *protocol.Flush) (protocol.Responses, error)
}

type MessageConn interface {
	// SetStartupMessage sets a startup protocol.
	SetStartupMessage(msg *protocol.Startup)
	// StartupMessage return the startup protocol.
	StartupMessage() (*protocol.Startup, bool)
	// MessageReader returns a message reader.
	MessageReader() *protocol.MessageReader
	// ResponseMessage sends a response protocol.
	ResponseMessage(resMsg protocol.Response) error
	// ResponseMessages sends response messages.
	ResponseMessages(resMsgs protocol.Responses) error
	// ResponseError sends an error response.
	ResponseError(err error) error
	// SkipMessage skips a protocol.
	SkipMessage() error
	// ReadyForMessage sends a ready for protocol.
	ReadyForMessage(status protocol.TransactionStatus) error
}

type PreparedConn interface {
	// PreparedStatement returns a prepared statement.
	PreparedStatement(name string) (*PreparedStatement, error)
	// SetPreparedStatement sets a prepared statement.
	SetPreparedStatement(msg *protocol.Parse) error
	// RemovePreparedStatement removes a prepared statement.
	RemovePreparedStatement(name string) error
	// PreparedPortal returns a prepared query statement.
	PreparedPortal(name string) (*PreparedPortal, error)
	// SetPreparedPortal sets a prepared query statement.
	SetPreparedPortal(name string, query *PreparedPortal) error
	// RemovePreparedPortal removes a prepared query statement.
	RemovePreparedPortal(name string) error
}

type PreparedPortal = protocol.Query

type PreparedPortalMap map[string]PreparedPortal

type PreparedStatement struct {
	*protocol.Parse
	ParsedStatement *query.Statement
}

type PreparedStatementMap map[string]*PreparedStatement

type QueryExecutor interface {
	DDOExecutor
	DMOExecutor
}

type QueryExtraExecutor interface {
	DMOExtraExecutor
}

type Server interface {
	tracer.Tracer
	Config
	AuthManager
	UserExecutorHandler
	SetTracer(tracer.Tracer)
	Start() error
	Stop() error
	Restart() error
}

type StartupAuthHandler interface {
	AuthManager
	// Authenticate handles the Start-up message and returns an Authentication or ErrorResponse protocol.
	Authenticate(Conn) (protocol.Response, error)
}

type StartupHandler interface {
	StartupAuthHandler
	// ParameterStatuses returns the parameter statuses.
	ParameterStatuses(Conn) (protocol.Responses, error)
	// BackendKeyData returns the backend key data.
	BackendKeyData(Conn) (protocol.Response, error)
}

type SystemExecutor interface {
	StartupHandler
}

type SystemExecutorHandler interface {
	// SetStartupHandler sets a user startup handler.
	SetStartupHandler(StartupHandler)
}

type SystemQueryExecutor interface {
	// SystemSelect handles a SELECT query for system tables.
	SystemSelect(Conn, query.Select) (protocol.Responses, error)
}

type TCLExecutor interface {
	// Begin handles a BEGIN query.
	Begin(Conn, query.Begin) (protocol.Responses, error)
	// Commit handles a COMMIT query.
	Commit(Conn, query.Commit) (protocol.Responses, error)
	// Rollback handles a ROLLBACK query.
	Rollback(Conn, query.Rollback) (protocol.Responses, error)
}

type TLSConfig interface {
	// SetTLSEnabled sets a TLS enabled flag.
	SetTLSEnabled(enabled bool)
	// IsEnabled returns true if the TLS is enabled.
	IsTLSEnabled() bool
	// SetClientAuthType sets a client authentication type.
	SetClientAuthType(authType tls.ClientAuthType)
	// SetServerKeyFile sets a SSL server key file.
	SetServerKeyFile(file string) error
	// SetServerCertFile sets a SSL server certificate file.
	SetServerCertFile(file string) error
	// SetRootCertFile sets a SSL root certificates.
	SetRootCertFiles(files ...string) error
	// SetServerKey sets a SSL server key.
	SetServerKey(key []byte)
	// SetServerCert sets a SSL server certificate.
	SetServerCert(cert []byte)
	// SetRootCerts sets a SSL root certificates.
	SetRootCerts(certs ...[]byte)
	// SetTLSConfig sets a TLS configuration.
	SetTLSConfig(tlsConfig *tls.Config)
	// TLSConfig returns a TLS configuration from the configuration.
	TLSConfig() (*tls.Config, error)
}

type TLSConn interface {
	// IsTLSConnection return true if the connection is enabled TLS.
	IsTLSConnection() bool
	// TLSConnectionState returns the TLS connection state.
	TLSConnectionState() (*tls.ConnectionState, bool)
}

type UserExecutor interface {
	QueryExecutor
	QueryExtraExecutor
	TCLExecutor
	ExtendedQueryExecutor
	SystemQueryExecutor
	BulkExecutor
	ErrorHandler
}

type UserExecutorHandler interface {
	// SetQueryExecutor sets a user query executor.
	SetQueryExecutor(QueryExecutor)
	// SetQueryExecutor sets a user query executor.
	SetQueryExtraExecutor(QueryExtraExecutor)
	// SetTransactionExecutor sets a user transaction executor.
	SetTransactionExecutor(TCLExecutor)
	// SetSystemQueryExecutor sets a system query executor.
	SetSystemQueryExecutor(SystemQueryExecutor)
	// SetBulkExecutor sets a user bulk executor.
	SetBulkExecutor(BulkExecutor)
	// SetErrorHandler sets a user error handler.
	SetErrorHandler(ErrorHandler)
}