nntppool

type ActiveConnectionMetrics struct {
	Count                int           `json:"count"`
	TotalBytesDownloaded int64         `json:"total_bytes_downloaded"`
	TotalBytesUploaded   int64         `json:"total_bytes_uploaded"`
	TotalCommands        int64         `json:"total_commands"`
	TotalCommandErrors   int64         `json:"total_command_errors"`
	SuccessRate          float64       `json:"success_rate_percent"`
	AverageConnectionAge time.Duration `json:"average_connection_age"`
}

type AggregatedMetrics struct {
	TotalBytesDownloaded   int64
	TotalBytesUploaded     int64
	TotalCommands          int64
	TotalCommandErrors     int64
	TotalArticlesRetrieved int64
	TotalArticlesPosted    int64
	SuccessRate            float64
	AverageConnectionAge   time.Duration
}

type Config struct {
	Logger              Logger
	NntpCli             nntpcli.Client
	Providers           []UsenetProviderConfig
	HealthCheckInterval time.Duration
	MinConnections      int
	MaxRetries          uint
	DelayType           DelayType
	// Deprecated: now is always false
	SkipProvidersVerificationOnCreation bool
	RetryDelay                          time.Duration
	ShutdownTimeout                     time.Duration
	DefaultConnectionLease              time.Duration
	ProviderReconnectInterval           time.Duration
	ProviderMaxReconnectInterval        time.Duration
	ProviderHealthCheckStagger          time.Duration
	ProviderHealthCheckTimeout          time.Duration
	// contains filtered or unexported fields
}

type ConnectionCleanupStats struct {
	TrackedConnections int           `json:"tracked_connections"`
	LastCleanupTime    time.Time     `json:"last_cleanup_time"`
	CleanupInterval    time.Duration `json:"cleanup_interval"`
	ConnectionTimeout  time.Duration `json:"connection_timeout"`
	NextCleanupTime    time.Time     `json:"next_cleanup_time"`
}

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

type ConnectionProviderInfo struct {
	Host                     string        `json:"host"`
	Username                 string        `json:"username"`
	MaxConnections           int           `json:"maxConnections"`
	MaxConnectionIdleTimeout time.Duration `json:"maxConnectionIdleTimeout"`
	State                    ProviderState `json:"state"`
}

type DelayType int

type Logger interface {
	Debug(msg string, args ...any)
	Info(msg string, args ...any)
	Warn(msg string, args ...any)
	Error(msg string, args ...any)
	DebugContext(ctx context.Context, msg string, args ...any)
	InfoContext(ctx context.Context, msg string, args ...any)
	WarnContext(ctx context.Context, msg string, args ...any)
	ErrorContext(ctx context.Context, msg string, args ...any)
}

type MemoryUsage struct {
	AllocatedBytes      uint64 `json:"allocated_bytes"`
	TotalAllocatedBytes uint64 `json:"total_allocated_bytes"`
	SystemBytes         uint64 `json:"system_bytes"`
	GCCount             uint32 `json:"gc_count"`
	ThresholdBytes      uint64 `json:"threshold_bytes"`
	ThresholdExceeded   bool   `json:"threshold_exceeded"`
}

type MetricRetentionConfig struct {
	// How long to keep detailed metrics before rotation
	DetailedRetentionDuration time.Duration

	// How often to perform metric rotation
	RotationInterval time.Duration

	// Maximum number of historical windows to keep
	MaxHistoricalWindows int

	// Memory threshold (bytes) that triggers aggressive cleanup
	MemoryThresholdBytes uint64

	// Enable automatic cleanup when memory thresholds are exceeded
	AutoCleanupEnabled bool
}

