plugin

type AllowNullFunc func(arrow.DataType) bool

type AssessCategory int

type AssessOptions struct {
	MigrateForce bool
}

type Assessor interface {
	AssessTables(ctx context.Context, tables []TablePair, options AssessOptions) ([]TableFinding, error)
}

type BackendOptions struct {
	TableName  string
	Connection string
}

type BuildTarget struct {
	OS             string   `json:"os"`
	Arch           string   `json:"arch"`
	CGO            bool     `json:"cgo"`
	Env            []string `json:"env"`
	Tags           []string `json:"tags"`
	IncludeSymbols bool     `json:"include_symbols"`
}

type Client interface {
	SourceClient
	DestinationClient
	TransformerClient
}

type ColumnFinding struct {
	ColumnName         string
	Category           AssessCategory
	OldType            string
	NewType            string
	SafeModeBehavior   string
	ForcedModeBehavior string
	Evidence           []Evidence
}

type ConnectionTester func(ctx context.Context, logger zerolog.Logger, spec []byte) error

type DestinationClient interface {
	Close(ctx context.Context) error
	Read(ctx context.Context, table *schema.Table, res chan<- arrow.RecordBatch) error
	Write(ctx context.Context, res <-chan message.WriteMessage) error
}

type Evidence struct {
	SyntheticValue string
	Before         string
	After          string
}

type Kind string

type Meta struct {
	Team            cqapi.PluginTeam
	Kind            cqapi.PluginKind
	Name            cqapi.PluginName
	SkipUsageClient bool
}

type MigrateMode int

type Migrations struct {
	RemoveUniqueConstraint bool
	MovePKToCQOnly         bool
}

type NewClientFunc func(context.Context, zerolog.Logger, []byte, NewClientOptions) (Client, error)

type NewClientOptions struct {
	NoConnection bool
	InvocationID string
	PluginMeta   Meta
}

type NewPluginFunc func() *Plugin

type NewSourceClientFunc func(context.Context, zerolog.Logger, any) (SourceClient, error)

type OnBeforeSender interface {
	OnBeforeSend(context.Context, message.SyncMessage) (message.SyncMessage, error)
}

type OnSyncFinisher interface {
	OnSyncFinish(context.Context) error
}

type Option func(*Plugin)

type PackageType string

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

type SafeMigrations struct {
	AddColumn              bool
	AddColumnNotNull       bool
	RemoveColumn           bool
	RemoveColumnNotNull    bool
	ChangeColumn           bool
	RemoveUniqueConstraint bool
	MovePKToCQOnly         bool
}

type Shard struct {
	Num   int32
	Total int32
}

type SourceClient interface {
	Close(ctx context.Context) error
	Tables(ctx context.Context, options TableOptions) (schema.Tables, error)
	Sync(ctx context.Context, options SyncOptions, res chan<- message.SyncMessage) error
}

type SyncOptions struct {
	Tables              []string
	SkipTables          []string
	SkipDependentTables bool
	DeterministicCQID   bool
	BackendOptions      *BackendOptions
	Shard               *Shard
}

type TableFinding struct {
	TableName                string
	Category                 AssessCategory
	SafeModeBehavior         string
	ForcedModeBehavior       string
	Columns                  []ColumnFinding
	Evidence                 []Evidence
	IncompleteCoverageReason string
}

type TableOptions struct {
	Tables              []string
	SkipTables          []string
	SkipDependentTables bool
}

type TablePair struct {
	Old *schema.Table
	New *schema.Table
}

type TestConnError struct {
	Code    string
	Message error
}

type TransformerClient interface {
	Transform(context.Context, <-chan arrow.RecordBatch, chan<- arrow.RecordBatch) error
	TransformSchema(context.Context, *arrow.Schema) (*arrow.Schema, error)
}

type UnimplementedDestination struct {
	UnimplementedTransformer
}

type UnimplementedSource struct {
	UnimplementedTransformer
}

type UnimplementedTransformer struct{}

type WriteTests struct {
	DuplicatePK bool
}

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

type WriterTestSuiteTests struct {
	// SkipUpsert skips testing with message.Insert and Upsert=true.
	// Usually when a destination is not supporting primary keys
	SkipUpsert bool

	// SkipDeleteStale skips testing message.Delete events.
	SkipDeleteStale bool

	// SkipDeleteRecord skips testing message.DeleteRecord events.
	SkipDeleteRecord bool

	// SkipAppend skips testing message.Insert and Upsert=false.
	SkipInsert bool

	// SkipMigrate skips testing migration
	SkipMigrate bool

	// SafeMigrations defines which tests should work with force migration
	// and which should pass with safe migration
	SafeMigrations SafeMigrations

	SkipSpecificMigrations Migrations

	SkipSpecificWriteTests WriteTests
}