flink

type APIResponse struct {
	Versions []string `json:"versions"`
}

type Client struct {
	// BaseURL should be the root of the SQL Gateway REST API, e.g.
	// "http://localhost:8083".  Do not include a trailing slash; the
	// Client automatically appends endpoint paths.
	BaseURL *url.URL
	// HTTPClient is the http.Client used for sending requests.  If
	// nil, http.DefaultClient is used.  You can set custom timeouts
	// or transports on this client for better control.
	HTTPClient *http.Client
	// APIVersion controls which API version is used for requests.
	// Valid values are "v1", "v2" or "v3".  If empty, "v3" is used.
	APIVersion string
}

type ColumnInfo struct {
	Name        string      `json:"name"`
	LogicalType LogicalType `json:"logicalType"`
	Comment     string      `json:"comment"`
}

type CompleteStatementRequest struct {
	Position  int    `json:"position"`
	Statement string `json:"statement"`
}

type CompleteStatementResponse struct {
	Candidates []string `json:"candidates"`
}

type ConfigureSessionRequest struct {
	// ExecutionTimeout specifies a timeout in seconds for configuring
	// the session.  Use int64 to match the OpenAPI specification.
	ExecutionTimeout int64  `json:"executionTimeout,omitempty"`
	Statement        string `json:"statement"`
}

type ConnConfig struct {
	GatewayURL string
	Client     *http.Client
	APIVersion string
	Properties map[string]string
}

type ConnOption func(c *ConnConfig)

type ExecuteStatementRequest struct {
	ExecutionConfig map[string]string `json:"executionConfig,omitempty"`
	// ExecutionTimeout specifies a timeout in seconds for the statement
	// execution.  Use int64 to match the OpenAPI specification.
	ExecutionTimeout int64  `json:"executionTimeout,omitempty"`
	Statement        string `json:"statement"`
}

type ExecuteStatementResponse struct {
	OperationHandle string `json:"operationHandle"`
}

type FetchResultsResponseBody struct {
	JobID         string     `json:"jobID"`
	NextResultUri string     `json:"nextResultUri"`
	IsQueryResult bool       `json:"isQueryResult"`
	ResultKind    ResultKind `json:"resultKind"`
	ResultType    ResultType `json:"resultType"`
	Results       struct {
		Columns   []ColumnInfo `json:"columns"`
		RowFormat string       `json:"rowFormat"`
		Data      []RowData    `json:"data"`
	} `json:"results"`
}

type GatewayClient interface {
	GetInfo(ctx context.Context) (*InfoResponse, error)
	GetAPIVersions(ctx context.Context) ([]string, error)
	OpenSession(ctx context.Context, reqBody *OpenSessionRequest) (string, error)
	CloseSession(ctx context.Context, sessionHandle string) error
	GetSessionConfig(ctx context.Context, sessionHandle string) (map[string]string, error)
	CompleteStatement(ctx context.Context, sessionHandle string, reqBody *CompleteStatementRequest) ([]string, error)
	ConfigureSession(ctx context.Context, sessionHandle string, reqBody *ConfigureSessionRequest) error
	Heartbeat(ctx context.Context, sessionHandle string) error
	RefreshMaterializedTable(ctx context.Context, sessionHandle, identifier string, reqBody *RefreshMaterializedTableRequest) (string, error)
	CancelOperation(ctx context.Context, sessionHandle, operationHandle string) (string, error)
	CloseOperation(ctx context.Context, sessionHandle, operationHandle string) (string, error)
	GetOperationStatus(ctx context.Context, sessionHandle, operationHandle string) (string, error)
	ExecuteStatement(ctx context.Context, sessionHandle string, reqBody *ExecuteStatementRequest) (string, error)
	FetchResults(ctx context.Context, sessionHandle, operationHandle, token string, rowFormat string) (*FetchResultsResponseBody, error)
}

type InfoResponse struct {
	ProductName string `json:"productName"`
	Version     string `json:"version"`
}

type LogicalType struct {
	Type       string  `json:"type"`
	Nullable   bool    `json:"nullable"`
	Length     *int64  `json:"length"`
	Precision  *int    `json:"precision"`
	Scale      *int    `json:"scale"`
	Resolution *string `json:"resolution"`
}

type OpenSessionRequest struct {
	Properties  map[string]string `json:"properties,omitempty"`
	SessionName string            `json:"sessionName,omitempty"`
}

type OpenSessionResponse struct {
	SessionHandle string `json:"sessionHandle"`
}

type OperationStatusResponse struct {
	Status string `json:"status"`
}

type RefreshMaterializedTableRequest struct {
	DynamicOptions   map[string]string `json:"dynamicOptions,omitempty"`
	ExecutionConfig  map[string]string `json:"executionConfig,omitempty"`
	IsPeriodic       bool              `json:"isPeriodic,omitempty"`
	Periodic         bool              `json:"periodic,omitempty"`
	ScheduleTime     string            `json:"scheduleTime,omitempty"`
	StaticPartitions map[string]string `json:"staticPartitions,omitempty"`
}

type RefreshMaterializedTableResponse struct {
	OperationHandle string `json:"operationHandle"`
}

type ResultKind string

type ResultType string

type RowData struct {
	Kind   string            `json:"kind"`
	Fields []json.RawMessage `json:"fields"`
}

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