queue

type AttackConfiguration struct {
	MaxRetries      int           `json:"max_retries"`
	Timeout         time.Duration `json:"timeout"`
	RateLimit       int           `json:"rate_limit"`
	ConcurrentLimit int           `json:"concurrent_limit"`
	StopOnSuccess   bool          `json:"stop_on_success"`
	CollectMetrics  bool          `json:"collect_metrics"`
}

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

type AttackJobPayload struct {
	ProviderType  core.ProviderType      `json:"provider_type"`
	Model         string                 `json:"model"`
	TemplateID    string                 `json:"template_id"`
	Parameters    map[string]interface{} `json:"parameters"`
	Configuration AttackConfiguration    `json:"configuration"`
	Metadata      map[string]interface{} `json:"metadata"`
}

type AttackResult struct {
	JobID         string                 `json:"job_id"`
	Success       bool                   `json:"success"`
	Response      string                 `json:"response"`
	Confidence    float64                `json:"confidence"`
	AttackType    string                 `json:"attack_type"`
	ProviderType  core.ProviderType      `json:"provider_type"`
	Model         string                 `json:"model"`
	ExecutionTime time.Duration          `json:"execution_time"`
	TokensUsed    int                    `json:"tokens_used"`
	Cost          float64                `json:"cost"`
	Metadata      map[string]interface{} `json:"metadata"`
	Error         string                 `json:"error,omitempty"`
	Timestamp     time.Time              `json:"timestamp"`
}

type BatchAttackPayload struct {
	Attacks       []*AttackJobPayload `json:"attacks"`
	Configuration AttackConfiguration `json:"configuration"`
}

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

type ExecutionResult struct {
	Success    bool                   `json:"success"`
	Response   string                 `json:"response"`
	Confidence float64                `json:"confidence"`
	TokensUsed int                    `json:"tokens_used"`
	Cost       float64                `json:"cost"`
	Metadata   map[string]interface{} `json:"metadata"`
}

type Job struct {
	ID          string                 `json:"id"`
	Type        string                 `json:"type"`
	Queue       string                 `json:"queue"`
	Priority    int                    `json:"priority"`
	Payload     map[string]interface{} `json:"payload"`
	CreatedAt   time.Time              `json:"created_at"`
	ScheduledAt time.Time              `json:"scheduled_at"`
	StartedAt   *time.Time             `json:"started_at,omitempty"`
	CompletedAt *time.Time             `json:"completed_at,omitempty"`
	FailedAt    *time.Time             `json:"failed_at,omitempty"`
	Attempts    int                    `json:"attempts"`
	MaxRetries  int                    `json:"max_retries"`
	Error       string                 `json:"error,omitempty"`
	Result      interface{}            `json:"result,omitempty"`
	Status      JobStatus              `json:"status"`
}

type JobHandler interface {
	ProcessJob(ctx context.Context, job *Job) error
	GetJobTypes() []string
}

type JobStatus string

type Logger interface {
	Info(msg string, args ...interface{})
	Warn(msg string, args ...interface{})
	Error(msg string, args ...interface{})
	Debug(msg string, args ...interface{})
}

type Provider interface {
	GetModels(ctx context.Context) ([]string, error)
	Execute(ctx context.Context, request map[string]interface{}) (*ExecutionResult, error)
	GetName() string
}

type ProviderManager interface {
	GetProvider(providerType core.ProviderType) (Provider, error)
	ListProviders() []string
}

type QueueMetrics struct {
	JobsEnqueued   int64         `json:"jobs_enqueued"`
	JobsProcessed  int64         `json:"jobs_processed"`
	JobsFailed     int64         `json:"jobs_failed"`
	JobsRetried    int64         `json:"jobs_retried"`
	ActiveWorkers  int64         `json:"active_workers"`
	QueueLength    int64         `json:"queue_length"`
	AverageLatency time.Duration `json:"average_latency"`
	ThroughputRPS  float64       `json:"throughput_rps"`
}

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

type RedisQueueConfig struct {
	// Redis connection
	RedisAddr     string `json:"redis_addr"`
	RedisPassword string `json:"redis_password"`
	RedisDB       int    `json:"redis_db"`

	// Queue configuration
	QueuePrefix       string        `json:"queue_prefix"`
	DefaultPriority   int           `json:"default_priority"`
	MaxRetries        int           `json:"max_retries"`
	RetryDelay        time.Duration `json:"retry_delay"`
	JobTimeout        time.Duration `json:"job_timeout"`
	VisibilityTimeout time.Duration `json:"visibility_timeout"`

	// Worker configuration
	WorkerCount  int           `json:"worker_count"`
	PollInterval time.Duration `json:"poll_interval"`
	BatchSize    int           `json:"batch_size"`

	// Performance settings
	EnablePipelining bool          `json:"enable_pipelining"`
	PoolSize         int           `json:"pool_size"`
	MinIdleConns     int           `json:"min_idle_conns"`
	MaxConnAge       time.Duration `json:"max_conn_age"`

	// Monitoring
	EnableMetrics   bool          `json:"enable_metrics"`
	MetricsInterval time.Duration `json:"metrics_interval"`
}

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

type WorkerMetrics struct {
	WorkerID       string        `json:"worker_id"`
	JobsProcessed  int64         `json:"jobs_processed"`
	JobsFailed     int64         `json:"jobs_failed"`
	AverageLatency time.Duration `json:"average_latency"`
	LastActive     time.Time     `json:"last_active"`
	Status         string        `json:"status"`
}