clickhouse

type Agent struct {
	Name string
	Ver  Ver
}

type ApplyToCredentialser interface {
	ApplyToCredentials(*Credentials) error
}

type BaseOption string

type Block struct {
	Table string

	Rows int
	Cols []Column
}

type Client interface {
	NextPacket(context.Context) (ServerPacket, error)

	SendQuery(context.Context, *Query) (QueryMeta, error)
	CancelQuery(context.Context) error

	SendBlock(ctx context.Context, b *Block, compr bool) error
	RecvBlock(ctx context.Context, compr bool) (b *Block, err error)

	RecvException(context.Context) error
	RecvProgress(context.Context) (Progress, error)
	RecvProfileInfo(context.Context) (ProfileInfo, error)
}

type ClientOption interface {
	fmt.Stringer
}

type ClientPacket int

type ClientPool interface {
	Get(context.Context, ...ClientOption) (Client, error)
	Put(context.Context, Client, error) error

	io.Closer
}

type Column struct {
	Name string
	Type string

	RawData []byte
}

type Credentials struct {
	Database string
	User     string
	Password string
}

type CredentialsOption struct {
	BaseOption
	F func(*Credentials) error
}

type Exception struct {
	Code       int32
	Name       string
	Message    string
	StackTrace string

	Cause *Exception
}

type Pinger interface {
	SendPing(context.Context) error
}

type ProfileInfo struct {
	Rows   uint64
	Blocks uint64
	Bytes  uint64

	AppliedLimit        uint64
	RowsBeforeLimit     uint64
	CalcRowsBeforeLimit uint64
}

type Progress struct {
	Rows      uint64
	Bytes     uint64
	TotalRows uint64
}

type Query struct {
	Query string

	ID       string
	QuotaKey string

	Compressed bool

	Client Agent
}

type QueryMeta []Column

type Server interface {
	Serve(context.Context, net.Listener) error
	HandleConn(context.Context, net.Conn) error
	HandleRequest(context.Context, ServerConn, ...ClientOption) error

	io.Closer
}

type ServerConn interface {
	NextPacket(context.Context) (ClientPacket, error)

	RecvQuery(context.Context) (*Query, error)
	SendQueryMeta(ctx context.Context, m QueryMeta, compr bool) error

	SendBlock(ctx context.Context, b *Block, compr bool) error
	RecvBlock(ctx context.Context, compr bool) (b *Block, err error)

	SendEndOfStream(context.Context) error

	SendException(context.Context, error) error
	SendProgress(context.Context, Progress) error
	SendProfileInfo(context.Context, ProfileInfo) error
}

type ServerPacket int

type Ver [3]int