godatabend

type APIClient struct {
	SessionID string
	QuerySeq  int64

	WaitTimeSeconds      int64
	MaxRowsInBuffer      int64
	MaxRowsPerPage       int64
	PresignedURLDisabled bool
	EmptyFieldAs         string
	// contains filtered or unexported fields
}

type APIError struct {
	RespBody   APIErrorResponseBody
	RespText   string
	StatusCode int
	Hint       string
}

type APIErrorResponseBody struct {
	Error   string `json:"error"`
	Message string `json:"message"`
}

type AccessTokenLoader interface {
	// LoadAccessToken is called whenever a new request is made to the server.
	LoadAccessToken(ctx context.Context, forceRotate bool) (string, error)
}

type AuthMethod string

type Batch interface {
	AppendToFile(v []driver.Value) error
	BatchInsert() error
}

type Config struct {
	Tenant    string // Tenant
	Warehouse string // Warehouse
	User      string // Username
	Password  string // Password (requires User)
	Database  string // Database name

	Role string // Role is the databend role you want to use for the current connection

	AccessToken       string
	AccessTokenFile   string // path to file containing access token, it can be used to rotate access token
	AccessTokenLoader AccessTokenLoader

	Host    string
	Timeout time.Duration
	/* Pagination params: WaitTimeSecs,  MaxRowsInBuffer, MaxRowsPerPage
	Pagination: critical conditions for each HTTP request to return (before all remaining result is ready to return)
	Related docs:https://databend.rs/doc/integrations/api/rest#query-request
	*/
	WaitTimeSecs    int64
	MaxRowsInBuffer int64
	MaxRowsPerPage  int64
	Location        *time.Location
	Debug           bool
	GzipCompression bool
	Params          map[string]string
	TLSConfig       string
	SSLMode         string

	// track the progress of query execution
	StatsTracker QueryStatsTracker

	// used on the storage which does not support presigned url like HDFS, local fs
	PresignedURLDisabled bool

	// Specifies the value that should be used when encountering empty fields, including both ,, and ,"",, in the CSV data being loaded into the table.
	// https://docs.databend.com/sql/sql-reference/file-format-options#empty_field_as
	// default is `string`
	// databend version should >= v1.2.345-nightly
	EmptyFieldAs        string
	EnableOpenTelemetry bool
}

type ContextKey string

type DBLogger interface {
	rlog.Ext1FieldLogger
	SetLogLevel(level string) error
	WithContext(ctx context.Context) *rlog.Entry
	SetOutput(output io.Writer)
}

type DataField struct {
	Name string `json:"name"`
	Type string `json:"type"`
}

type DataParser interface {
	Parse(io.RuneScanner) (driver.Value, error)
	Type() reflect.Type
}

type DataParserOptions struct {
	// Location describes default location for DateTime and Date field without Timezone argument.
	Location *time.Location
	// UseDBLocation if false: always use Location, ignore DateTime argument.
	UseDBLocation bool
}

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

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

type Error struct {
	Code    int
	Message string
}

type FileAccessTokenData struct {
	AccessToken string `toml:"access_token"`
}

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

type PaginationConfig struct {
	WaitTime        int64 `json:"wait_time_secs,omitempty"`
	MaxRowsInBuffer int64 `json:"max_rows_in_buffer,omitempty"`
	MaxRowsPerPage  int64 `json:"max_rows_per_page,omitempty"`
}

type PresignedResponse struct {
	Method  string
	Headers map[string]string
	URL     string
}

type QueryError struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Kind    string `json:"kind"`
}

type QueryIDGenerator func() string

type QueryProgress struct {
	Bytes uint64 `json:"bytes"`
	Rows  uint64 `json:"rows"`
}

type QueryRequest struct {
	Session    *json.RawMessage  `json:"session,omitempty"`
	SQL        string            `json:"sql"`
	Pagination *PaginationConfig `json:"pagination,omitempty"`

	StageAttachment *StageAttachmentConfig `json:"stage_attachment,omitempty"`
}

type QueryResponse struct {
	ID      string           `json:"id"`
	Session *json.RawMessage `json:"session"`
	Schema  *[]DataField     `json:"schema"`
	Data    [][]string       `json:"data"`
	State   string           `json:"state"`
	Error   *QueryError      `json:"error"`
	Stats   *QueryStats      `json:"stats"`
	// TODO: Affect rows
	StatsURI string `json:"stats_uri"`
	FinalURI string `json:"final_uri"`
	NextURI  string `json:"next_uri"`
	KillURI  string `json:"kill_uri"`
}

type QueryStats struct {
	RunningTimeMS  float64       `json:"running_time_ms"`
	ScanProgress   QueryProgress `json:"scan_progress"`
	WriteProgress  QueryProgress `json:"write_progress"`
	ResultProgress QueryProgress `json:"result_progress"`
}

type QueryStatsTracker func(queryID string, stats *QueryStats)

type RequestType int

type ServerInfo struct {
	Id        string `json:"id"`
	StartTime string `json:"start_time"`
}

type SessionState struct {
	Database       string    `json:"database,omitempty"`
	Role           string    `json:"role,omitempty"`
	SecondaryRoles *[]string `json:"secondary_roles,omitempty"`

	Settings map[string]string `json:"settings,omitempty"`

	// txn
	TxnState string `json:"txn_state,omitempty"`
}

type StageAttachmentConfig struct {
	Location          string            `json:"location"`
	FileFormatOptions map[string]string `json:"file_format_options,omitempty"`
	CopyOptions       map[string]string `json:"copy_options,omitempty"`
}

type StageLocation struct {
	Name string
	Path string
}

type StaticAccessTokenLoader struct {
	AccessToken string
}

type TypeDesc struct {
	Name string
	Args []*TypeDesc
}