type MetricSummary struct {
	StartTime time.Time `json:"start_time"`
	EndTime   time.Time `json:"end_time"`

	// Aggregated core metrics
	TotalConnectionsCreated   int64 `json:"total_connections_created"`
	TotalConnectionsDestroyed int64 `json:"total_connections_destroyed"`
	TotalAcquires             int64 `json:"total_acquires"`
	TotalReleases             int64 `json:"total_releases"`
	TotalErrors               int64 `json:"total_errors"`
	TotalRetries              int64 `json:"total_retries"`
	TotalAcquireWaitTime      int64 `json:"total_acquire_wait_time_ns"`

	// Aggregated traffic metrics
	TotalBytesDownloaded   int64 `json:"total_bytes_downloaded"`
	TotalBytesUploaded     int64 `json:"total_bytes_uploaded"`
	TotalArticlesRetrieved int64 `json:"total_articles_retrieved"`
	TotalArticlesPosted    int64 `json:"total_articles_posted"`
	TotalCommandCount      int64 `json:"total_command_count"`
	TotalCommandErrors     int64 `json:"total_command_errors"`

	// Computed metrics
	AverageConnectionsPerHour float64 `json:"average_connections_per_hour"`
	AverageErrorRate          float64 `json:"average_error_rate_percent"`
	AverageSuccessRate        float64 `json:"average_success_rate_percent"`
	AverageAcquireWaitTime    int64   `json:"average_acquire_wait_time_ns"`

	// Number of windows that were summarized
	WindowCount int `json:"window_count"`
}

type MetricWindow struct {
	StartTime time.Time `json:"start_time"`
	EndTime   time.Time `json:"end_time"`

	// Core metrics for this window
	ConnectionsCreated   int64 `json:"connections_created"`
	ConnectionsDestroyed int64 `json:"connections_destroyed"`
	Acquires             int64 `json:"acquires"`
	Releases             int64 `json:"releases"`
	Errors               int64 `json:"errors"`
	Retries              int64 `json:"retries"`
	AcquireWaitTime      int64 `json:"acquire_wait_time_ns"`

	// Traffic metrics
	BytesDownloaded   int64 `json:"bytes_downloaded"`
	BytesUploaded     int64 `json:"bytes_uploaded"`
	ArticlesRetrieved int64 `json:"articles_retrieved"`
	ArticlesPosted    int64 `json:"articles_posted"`
	CommandCount      int64 `json:"command_count"`
	CommandErrors     int64 `json:"command_errors"`
}

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

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

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

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

type Option func(*Config)

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

type PoolMetricsSnapshot struct {
	// Timestamp and uptime
	Timestamp time.Time     `json:"timestamp"`
	Uptime    time.Duration `json:"uptime"`

	// Connection metrics
	TotalConnectionsCreated   int64 `json:"total_connections_created"`
	TotalConnectionsDestroyed int64 `json:"total_connections_destroyed"`
	ActiveConnections         int64 `json:"active_connections"`
	TotalAcquires             int64 `json:"total_acquires"`
	TotalReleases             int64 `json:"total_releases"`

	// Pool utilization
	AcquiredConnections int32 `json:"acquired_connections"`
	IdleConnections     int32 `json:"idle_connections"`
	TotalConnections    int32 `json:"total_connections"`

	// Traffic metrics
	TotalBytesDownloaded    int64   `json:"total_bytes_downloaded"`
	TotalBytesUploaded      int64   `json:"total_bytes_uploaded"`
	TotalArticlesRetrieved  int64   `json:"total_articles_retrieved"`
	TotalArticlesPosted     int64   `json:"total_articles_posted"`
	DownloadSpeed           float64 `json:"download_speed_bytes_per_sec"`            // Recent speed (based on time window)
	UploadSpeed             float64 `json:"upload_speed_bytes_per_sec"`              // Recent speed (based on time window)
	HistoricalDownloadSpeed float64 `json:"historical_download_speed_bytes_per_sec"` // Average since pool start
	HistoricalUploadSpeed   float64 `json:"historical_upload_speed_bytes_per_sec"`   // Average since pool start
	SpeedCalculationWindow  float64 `json:"speed_calculation_window_seconds"`        // Time window used for recent speed
	SpeedCacheDuration      float64 `json:"speed_cache_duration_seconds"`            // Cache duration for speed calculations
	SpeedCacheAge           float64 `json:"speed_cache_age_seconds"`                 // Age of current cached speed values

	// Performance metrics
	TotalCommandCount      int64         `json:"total_command_count"`
	TotalCommandErrors     int64         `json:"total_command_errors"`
	CommandSuccessRate     float64       `json:"command_success_rate_percent"`
	AverageAcquireWaitTime time.Duration `json:"average_acquire_wait_time"`

	// Error metrics
	TotalErrors  int64   `json:"total_errors"`
	TotalRetries int64   `json:"total_retries"`
	ErrorRate    float64 `json:"error_rate_percent"`

	// Rolling metrics information
	CurrentWindowStartTime  time.Time `json:"current_window_start_time"`
	CurrentWindowEndTime    time.Time `json:"current_window_end_time"`
	HistoricalWindowCount   int       `json:"historical_window_count"`
	WindowRotationInterval  float64   `json:"window_rotation_interval_seconds"`
	DetailedRetentionPeriod float64   `json:"detailed_retention_period_hours"`
	MaxHistoricalWindows    int       `json:"max_historical_windows"`

	// Memory management
	MemoryThreshold    uint64    `json:"memory_threshold_bytes"`
	CurrentMemoryUsage uint64    `json:"current_memory_usage_bytes"`
	AutoCleanupEnabled bool      `json:"auto_cleanup_enabled"`
	LastMemoryCheck    time.Time `json:"last_memory_check"`

	// Connection cleanup statistics
	ConnectionCleanupStats ConnectionCleanupStats `json:"connection_cleanup_stats"`

	// Daily and weekly summaries
	DailySummary  *MetricSummary `json:"daily_summary,omitempty"`
	WeeklySummary *MetricSummary `json:"weekly_summary,omitempty"`

	// Provider-specific metrics
	ProviderMetrics []ProviderMetricsSnapshot `json:"provider_metrics"`
}

