simulator

type Simulator struct {
	// Duration of the simulation
	Duration time.Duration
	// MaxGeneratedItems is the maximum number of items to generate.
	// If set, the simulation will run until the number of generated items is reached instead of the duration.
	MaxGeneratedItems int

	Stages []*Stage
	Mu     sync.RWMutex
	Ctx    context.Context
	Cancel context.CancelFunc
	Quit   chan struct{}
	Wg     sync.WaitGroup
}

type Stage struct {
	Name string

	Input  chan any
	Output chan any
	Sem    chan struct{}

	Config  *StageConfig
	Metrics *StageMetrics
	IdleSpy *tracker.GoroutineManager

	IsFinal           bool
	MaxGeneratedItems int
	Stop              func()
	// contains filtered or unexported fields
}

type StageConfig struct {
	// Rate at which items are generated (generator only)
	InputRate time.Duration
	// Custom item generator function
	ItemGenerator func() any

	// Handles load spikes and burst patterns
	// Generates input bursts at intervals
	InputBurst func() []any
	// Total number of bursts to inject
	BurstCountTotal int
	// Interval between bursts
	BurstInterval time.Duration

	// Number of goroutines per stage
	RoutineNum int
	// Channel buffer size per stage
	BufferSize int
	// Simulated delay per item
	WorkerDelay time.Duration
	// Number of times to retry on error
	RetryCount int

	// Drop input if channel is full
	DropOnBackpressure bool
	// Whether the stage is a generator
	IsGenerator bool

	// Core processing function
	// Worker function that processes each item
	WorkerFunc func(item any) (any, error)
	// Context for cancellation and deadlines
	Ctx context.Context
}

type StageMetrics struct {

	// Counters
	ProcessedItems uint64
	DroppedItems   uint64
	OutputItems    uint64

	// State
	StartTime time.Time
	EndTime   time.Time

	// Generator stats
	GeneratedItems uint64
	// contains filtered or unexported fields
}

type StageStats struct {
	StageName      string  `json:"stage_name"`
	ProcessedItems uint64  `json:"processed_items"`
	OutputItems    uint64  `json:"output_items"`
	Throughput     float64 `json:"throughput"`
	DroppedItems   uint64  `json:"dropped_items"`
	DropRate       float64 `json:"drop_rate"`
	GeneratedItems uint64  `json:"generated_items,omitempty"`
	ThruDiffPct    float64 `json:"-"`
	ProcDiffPct    float64 `json:"-"`
}