ydb

type AuthTokenCredentials struct {
	AuthToken string
	// contains filtered or unexported fields
}

type Backoff interface {
	// Wait maps index of the operationCompleted to a channel which fulfillment means that
	// delay is over.
	//
	// Note that operationCompleted index begins from 0 and 0-th index means that it is the
	// first operationCompleted attempt after an initial error.
	Wait(n int) <-chan time.Time
}

type BackoffFunc func(n int) <-chan time.Time

type BackoffType uint8

type BalancingMethod uint

type CallInfo interface {
	EndpointInfo
}

type ConnState int8

type ConnStats struct {
	State        ConnState
	OpStarted    uint64
	OpFailed     uint64
	OpSucceed    uint64
	OpPerMinute  float64
	ErrPerMinute float64
	AvgOpTime    time.Duration
}

type Credentials interface {
	Token(context.Context) (string, error)
}

type CredentialsFunc func(context.Context) (string, error)

type CustomScanner interface {
	UnmarshalYDB(res RawScanner) error
}

type DialDoneInfo struct {
	Context context.Context
	Address string
	Error   error
}

type DialStartInfo struct {
	Context context.Context
	Address string
}

type Dialer struct {
	// DriverConfig is a driver configuration.
	DriverConfig *DriverConfig

	// TLSConfig specifies the TLS configuration to use for tls client.
	// If TLSConfig is zero then connections are insecure.
	TLSConfig *tls.Config

	// Timeout is the maximum amount of time a dial will wait for a connect to
	// complete.
	// If Timeout is zero then no timeout is used.
	Timeout time.Duration

	// Keepalive is the interval used to check whether inner connections are
	// still valid.
	// Dialer could increase keepalive interval if given value is too small.
	Keepalive time.Duration

	// NetDial is an optional function that may replace default network dialing
	// function such as net.Dial("tcp").
	// Deprecated: Use it for test purposes and special cases only. In most cases should be left empty.
	NetDial func(context.Context, string) (net.Conn, error)
}

type DiscoveryDoneInfo struct {
	Context   context.Context
	Endpoints []Endpoint
	Error     error
}

type DiscoveryStartInfo struct {
	Context context.Context
	Address string
}

type Driver interface {
	Call(context.Context, Operation) (CallInfo, error)
	StreamRead(context.Context, StreamOperation) (CallInfo, error)
	Close() error
}

type DriverConfig struct {
	// Database is a required database name.
	Database string

	// Credentials is an ydb client credentials.
	// In most cases Credentials are required.
	Credentials Credentials

	// Trace contains driver tracing options.
	Trace DriverTrace

	// RequestTimeout is the maximum amount of time a Call() will wait for an
	// operation to complete.
	// If RequestTimeout is zero then no timeout is used.
	RequestTimeout time.Duration

	// StreamTimeout is the maximum amount of time a StreamRead() will wait for
	// an operation to complete.
	// If StreamTimeout is zero then no timeout is used.
	StreamTimeout time.Duration

	// OperationTimeout is the maximum amount of time a YDB server will process
	// an operation. After timeout exceeds YDB will try to cancel operation and
	// regardless of the cancellation appropriate error will be returned to
	// the client.
	// If OperationTimeout is zero then no timeout is used.
	OperationTimeout time.Duration

	// OperationCancelAfter is the maximum amount of time a YDB server will process an
	// operation. After timeout exceeds YDB will try to cancel operation and if
	// it succeeds appropriate error will be returned to the client; otherwise
	// processing will be continued.
	// If OperationCancelAfter is zero then no timeout is used.
	OperationCancelAfter time.Duration

	// DiscoveryInterval is the frequency of background tasks of ydb endpoints
	// discovery.
	// If DiscoveryInterval is zero then the DefaultDiscoveryInterval is used.
	// If DiscoveryInterval is negative, then no background discovery prepared.
	DiscoveryInterval time.Duration

	// BalancingMethod is an algorithm used by the driver for endpoint
	// selection.
	// If BalancingMethod is zero then the DefaultBalancingMethod is used.
	BalancingMethod BalancingMethod

	// BalancingConfig is an optional configuration related to selected
	// BalancingMethod. That is, some balancing methods allow to be configured.
	BalancingConfig interface{}

	// PreferLocalEndpoints adds endpoint selection logic when local endpoints
	// are always used first.
	// When no alive local endpoints left other endpoints will be used.
	//
	// NOTE: some balancing methods (such as p2c) also may use knowledge of
	// endpoint's locality. Difference is that with PreferLocalEndpoints local
	// endpoints selected separately from others. That is, if there at least
	// one local endpoint it will be used regardless of its performance
	// indicators.
	//
	// NOTE: currently driver (and even ydb itself) does not track load factor
	// of each endpoint properly. Enabling this option may lead to the
	// situation, when all but one nodes in local datacenter become inactive
	// and all clients will overload this single instance very quickly. That
	// is, currently this option may be called as experimental.
	// You have been warned.
	PreferLocalEndpoints bool

	// RequestsType set an additional type hint to all requests.
	// It is needed only for debug purposes and advanced cases.
	RequestsType string

	// FastDial will make dialer return Driver as soon as 1st connection succeeds.
	// NB: it may be not the fastest node to serve requests.
	FastDial bool
}