type PooledConnection interface {
	Connection() nntpcli.Connection
	Close() error
	Free() error
	Provider() ConnectionProviderInfo
	CreatedAt() time.Time
	IsLeaseExpired() bool
	IsMarkedForReplacement() bool
	MarkForReplacement()
	ExtendLease(duration time.Duration)
}

type Provider struct {
	Host                           string
	Username                       string
	Password                       string
	Port                           int
	MaxConnections                 int
	MaxConnectionIdleTimeInSeconds int
	IsBackupProvider               bool
}

type ProviderChange struct {
	ID         string                `json:"id"`
	ChangeType ProviderChangeType    `json:"change_type"`
	OldConfig  *UsenetProviderConfig `json:"old_config,omitempty"`
	NewConfig  *UsenetProviderConfig `json:"new_config,omitempty"`
	Priority   int                   `json:"priority"` // Migration priority (0 = highest)
}

type ProviderChangeType int

type ProviderConfig interface {
	ID() string
	GetHost() string
	GetUsername() string
	GetPassword() string
	GetPort() int
	GetMaxConnections() int
	GetMaxConnectionIdleTimeInSeconds() int
	GetMaxConnectionTTLInSeconds() int
	GetTLS() bool
	GetInsecureSSL() bool
	GetIsBackupProvider() bool
	GetVerifyCapabilities() []string
}

type ProviderInfo struct {
	Host                     string        `json:"host"`
	Username                 string        `json:"username"`
	UsedConnections          int           `json:"usedConnections"`
	MaxConnections           int           `json:"maxConnections"`
	MaxConnectionIdleTimeout time.Duration `json:"maxConnectionIdleTimeout"`
	State                    ProviderState `json:"state"`
	LastConnectionAttempt    time.Time     `json:"lastConnectionAttempt"`
	LastSuccessfulConnect    time.Time     `json:"lastSuccessfulConnect"`
	FailureReason            string        `json:"failureReason"`
	RetryCount               int           `json:"retryCount"`
	NextRetryAt              time.Time     `json:"nextRetryAt"`
}

