renamemain

1 files changed, 18 insertions, 8 deletions

diff --git a/internal/cpu/stresser.go b/internal/cpu/stresser.go
index 906467a..62ae3b7 100644
--- a/internal/cpu/stresser.go
+++ b/internal/cpu/stresser.go
@@ -13,12 +13,13 @@ import (
 // Stresser uses pointer receivers because it contains a sync.Mutex
 // and shared mutable state that must not be copied.
 type Stresser struct {
-	config  Config
-	cancel  context.CancelFunc
-	wg      sync.WaitGroup
-	mu      sync.Mutex
-	current Mode
-	rand    *rand.Rand
+	config      Config
+	cancel      context.CancelFunc
+	wg          sync.WaitGroup
+	mu          sync.Mutex
+	current     Mode
+	currentCPUs int
+	rand        *rand.Rand
 }
 
 func New(cfg Config) *Stresser {
@@ -34,6 +35,11 @@ func New(cfg Config) *Stresser {
 	if cfg.Mode == "" {
 		cfg.Mode = ModePrime
 	}
+	if !cfg.ChaosEnabled && cfg.MinCPUs <= 0 {
+		cfg.MinCPUs = cfg.Workers
+		cfg.MaxCPUs = cfg.Workers
+	}
+	cfg.normalize()
 	return &Stresser{
 		config: cfg,
 		rand:   rand.New(rand.NewSource(time.Now().UnixNano())),
@@ -44,8 +50,12 @@ func (s *Stresser) Start() {
 	ctx, cancel := context.WithCancel(context.Background())
 	s.cancel = cancel
 
-	s.wg.Add(s.config.Workers)
-	for i := 0; i < s.config.Workers; i++ {
+	s.mu.Lock()
+	s.currentCPUs = s.determineCPUCount()
+	s.mu.Unlock()
+
+	s.wg.Add(s.currentCPUs)
+	for i := 0; i < s.currentCPUs; i++ {
 		go s.worker(ctx, i)
 	}