type DriverTrace struct {
	OnDial func(DialStartInfo) func(DialDoneInfo)

	OnGetConn func(GetConnStartInfo) func(GetConnDoneInfo)

	OnPessimization func(PessimizationStartInfo) func(PessimizationDoneInfo)

	// Only for background.
	TrackConnStart func(TrackConnStartInfo)
	TrackConnDone  func(TrackConnDoneInfo)

	OnGetCredentials func(GetCredentialsStartInfo) func(GetCredentialsDoneInfo)

	OnDiscovery func(DiscoveryStartInfo) func(DiscoveryDoneInfo)

	OnOperation     func(OperationStartInfo) func(OperationDoneInfo)
	OnOperationWait func(OperationWaitInfo)

	OnStream func(StreamStartInfo) func(StreamDoneInfo)

	OnStreamRecv func(StreamRecvStartInfo) func(StreamRecvDoneInfo)
}

type Endpoint struct {
	Addr       string
	Port       int
	LoadFactor float32
	Local      bool
}

type EndpointInfo interface {
	// Deprecated: using raw conn does not handle endpoint reconnection
	Conn() *conn
	Address() string
}

type FeatureFlag = internal.FeatureFlag

type GetConnDoneInfo struct {
	Context context.Context
	Address string
	Error   error
}

type GetConnStartInfo struct {
	Context context.Context
}

type GetCredentialsDoneInfo struct {
	Context context.Context
	Token   bool
	Error   error
}

type GetCredentialsStartInfo struct {
	Context context.Context
}

type Issue struct {
	Message  string
	Code     uint32
	Severity uint32
}

type IssueIterator []*Ydb_Issue.IssueMessage

type LogBackoff struct {
	// SlotDuration is a size of a single time slot used in backoff delay
	// calculation.
	// If SlotDuration is less or equal to zero, then the time.Second value is
	// used.
	SlotDuration time.Duration

	// Ceiling is a maximum degree of backoff delay growth.
	// If Ceiling is less or equal to zero, then the default ceiling of 1 is
	// used.
	Ceiling uint

	// JitterLimit controls fixed and random portions of backoff delay.
	// Its value can be in range [0, 1].
	// If JitterLimit is non zero, then the backoff delay will be equal to (F + R),
	// where F is a result of multiplication of this value and calculated delay
	// duration D; and R is a random sized part from [0,(D - F)].
	JitterLimit float64
}

type MetaDatabaseCtxKey struct{}

type Method string

type NoOpResponse = internal.NoOpResponse

type OpError struct {
	Reason StatusCode
	// contains filtered or unexported fields
}

type OpResponse = internal.OpResponse

type Operation = internal.Operation

type OperationCompleted uint8

type OperationDoneInfo struct {
	Context context.Context
	Address string
	Method  Method
	Params  OperationParams
	OpID    string
	Issues  IssueIterator
	Error   error
}

type OperationMode uint

type OperationParams struct {
	Timeout     time.Duration
	CancelAfter time.Duration
	Mode        OperationMode
}

type OperationStartInfo struct {
	Context context.Context
	Address string
	Method  Method
	Params  OperationParams
}

type OperationWaitInfo struct {
	Context context.Context
	Address string
	Method  Method
	Params  OperationParams
	OpID    string
}

type P2CConfig struct {
	// PreferLocal reports whether p2c balancer should prefer local endpoint
	// when all other runtime indicators are the same (such as error rate or
	// average response time).
	PreferLocal bool

	// OpTimeThreshold specifies such difference between endpoint average
	// operation time when it becomes significant to be used during comparison.
	OpTimeThreshold time.Duration
}

type PessimizationDoneInfo struct {
	Context context.Context
	Address string
	Error   error
}

type PessimizationStartInfo struct {
	Context context.Context
	Address string
	Cause   error
}

