clickhouse

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

type ClientInterface interface {
	// Start initializes the client
	Start() error
	// Stop closes the client
	Stop() error
	// SetNetwork updates the network name for metrics labeling
	SetNetwork(network string)
	// Do executes a ch-go query directly for streaming operations
	Do(ctx context.Context, query ch.Query) error
	// Execute runs a query without expecting results
	Execute(ctx context.Context, query string) error
	// IsStorageEmpty checks if a table has any records matching the given conditions
	IsStorageEmpty(ctx context.Context, table string, conditions map[string]any) (bool, error)

	// QueryUInt64 executes a query and returns a single UInt64 value from the specified column.
	// Returns nil if no rows are found.
	QueryUInt64(ctx context.Context, query string, columnName string) (*uint64, error)
	// QueryMinMaxUInt64 executes a query that returns min and max UInt64 values.
	// The query must return columns named "min" and "max".
	// Returns nil for both values if no rows are found.
	QueryMinMaxUInt64(ctx context.Context, query string) (minVal, maxVal *uint64, err error)
	// QueryUInt64Slice executes a query and returns all UInt64 values from the specified column.
	// Returns an empty slice if no rows are found.
	QueryUInt64Slice(ctx context.Context, query string, columnName string) ([]uint64, error)
}

type Config struct {
	// Connection
	Addr     string `yaml:"addr"`     // Native protocol address, e.g., "localhost:9000"
	Database string `yaml:"database"` // Database name, default: "default"
	Username string `yaml:"username"`
	Password string `yaml:"password"` //nolint:gosec // G117: config field, not a hardcoded secret

	// Pool settings
	MaxConns          int32         `yaml:"max_conns"`           // Maximum connections in pool, default: 10
	MinConns          int32         `yaml:"min_conns"`           // Minimum connections in pool, default: 2
	ConnMaxLifetime   time.Duration `yaml:"conn_max_lifetime"`   // Maximum connection lifetime, default: 1h
	ConnMaxIdleTime   time.Duration `yaml:"conn_max_idle_time"`  // Maximum idle time, default: 30m
	HealthCheckPeriod time.Duration `yaml:"health_check_period"` // Health check period, default: 1m
	DialTimeout       time.Duration `yaml:"dial_timeout"`        // Dial timeout, default: 10s

	// Performance
	Compression string `yaml:"compression"` // Compression: lz4, zstd, none (default: lz4)

	// Retry settings
	MaxRetries     int           `yaml:"max_retries"`      // Maximum retry attempts, default: 3
	RetryBaseDelay time.Duration `yaml:"retry_base_delay"` // Base delay for exponential backoff, default: 100ms
	RetryMaxDelay  time.Duration `yaml:"retry_max_delay"`  // Max delay between retries, default: 10s

	// Timeout settings
	QueryTimeout time.Duration `yaml:"query_timeout"` // Query timeout per attempt, default: 60s

	// Metrics labels
	Network   string `yaml:"network"`
	Processor string `yaml:"processor"`
	Debug     bool   `yaml:"debug"`
}

type MockCall struct {
	Method string
	Args   []any
}

type MockClient struct {
	// Function fields that can be set by tests
	ExecuteFunc           func(ctx context.Context, query string) error
	IsStorageEmptyFunc    func(ctx context.Context, table string, conditions map[string]any) (bool, error)
	StartFunc             func() error
	StopFunc              func() error
	DoFunc                func(ctx context.Context, query ch.Query) error
	QueryUInt64Func       func(ctx context.Context, query string, columnName string) (*uint64, error)
	QueryMinMaxUInt64Func func(ctx context.Context, query string) (minVal, maxVal *uint64, err error)
	QueryUInt64SliceFunc  func(ctx context.Context, query string, columnName string) ([]uint64, error)

	// Track calls for assertions
	Calls []MockCall
}