renamemain

1 files changed, 10 insertions, 0 deletions

diff --git a/internal/cpu/helpers.go b/internal/cpu/helpers.go
index 9fcc6b9..138cf8c 100644
--- a/internal/cpu/helpers.go
+++ b/internal/cpu/helpers.go
@@ -11,6 +11,16 @@ func (s *Stresser) randomIntensity() int {
 	return s.rand.Intn(s.config.IntensityMax-s.config.IntensityMin+1) + s.config.IntensityMin
 }
 
+func (s *Stresser) determineCPUCount() int {
+	if s.config.ChaosEnabled {
+		return s.rand.Intn(s.config.MaxCPUs-s.config.MinCPUs+1) + s.config.MinCPUs
+	}
+	if s.config.Workers > 0 {
+		return s.config.Workers
+	}
+	return s.config.MaxCPUs
+}
+
 func (s *Stresser) getRandomMode() Mode {
 	return allModes[s.rand.Intn(len(allModes))]
 }