type RawScanner interface {
	HasItems() bool
	HasNextItem() bool

	// NextItem selects next item to parse in the current row.
	// It returns false if there are no more items in the row.
	//
	// Note that NextItem() differs from NextRow() and NextSet() – if it return
	// false it fails the Result such that no further operations may be processed.
	// That is, res.Err() becomes non-nil.
	NextItem() (ok bool)

	// SeekItem finds the column with given name in the result set and selects
	// appropriate item to parse in the current row.
	SeekItem(name string) bool
	Path() string
	WritePathTo(w io.Writer) (n int64, err error)
	Type() Type
	Bool() (v bool)
	Int8() (v int8)
	Uint8() (v uint8)
	Int16() (v int16)
	Uint16() (v uint16)
	Int32() (v int32)
	Uint32() (v uint32)
	Int64() (v int64)
	Uint64() (v uint64)
	Float() (v float32)
	Double() (v float64)
	Date() (v time.Time)
	Datetime() (v time.Time)
	Timestamp() (v time.Time)
	Interval() (v time.Duration)
	TzDate() (v time.Time)
	TzDatetime() (v time.Time)
	TzTimestamp() (v time.Time)
	String() (v string)
	UTF8() (v string)
	YSON() (v []byte)
	JSON() (v []byte)
	UUID() (v [16]byte)
	JSONDocument() (v []byte)
	DyNumber() (v string)
	Value() Value

	// ListIn interprets current item under scan as a ydb's list.
	// It returns the size of the nested items.
	// If current item under scan is not a list type, it returns -1.
	ListIn() (size int)

	// ListItem selects current item i-th element as an item to scan.
	// ListIn() must be called before.
	ListItem(i int)

	// ListOut leaves list entered before by ListIn() call.
	ListOut()

	// TupleIn interprets current item under scan as a ydb's tuple.
	// It returns the size of the nested items.
	TupleIn() (size int)

	// TupleItem selects current item i-th element as an item to scan.
	// Note that TupleIn() must be called before.
	// It panics if it is out of bounds.
	TupleItem(i int)

	// TupleOut leaves tuple entered before by TupleIn() call.
	TupleOut()

	// StructIn interprets current item under scan as a ydb's struct.
	// It returns the size of the nested items – the struct fields values.
	// If there is no current item under scan it returns -1.
	StructIn() (size int)

	// StructField selects current item i-th field value as an item to scan.
	// Note that StructIn() must be called before.
	// It panics if i is out of bounds.
	StructField(i int) (name string)

	// StructOut leaves struct entered before by StructIn() call.
	StructOut()

	// DictIn interprets current item under scan as a ydb's dict.
	// It returns the size of the nested items pairs.
	// If there is no current item under scan it returns -1.
	DictIn() (size int)

	// DictKey selects current item i-th pair key as an item to scan.
	// Note that DictIn() must be called before.
	// It panics if i is out of bounds.
	DictKey(i int)

	// DictPayload selects current item i-th pair value as an item to scan.
	// Note that DictIn() must be called before.
	// It panics if i is out of bounds.
	DictPayload(i int)

	// DictOut leaves dict entered before by DictIn() call.
	DictOut()

	// Variant unwraps current item under scan interpreting it as Variant<T> type.
	// It returns non-empty name of a field that is filled for struct-based
	// variant.
	// It always returns an index of filled field of a T.
	Variant() (name string, index uint32)

	// Decimal returns decimal value represented by big-endian 128 bit signed integer.
	Decimal(t Type) (v [16]byte)
	ODecimal(t Type) (v [16]byte)

	// UnwrapDecimal returns decimal value represented by big-endian 128 bit signed
	// integer and its type information.
	UnwrapDecimal() (v [16]byte, precision, scale uint32)

	// Unwrap unwraps current item under scan interpreting it as Optional<T> type.
	Unwrap()
	IsNull() bool
	IsOptional() bool
	IsDecimal() bool
	Err() error
}

