coordinator

type Capacity struct {
	CPUs        int      `json:"cpus"`
	MemoryBytes uint64   `json:"memoryBytes"`
	MaxVUs      int      `json:"maxVUs"`
	Protocols   []string `json:"protocols"`
}

type Config struct {
	// JoinToken is the secret workers present in Hello. Workers are
	// refused while it is empty.
	JoinToken string
	// HeartbeatInterval is how often workers send heartbeats (default
	// 1s). Three missed heartbeats mark a worker lost.
	HeartbeatInterval time.Duration
	Logger            *slog.Logger

	// ClockSamples is the number of ping/pong round trips per clock
	// synchronisation (default 8).
	ClockSamples int
	// CA, when set, turns on mutual TLS: workers enroll for a certificate
	// with Enroll and must present it on Connect, where it replaces the
	// join token as their credential. The gRPC server must use
	// CA.ServerTLS.
	CA *pki.CA
	// CertValidity is the lifetime of enrolled worker certificates
	// (default DefaultCertValidity).
	CertValidity time.Duration

	// MergeGrace is how long after an interval ends the merged snapshot
	// waits for slow workers before it is emitted without them (default
	// two snapshot intervals).
	MergeGrace time.Duration

	// TakeoverLead is how far ahead a spare worker that takes over a lost
	// worker's share is told to start (default 2s): time to synchronise
	// its clock and prepare the scenario.
	TakeoverLead time.Duration
	// NoTakeover keeps lost shares unassigned: the run continues degraded.
	NoTakeover bool
}

type Coordinator struct {
	workerv1.UnimplementedWorkerServiceServer
	// contains filtered or unexported fields
}

type EventType string

type Result struct {
	RunID      string
	T0         time.Time
	End        time.Time
	Interval   time.Duration
	StopReason string
	// Phases are run-wide per-step phase histograms merged across workers.
	Phases map[int]*[metrics.NumPhases]*metrics.Histogram
	// PeakVUs sums the workers' peaks: exact for closed-model plans,
	// whose shares peak together.
	PeakVUs int
	// Snapshots are the merged intervals in order, including data that
	// arrived too late for the live stream.
	Snapshots []*metrics.Snapshot
	Workers   []WorkerSummary
	// Degraded is true when a worker was lost or failed mid-run.
	Degraded bool
	// LateSnapshots counts worker snapshots that missed the live stream.
	LateSnapshots uint64
}

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

type RunEvent struct {
	Time       time.Time `json:"time"`
	Type       EventType `json:"type"`
	Worker     string    `json:"worker,omitempty"`
	WorkerName string    `json:"workerName,omitempty"`
	Interval   int64     `json:"interval"`
	Message    string    `json:"message"`
	Reasons    []string  `json:"reasons,omitempty"`
	Workers    []string  `json:"workers,omitempty"`
}

type RunSpec struct {
	ID       string
	Scenario []byte // scenario YAML
	Env      map[string]string
	Secrets  map[string]string
	// AllowHosts lists the public hosts workers may send requests to,
	// normally the verified target host. Private hosts are always allowed.
	AllowHosts []string
	// Workers is how many workers to use (0 = every idle connected one).
	Workers int
	// Regions optionally splits load by worker region, for example
	// {"mumbai": 0.5, "frankfurt": 0.5}. Fractions are normalised.
	Regions map[string]float64
	// StartDelay is the lead time before the synchronised T0 (default 3s).
	StartDelay time.Duration
}

type Window struct {
	From int64 `json:"from"`
	To   int64 `json:"to"`
}

type WorkerInfo struct {
	ID             string            `json:"id"`
	Name           string            `json:"name"`
	Version        string            `json:"version"`
	Region         string            `json:"region,omitempty"`
	Labels         map[string]string `json:"labels,omitempty"`
	Capacity       Capacity          `json:"capacity"`
	Connected      bool              `json:"connected"`
	ConnectedSince time.Time         `json:"connectedSince"`
	LastHeartbeat  time.Time         `json:"lastHeartbeat"`
	Health         wire.Health       `json:"health"`
	// ClockOffset is the worker's clock minus the server's, from the
	// latest synchronisation.
	ClockOffset time.Duration `json:"clockOffset"`
	CurrentRun  string        `json:"currentRun,omitempty"`
}

type WorkerSummary struct {
	ID          string        `json:"id"`
	Name        string        `json:"name"`
	Region      string        `json:"region,omitempty"`
	Index       int           `json:"index"`
	ShareLo     float64       `json:"shareLo"`
	ShareHi     float64       `json:"shareHi"`
	ClockOffset time.Duration `json:"clockOffset"`
	ClockRTT    time.Duration `json:"clockRTT"`
	State       string        `json:"state"`
	StopReason  string        `json:"stopReason,omitempty"`
	Error       string        `json:"error,omitempty"`
	PeakVUs     int           `json:"peakVUs"`
	Requests    uint64        `json:"requests"`
	// Iterations counts iterations started.
	Iterations uint64 `json:"iterations"`
	Snapshots  int    `json:"snapshots"`
	// Saturated lists the intervals in which the worker reported itself
	// saturated, and SaturationReasons the distinct reasons.
	Saturated         []Window `json:"saturated,omitempty"`
	SaturationReasons []string `json:"saturationReasons,omitempty"`
	// Lost is the window from the loss until the share was taken over, or
	// to the end of the run.
	Lost *Window `json:"lost,omitempty"`
	// Replaces names the lost worker whose share this one took over, from
	// interval JoinedAt.
	Replaces string `json:"replaces,omitempty"`
	JoinedAt int64  `json:"joinedAt,omitempty"`
	// Missing lists intervals emitted live without this worker's data.
	Missing []int64 `json:"missing,omitempty"`
}