utils

Documentation

Overview

Package utils contains helper types and functions that can be used by any relational data source.

Index

• func MakeReadSplitQuery(logger log.Logger, formatter SQLFormatter, request *api_service_protos.TSelect) (string, []any, error)
• type Connection
• type ConnectionManager
• type ConnectionManagerBase
• type ConnectionManagerMock
  • func (m *ConnectionManagerMock) Make(ctx context.Context, logger log.Logger, ...) (Connection, error)
  • func (m *ConnectionManagerMock) Release(logger log.Logger, conn Connection)
• type ConnectionMock
  • func (m *ConnectionMock) Close() error
  • func (m *ConnectionMock) Query(ctx context.Context, query string, params ...any) (Rows, error)
• type DefaultSchemaProvider
  • func (f *DefaultSchemaProvider) GetSchema(ctx context.Context, logger my_log.Logger, conn Connection, ...) (*api_service_protos.TSchema, error)
• type RowTransformerMock
  • func (t *RowTransformerMock) AppendToArrowBuilders(builder []array.Builder) error
  • func (t *RowTransformerMock) GetAcceptors() []any
  • func (t *RowTransformerMock) SetAcceptors([]any)
• type Rows
• type RowsMock
  • func (m *RowsMock) Close() error
  • func (m *RowsMock) Err() error
  • func (m *RowsMock) MakeTransformer(ydbType []*Ydb.Type) (utils.RowTransformer[any], error)
  • func (m *RowsMock) Next() bool
  • func (m *RowsMock) Scan(dest ...any) error
• type SQLFormatter
• type SchemaBuilder
  • func NewSchemaBuilder(typeMapper utils.TypeMapper, ...) *SchemaBuilder
  • func (sb *SchemaBuilder) AddColumn(columnName, columnType string) error
  • func (sb *SchemaBuilder) Build(logger log.Logger) (*api_service_protos.TSchema, error)
• type SchemaProvider
  • func NewDefaultSchemaProvider(typeMapper utils.TypeMapper, ...) SchemaProvider

Functions

func MakeReadSplitQuery

func MakeReadSplitQuery(logger log.Logger, formatter SQLFormatter, request *api_service_protos.TSelect) (string, []any, error)

Types

type Connection

type Connection interface {
	Query(ctx context.Context, query string, args ...any) (Rows, error)
	Close() error
}

type ConnectionManager

type ConnectionManager interface {
	Make(ctx context.Context, logger log.Logger, dataSourceInstance *api_common.TDataSourceInstance) (Connection, error)
	Release(logger log.Logger, connection Connection)
}

type ConnectionManagerBase

type ConnectionManagerBase struct {
	QueryLoggerFactory utils.QueryLoggerFactory
}

type ConnectionManagerMock

type ConnectionManagerMock struct {
	mock.Mock
}

func (*ConnectionManagerMock) Make

func (m *ConnectionManagerMock) Make(
	ctx context.Context,
	logger log.Logger,
	dataSourceInstance *api_common.TDataSourceInstance) (Connection, error)

func (*ConnectionManagerMock) Release

func (m *ConnectionManagerMock) Release(logger log.Logger, conn Connection)

type ConnectionMock

type ConnectionMock struct {
	mock.Mock
}

func (*ConnectionMock) Close

func (m *ConnectionMock) Close() error

func (*ConnectionMock) Query

func (m *ConnectionMock) Query(ctx context.Context, query string, params ...any) (Rows, error)

type DefaultSchemaProvider

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

func (*DefaultSchemaProvider) GetSchema

func (f *DefaultSchemaProvider) GetSchema(
	ctx context.Context,
	logger my_log.Logger,
	conn Connection,
	request *api_service_protos.TDescribeTableRequest,
) (*api_service_protos.TSchema, error)

type RowTransformerMock

type RowTransformerMock struct {
	mock.Mock
	Acceptors []any
}

func (*RowTransformerMock) AppendToArrowBuilders

func (t *RowTransformerMock) AppendToArrowBuilders(builder []array.Builder) error

func (*RowTransformerMock) GetAcceptors

func (t *RowTransformerMock) GetAcceptors() []any

func (*RowTransformerMock) SetAcceptors

func (t *RowTransformerMock) SetAcceptors([]any)

type Rows

type Rows interface {
	Close() error
	Err() error
	Next() bool
	Scan(dest ...any) error
	MakeTransformer(ydbTypes []*Ydb.Type) (utils.RowTransformer[any], error)
}

type RowsMock

type RowsMock struct {
	mock.Mock
	PredefinedData [][]any
	// contains filtered or unexported fields
}

func (*RowsMock) Close

func (m *RowsMock) Close() error

func (*RowsMock) Err

func (m *RowsMock) Err() error

func (*RowsMock) MakeTransformer

func (m *RowsMock) MakeTransformer(ydbType []*Ydb.Type) (utils.RowTransformer[any], error)

func (*RowsMock) Next

func (m *RowsMock) Next() bool

func (*RowsMock) Scan

func (m *RowsMock) Scan(dest ...any) error

type SQLFormatter

type SQLFormatter interface {
	// Get placeholder for n'th argument (starting from 0) for prepared statement
	GetPlaceholder(n int) string

	// Sanitize names of databases, tables, columns, views, schemas
	SanitiseIdentifier(ident string) string

	// Support for high level expression (without subexpressions, they are checked separately)
	SupportsPushdownExpression(expression *api_service_protos.TExpression) bool
}

type SchemaBuilder

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

func NewSchemaBuilder

func NewSchemaBuilder(
	typeMapper utils.TypeMapper,
	typeMappingSettings *api_service_protos.TTypeMappingSettings,
) *SchemaBuilder

func (*SchemaBuilder) AddColumn

func (sb *SchemaBuilder) AddColumn(columnName, columnType string) error

func (*SchemaBuilder) Build

func (sb *SchemaBuilder) Build(logger log.Logger) (*api_service_protos.TSchema, error)

type SchemaProvider

type SchemaProvider interface {
	GetSchema(
		ctx context.Context,
		logger log.Logger,
		conn Connection,
		request *api_service_protos.TDescribeTableRequest,
	) (*api_service_protos.TSchema, error)
}

func NewDefaultSchemaProvider

func NewDefaultSchemaProvider(
	typeMapper utils.TypeMapper,
	getArgsAndQueryFunc func(request *api_service_protos.TDescribeTableRequest) (string, []any),
) SchemaProvider