models

type BenchmarkResult struct {
	Latencies []time.Duration
	QPS       float64
}

type BenchmarkWorkload struct {
	// Queries is a list of all the SQL queries and their arguments for the workload.
	Queries []QueryWithArgs `json:"queries"`
}

type ColumnSchema struct {
	Name         string `json:"name"`
	DataType     string `json:"data_type"`
	IsNullable   bool   `json:"is_nullable"`
	IsPrimaryKey bool   `json:"is_primary_key"`
	Default      string `json:"default"`
}

type Config struct {
	TracePath       string `json:"trace_path"`
	SchemaPath      string `json:"schema_path"`
	WorkloadPath    string `json:"workload_path"`
	BaseMetricsPath string `json:"base_metrics_path"`
	CandMetricsPath string `json:"cand_metrics_path"`
	ReportPath      string `json:"report_path"`
	Executor        string `json:"executor"`
	Target          string `json:"target"`
}

type DatabaseSchema struct {
	Name   string         `json:"name"`
	Tables []*TableSchema `json:"tables"`
}

type DistributionType string

type IndexSchema struct {
	Name     string   `json:"name"`
	Columns  []string `json:"columns"`
	IsUnique bool     `json:"is_unique"`
}

type Job struct {
	ID        string    `json:"id"`
	Status    JobStatus `json:"status"`
	Config    *Config   `json:"config"`
	Report    *Report   `json:"report,omitempty"`
	Error     string    `json:"error,omitempty"`
	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`
}

type JobStatus string

type ParameterModel struct {
	ParamName string           `json:"param_name"`
	DataType  string           `json:"data_type"` // e.g., "INT", "VARCHAR"
	DistType  DistributionType `json:"dist_type"` // Detected distribution

	// Zipf specific
	ZipfS float64 `json:"zipf_s,omitempty"` // Skewness parameter
	ZipfV float64 `json:"zipf_v,omitempty"` // Value range parameter (usually 1)

	// Stats
	Cardinality  int     `json:"cardinality"`
	HotspotRatio float64 `json:"hotspot_ratio"` // e.g., 0.8 means 20% items get 80% traffic

	// Sample Values (for reproduction)
	// We store TopValues and TopFrequencies to support empirical sampling of hotspots.
	TopValues      []interface{} `json:"top_values"`
	TopFrequencies []int         `json:"top_frequencies"`
}

type PerformanceMetrics struct {
	// QueriesExecuted is the total number of queries run during the benchmark.
	QueriesExecuted int64 `json:"queries_executed"`
	// Errors is the total number of errors that occurred during the benchmark.
	Errors int64 `json:"errors"`
	// Duration is the total time taken for the benchmark to complete.
	Duration time.Duration `json:"duration"`
	// Latencies is a slice of all the individual query latencies.
	Latencies []time.Duration `json:"-"` // Exclude from JSON report for brevity
	// P50 is the 50th percentile latency.
	P50 time.Duration `json:"p50"`
	// P90 is the 90th percentile latency.
	P90 time.Duration `json:"p90"`
	// P99 is the 99th percentile latency.
	P99 time.Duration `json:"p99"`
	// SlowQueries is the number of queries that exceeded the slow query threshold.
	SlowQueries int64 `json:"slow_queries"`
}

type QueryExecutionResult struct {
	Duration time.Duration
	Error    error
}

type QueryWithArgs struct {
	// Query is the SQL statement with '?' placeholders for parameters.
	Query string `json:"query"`
	// Args is a slice of arguments to be bound to the query's placeholders.
	Args []interface{} `json:"args"`
}

type Report struct {
	Version   string            `json:"version"`
	Timestamp time.Time         `json:"timestamp"`
	Metadata  *ReportMetadata   `json:"metadata"`
	Result    *ValidationResult `json:"result"`
}

type ReportMetadata struct {
	BaseTarget      string         `json:"base_target"`
	CandidateTarget string         `json:"candidate_target"`
	Threshold       float64        `json:"threshold"`
	TemplateSummary []*SQLTemplate `json:"template_summary"`
}

type SQLTemplate struct {
	RawSQL     string
	GroupKey   string
	Weight     int
	Parameters []string
}

type SQLTrace struct {
	Query      string
	Timestamp  time.Time
	Latency    time.Duration
	Parameters map[string]interface{}
}

type Schema struct {
	Databases []DatabaseSchema `json:"databases"`
}

type TableSchema struct {
	Name      string                  `json:"name"`
	Columns   []*ColumnSchema         `json:"columns"`
	PK        []string                `json:"pk"` // Primary Key columns
	Indexes   map[string]*IndexSchema `json:"indexes"`
	Engine    string                  `json:"engine,omitempty"` // e.g., "MergeTree() ORDER BY ..."
	CreateSQL string                  `json:"create_sql,omitempty"`
}

type TraceCollection struct {
	Traces []SQLTrace
}

type ValidationReport struct {
	Status         string
	QPSDeviation   float64
	LatencyP99Diff float64
}

type ValidationResult struct {
	BaseMetrics      *PerformanceMetrics `json:"base_metrics"`
	CandidateMetrics *PerformanceMetrics `json:"candidate_metrics"`
	Pass             bool                `json:"pass"`
	Reason           string              `json:"reason"`
}

type ValueDistribution struct {
	// Values is a slice of unique parameter values.
	Values []interface{}
	// Frequencies is a slice of corresponding frequencies for each value.
	Frequencies []int
	// Total is the total number of observations for this parameter.
	Total int
}

type WorkloadParameterModel struct {
	// TemplateParameters maps a template's GroupKey to a map of its parameters.
	// Each parameter is then mapped to its parameter model.
	TemplateParameters map[string]map[string]*ParameterModel
}