bench

type CleanupCounters struct {
	FinalizersQueued   uint64 `json:"finalizers_queued"`
	FinalizersExecuted uint64 `json:"finalizers_executed"`
	CleanupsQueued     uint64 `json:"cleanups_queued_estimate"`
	CleanupsExecuted   uint64 `json:"cleanups_executed_estimate"`
	GCCycles           uint64 `json:"gc_cycles"`
}

type ClientCounters struct {
	Observation string           `json:"observation"`
	Values      map[string]int64 `json:"values,omitempty"`
}

type Config struct {
	ClusterFile    string `json:"cluster_file"`
	Workload       string `json:"workload"`
	Iterations     int    `json:"iterations"`
	Warmup         int    `json:"warmup"`
	Workers        int    `json:"workers"`
	Output         string `json:"output"`
	RunID          string `json:"run_id"`
	Block          int    `json:"block"`
	SourceRevision string `json:"source_revision"`
	HarnessID      string `json:"harness_id"`
	AllowWrites    bool   `json:"allow_writes"`
	Prefix         string `json:"prefix"`
	TimeoutMS      int64  `json:"transaction_timeout_ms"`
	RetryLimit     int64  `json:"retry_limit"`
	Profile        string `json:"profile"`
	// CoalesceDelay is the Go client's connection write delay, "min,max" as Go
	// durations; empty keeps the client default.
	CoalesceDelay string `json:"coalesce_delay,omitempty"`
	// AdaptiveLinger is the Go client's Nagle-style adaptive linger window; 0 keeps the default.
	AdaptiveLinger time.Duration `json:"adaptive_linger_ns,omitempty"`
	// Rate > 0 paces the run: operations start at Poisson arrival times averaging
	// Rate per second, drawn from Seed, so every run with the same seed starts the
	// same operations at the same offsets. Rate 0 is a closed loop.
	Rate float64 `json:"rate,omitempty"`
	Seed uint64  `json:"seed"`
	// Duration selects fixed-time closed-loop mode; zero preserves count mode.
	Duration       time.Duration `json:"duration_ns,omitempty"`
	WarmupDuration time.Duration `json:"warmup_duration_ns,omitempty"`
}

type Database interface {
	Transact(readOnly bool, fn func(Transaction) error) error
	Local(fn func(Transaction) error) error
	Counters() ClientCounters
	Close()
}

type Dataset struct {
	Points          map[int]KV
	Range           []KV
	RangeEnd        []byte
	Writes          [][]byte
	Initial         []byte
	Warmup, Measure [][]byte
	BatchWrites     [][][]byte
	BatchInitial    [][]byte
	BatchValueBytes int
}

type FinalizerDrain struct {
	Before   CleanupCounters `json:"before"`
	After    CleanupCounters `json:"after"`
	Rounds   int             `json:"rounds"`
	WallNS   int64           `json:"wall_ns"`
	Complete bool            `json:"complete"`
}

type KV struct{ Key, Value []byte }

type Lifecycle struct {
	MeasurementDrain   FinalizerDrain `json:"measurement_drain"`
	MeasurementBarrier bool           `json:"measurement_barrier_complete"`
	AfterDrain         Stats          `json:"stats_after_measurement_drain"`
	ShutdownDrain      FinalizerDrain `json:"shutdown_drain"`
	DatabaseClosed     bool           `json:"database_closed"`
	NetworkStopped     bool           `json:"network_stopped"`
	AfterShutdown      Stats          `json:"stats_after_shutdown"`
	Complete           bool           `json:"complete"`
}

type LifecycleDatabase interface {
	Database
	Barrier() error
	Shutdown() (networkStopped bool, err error)
}

type Pacing struct {
	LagP50NS, LagP99NS, LagMaxNS int64
	LateOver100us, LateOver1ms   float64
	GapCVActual, GapCVScheduled  float64
	Samples                      int
}

type Phase struct {
	Samples                    []Sample
	WallNS                     int64
	Completed, Errors, Retries int
	Before, After              Stats
	StatsError                 error
}

type Sample struct {
	Index, Worker            int
	StartOffsetNS, LatencyNS int64
	// ScheduledOffsetNS is the paced start time; -1 in a closed loop.
	ScheduledOffsetNS int64
	Attempts          int
	Err               error
	Attempted         bool
}

type Stats struct {
	TotalAlloc uint64 `json:"go_total_alloc_bytes"`
	Mallocs    uint64 `json:"go_mallocs"`
	GC         uint32 `json:"go_gc_cycles"`
	GCPause    uint64 `json:"go_gc_pause_ns"`
	UserUS     int64  `json:"process_user_cpu_us"`
	SystemUS   int64  `json:"process_system_cpu_us"`
	MaxRSSKB   int64  `json:"process_high_water_rss_kb"`
}

type Transaction interface {
	Get([]byte) ([]byte, error)
	Set([]byte, []byte)
	Range([]byte, []byte, int) ([]KV, error)
	ReadVersion() (int64, error)
}