type RawValue interface {
	Path() string
	WritePathTo(w io.Writer) (n int64, err error)
	Type() Type
	Bool() (v bool)
	Int8() (v int8)
	Uint8() (v uint8)
	Int16() (v int16)
	Uint16() (v uint16)
	Int32() (v int32)
	Uint32() (v uint32)
	Int64() (v int64)
	Uint64() (v uint64)
	Float() (v float32)
	Double() (v float64)
	Date() (v time.Time)
	Datetime() (v time.Time)
	Timestamp() (v time.Time)
	Interval() (v time.Duration)
	TzDate() (v time.Time)
	TzDatetime() (v time.Time)
	TzTimestamp() (v time.Time)
	String() (v string)
	UTF8() (v string)
	YSON() (v []byte)
	JSON() (v []byte)
	UUID() (v [16]byte)
	JSONDocument() (v []byte)
	DyNumber() (v string)
	Value() Value

	// Any returns any primitive or optional value.
	// Currently, it may return one of these types:
	//
	//   bool
	//   int8
	//   uint8
	//   int16
	//   uint16
	//   int32
	//   uint32
	//   int64
	//   uint64
	//   float32
	//   float64
	//   []byte
	//   string
	//   [16]byte
	//
	Any() interface{}

	// Unwrap unwraps current item under scan interpreting it as Optional<T> type.
	Unwrap()
	AssertType(t Type) bool
	IsNull() bool
	IsOptional() bool

	// ListIn interprets current item under scan as a ydb's list.
	// It returns the size of the nested items.
	// If current item under scan is not a list type, it returns -1.
	ListIn() (size int)

	// ListItem selects current item i-th element as an item to scan.
	// ListIn() must be called before.
	ListItem(i int)

	// ListOut leaves list entered before by ListIn() call.
	ListOut()

	// TupleIn interprets current item under scan as a ydb's tuple.
	// It returns the size of the nested items.
	TupleIn() (size int)

	// TupleItem selects current item i-th element as an item to scan.
	// Note that TupleIn() must be called before.
	// It panics if it is out of bounds.
	TupleItem(i int)

	// TupleOut leaves tuple entered before by TupleIn() call.
	TupleOut()

	// StructIn interprets current item under scan as a ydb's struct.
	// It returns the size of the nested items – the struct fields values.
	// If there is no current item under scan it returns -1.
	StructIn() (size int)

	// StructField selects current item i-th field value as an item to scan.
	// Note that StructIn() must be called before.
	// It panics if i is out of bounds.
	StructField(i int) (name string)

	// StructOut leaves struct entered before by StructIn() call.
	StructOut()

	// DictIn interprets current item under scan as a ydb's dict.
	// It returns the size of the nested items pairs.
	// If there is no current item under scan it returns -1.
	DictIn() (size int)

	// DictKey selects current item i-th pair key as an item to scan.
	// Note that DictIn() must be called before.
	// It panics if i is out of bounds.
	DictKey(i int)

	// DictPayload selects current item i-th pair value as an item to scan.
	// Note that DictIn() must be called before.
	// It panics if i is out of bounds.
	DictPayload(i int)

	// DictOut leaves dict entered before by DictIn() call.
	DictOut()

	// Variant unwraps current item under scan interpreting it as Variant<T> type.
	// It returns non-empty name of a field that is filled for struct-based
	// variant.
	// It always returns an index of filled field of a T.
	Variant() (name string, index uint32)

	// Decimal returns decimal value represented by big-endian 128 bit signed integer.
	Decimal(t Type) (v [16]byte)
	ODecimal(t Type) (v [16]byte)

	// UnwrapDecimal returns decimal value represented by big-endian 128 bit signed
	// integer and its type information.
	UnwrapDecimal() (v [16]byte, precision, scale uint32)
	IsDecimal() bool
	Err() error
}

type Response = internal.Response

type RetryChecker struct {
	// RetryNotFound reports whether Repeater must operationCompleted ErrNotFound errors.
	// Deprecated: has no effect now
	RetryNotFound bool
}

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

type Scanner interface {
	UnmarshalYDB(res RawValue) error
}

type StatusCode int32

type StreamDoneInfo struct {
	Context context.Context
	Address string
	Method  Method
	Error   error
}

type StreamOperation = internal.StreamOperation

type StreamOperationResponse = internal.StreamOperationResponse

type StreamRecvDoneInfo struct {
	Context context.Context
	Address string
	Method  Method
	Issues  IssueIterator
	Error   error
}

type StreamRecvStartInfo struct {
	Context context.Context
	Address string
	Method  Method
}

type StreamStartInfo struct {
	Context context.Context
	Address string
	Method  Method
}

type StructOption func(*tStructType)

type StructValueOption func(*tStructValueProto)

type TrackConnDoneInfo struct {
	Address string
}

type TrackConnStartInfo struct {
	Address string
}

type TransportError struct {
	Reason TransportErrorCode
	// contains filtered or unexported fields
}

type TransportErrorCode int32

type Type interface {
	internal.T
}

type Value interface {
	internal.V
}

type WG interface {
	Done()
	Wait()
	Add(delta int)
}