type ProviderMetricsSnapshot struct {
	ProviderID string        `json:"provider_id"`
	Host       string        `json:"host"`
	Username   string        `json:"username"`
	State      ProviderState `json:"state"`

	// Pool statistics from puddle
	MaxConnections          int32         `json:"max_connections"`
	TotalConnections        int32         `json:"total_connections"`
	AcquiredConnections     int32         `json:"acquired_connections"`
	IdleConnections         int32         `json:"idle_connections"`
	ConstructingConnections int32         `json:"constructing_connections"`
	AcquireCount            int64         `json:"acquire_count"`
	AcquireDuration         time.Duration `json:"acquire_duration"`
	EmptyAcquireCount       int64         `json:"empty_acquire_count"`
	EmptyAcquireWaitTime    time.Duration `json:"empty_acquire_wait_time"`
	CanceledAcquireCount    int64         `json:"canceled_acquire_count"`

	// Connection-level aggregated metrics
	TotalBytesDownloaded   int64         `json:"total_bytes_downloaded"`
	TotalBytesUploaded     int64         `json:"total_bytes_uploaded"`
	TotalCommands          int64         `json:"total_commands"`
	TotalCommandErrors     int64         `json:"total_command_errors"`
	SuccessRate            float64       `json:"success_rate_percent"`
	AverageConnectionAge   time.Duration `json:"average_connection_age"`
	TotalArticlesRetrieved int64         `json:"total_articles_retrieved"`
	TotalArticlesPosted    int64         `json:"total_articles_posted"`
}

type ProviderState int

type ProviderStatus struct {
	ProviderID     string     `json:"provider_id"`
	Status         string     `json:"status"` // "pending", "migrating", "completed", "failed"
	ConnectionsOld int        `json:"connections_old"`
	ConnectionsNew int        `json:"connections_new"`
	StartTime      time.Time  `json:"start_time"`
	CompletedAt    *time.Time `json:"completed_at,omitempty"`
	Error          string     `json:"error,omitempty"`
}

type ReconfigurationStatus struct {
	ID          string                    `json:"id"`
	StartTime   time.Time                 `json:"start_time"`
	Status      string                    `json:"status"` // "running", "completed", "failed", "rolled_back"
	Changes     []ProviderChange          `json:"changes"`
	Progress    map[string]ProviderStatus `json:"progress"` // providerID -> status
	Error       string                    `json:"error,omitempty"`
	CompletedAt *time.Time                `json:"completed_at,omitempty"`
}

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

type RollingMetricsStatus struct {
	Enabled                   bool      `json:"enabled"`
	CurrentWindowActive       bool      `json:"current_window_active"`
	CurrentWindowStartTime    time.Time `json:"current_window_start_time"`
	CurrentWindowEndTime      time.Time `json:"current_window_end_time"`
	CurrentWindowDuration     float64   `json:"current_window_duration_seconds"`
	HistoricalWindowCount     int       `json:"historical_window_count"`
	MaxHistoricalWindows      int       `json:"max_historical_windows"`
	RotationInterval          float64   `json:"rotation_interval_seconds"`
	DetailedRetentionDuration float64   `json:"detailed_retention_duration_hours"`
	MemoryThreshold           uint64    `json:"memory_threshold_bytes"`
	AutoCleanupEnabled        bool      `json:"auto_cleanup_enabled"`
	LastMemoryCheck           time.Time `json:"last_memory_check"`
	MemoryCheckInterval       float64   `json:"memory_check_interval_seconds"`
}

type UsenetConnectionPool interface {
	GetConnection(
		ctx context.Context,

		skipProviders []string,

		useBackupProviders bool,
	) (PooledConnection, error)
	Body(
		ctx context.Context,
		msgID string,
		w io.Writer,
		nntpGroups []string,
	) (int64, error)
	BodyReader(
		ctx context.Context,
		msgID string,
		nntpGroups []string,
	) (nntpcli.ArticleBodyReader, error)
	Post(ctx context.Context, r io.Reader) error
	Stat(ctx context.Context, msgID string, nntpGroups []string) (int, error)
	GetProvidersInfo() []ProviderInfo
	GetProviderStatus(providerID string) (*ProviderInfo, bool)
	Reconfigure(...Config) error
	GetReconfigurationStatus(migrationID string) (*ReconfigurationStatus, bool)
	GetActiveReconfigurations() map[string]*ReconfigurationStatus
	GetMetrics() *PoolMetrics
	GetMetricsSnapshot() PoolMetricsSnapshot
	Quit()
}

type UsenetProviderConfig struct {
	Host                           string
	Username                       string
	Password                       string
	VerifyCapabilities             []string
	Port                           int
	MaxConnections                 int
	MaxConnectionIdleTimeInSeconds int
	MaxConnectionTTLInSeconds      int
	TLS                            bool
	InsecureSSL                    bool
	IsBackupProvider               bool
}