flightsqlingress

package

v0.0.0-...-a19b83b Latest Latest Go to latest Published: Mar 5, 2026 License: MIT Imports: 28 Imported by: 0

Details

Valid go.mod file
Redistributable license
Tagged version
Stable version
Learn more about best practices

Repository

github.com/posthog/duckgres

Links

Open Source Insights

Documentation ¶

Index ¶

Constants
type Config
type ControlPlaneFlightSQLHandler
- func NewControlPlaneFlightSQLHandler(sessions *flightAuthSessionStore, users map[string]string) (*ControlPlaneFlightSQLHandler, error)
type FlightIngress
- func NewFlightIngress(host string, port int, tlsConfig *tls.Config, users map[string]string, ...) (*FlightIngress, error)
type Hooks
type Options
type SessionProvider

Constants ¶

View Source

const (
	ReapTriggerPeriodic = "periodic"
	ReapTriggerForced   = "forced"
)

Variables ¶

This section is empty.

Functions ¶

This section is empty.

Types ¶

type Config ¶

type Config struct {
	SessionIdleTTL     time.Duration
	SessionReapTick    time.Duration
	HandleIdleTTL      time.Duration
	SessionTokenTTL    time.Duration
	WorkerQueueTimeout time.Duration // applied to CreateSession calls; 0 = use request context as-is
}

type ControlPlaneFlightSQLHandler ¶

type ControlPlaneFlightSQLHandler struct {
	flightsql.BaseServer
	// contains filtered or unexported fields
}

ControlPlaneFlightSQLHandler implements Flight SQL over control-plane sessions.

func NewControlPlaneFlightSQLHandler ¶

func NewControlPlaneFlightSQLHandler(sessions *flightAuthSessionStore, users map[string]string) (*ControlPlaneFlightSQLHandler, error)

func (*ControlPlaneFlightSQLHandler) BeginTransaction ¶

func (h *ControlPlaneFlightSQLHandler) BeginTransaction(ctx context.Context, req flightsql.ActionBeginTransactionRequest) ([]byte, error)

func (*ControlPlaneFlightSQLHandler) ClosePreparedStatement ¶

func (h *ControlPlaneFlightSQLHandler) ClosePreparedStatement(ctx context.Context, req flightsql.ActionClosePreparedStatementRequest) error

func (*ControlPlaneFlightSQLHandler) CloseSession ¶

func (h *ControlPlaneFlightSQLHandler) CloseSession(ctx context.Context, req *flight.CloseSessionRequest) (*flight.CloseSessionResult, error)

func (*ControlPlaneFlightSQLHandler) CreatePreparedStatement ¶

func (h *ControlPlaneFlightSQLHandler) CreatePreparedStatement(ctx context.Context, req flightsql.ActionCreatePreparedStatementRequest) (flightsql.ActionCreatePreparedStatementResult, error)

func (*ControlPlaneFlightSQLHandler) DoGetDBSchemas ¶

func (h *ControlPlaneFlightSQLHandler) DoGetDBSchemas(ctx context.Context, cmd flightsql.GetDBSchemas) (*arrow.Schema, <-chan flight.StreamChunk, error)

func (*ControlPlaneFlightSQLHandler) DoGetPreparedStatement ¶

func (h *ControlPlaneFlightSQLHandler) DoGetPreparedStatement(ctx context.Context, cmd flightsql.PreparedStatementQuery) (*arrow.Schema, <-chan flight.StreamChunk, error)

func (*ControlPlaneFlightSQLHandler) DoGetStatement ¶

func (h *ControlPlaneFlightSQLHandler) DoGetStatement(ctx context.Context, ticket flightsql.StatementQueryTicket) (*arrow.Schema, <-chan flight.StreamChunk, error)

func (*ControlPlaneFlightSQLHandler) DoGetTables ¶

func (h *ControlPlaneFlightSQLHandler) DoGetTables(ctx context.Context, cmd flightsql.GetTables) (*arrow.Schema, <-chan flight.StreamChunk, error)

func (*ControlPlaneFlightSQLHandler) DoPutCommandStatementUpdate ¶

func (h *ControlPlaneFlightSQLHandler) DoPutCommandStatementUpdate(ctx context.Context, cmd flightsql.StatementUpdate) (int64, error)

func (*ControlPlaneFlightSQLHandler) EndTransaction ¶

func (h *ControlPlaneFlightSQLHandler) EndTransaction(ctx context.Context, req flightsql.ActionEndTransactionRequest) error

func (*ControlPlaneFlightSQLHandler) GetFlightInfoPreparedStatement ¶

func (h *ControlPlaneFlightSQLHandler) GetFlightInfoPreparedStatement(ctx context.Context, cmd flightsql.PreparedStatementQuery, desc *flight.FlightDescriptor) (*flight.FlightInfo, error)

func (*ControlPlaneFlightSQLHandler) GetFlightInfoSchemas ¶

func (h *ControlPlaneFlightSQLHandler) GetFlightInfoSchemas(ctx context.Context, cmd flightsql.GetDBSchemas, desc *flight.FlightDescriptor) (*flight.FlightInfo, error)

func (*ControlPlaneFlightSQLHandler) GetFlightInfoStatement ¶

func (h *ControlPlaneFlightSQLHandler) GetFlightInfoStatement(ctx context.Context, cmd flightsql.StatementQuery, desc *flight.FlightDescriptor) (*flight.FlightInfo, error)

func (*ControlPlaneFlightSQLHandler) GetFlightInfoTables ¶

func (h *ControlPlaneFlightSQLHandler) GetFlightInfoTables(ctx context.Context, cmd flightsql.GetTables, desc *flight.FlightDescriptor) (*flight.FlightInfo, error)

type FlightIngress ¶

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

FlightIngress serves Arrow Flight SQL on the control plane with Basic auth. It reuses worker sessions via SessionProvider.

func NewFlightIngress ¶

func NewFlightIngress(host string, port int, tlsConfig *tls.Config, users map[string]string, provider SessionProvider, cfg Config, opts Options) (*FlightIngress, error)

func (*FlightIngress) Addr ¶

func (fi *FlightIngress) Addr() string

Addr returns the bound listener address.

func (*FlightIngress) Shutdown ¶

func (fi *FlightIngress) Shutdown()

Shutdown stops accepting new Flight connections and cleans up sessions.

func (*FlightIngress) Start ¶

func (fi *FlightIngress) Start()

Start begins serving in the background.

type Hooks ¶

type Hooks struct {
	OnSessionCountChanged func(int)
	OnSessionsReaped      func(trigger string, count int)
}

type Options ¶

type Options struct {
	Hooks       Hooks
	RateLimiter *server.RateLimiter
}

type SessionProvider ¶

type SessionProvider interface {
	CreateSession(ctx context.Context, username string, pid int32, memoryLimit string, threads int) (int32, *server.FlightExecutor, error)
	DestroySession(int32)
}

Source Files ¶

View all Source files

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL