move mutex

Author: skudasov

wasp/wasp.go

@@ -513,20 +513,20 @@ func (g *Generator) processSegment() bool {
 			return false
 		}
 		if oldVUs > g.currentSegment.From {
+			g.vusMu.Lock()
 			for i := 0; i < vusToSpawn; i++ {
-				g.vusMu.Lock()
 				g.vus[i].Stop(g)
-				g.vusMu.Unlock()
 			}
 			g.vus = g.vus[vusToSpawn:]
+			g.vusMu.Unlock()
 		} else {
+			g.vusMu.Lock()
 			for i := 0; i < vusToSpawn; i++ {
 				inst := g.vu.Clone(g)
 				g.runVU(inst)
-				g.vusMu.Lock()
 				g.vus = append(g.vus, inst)
-				g.vusMu.Unlock()
 			}
+			g.vusMu.Unlock()
 		}
 	}
 	return false
@@ -632,9 +632,9 @@ func (g *Generator) pacedCall() {
 	if !g.Stats().RunStarted.Load() {
 		return
 	}
-	//if g.rl.Load() == nil {
-	//	return
-	//}
+	if g.rl.Load() == nil {
+		return
+	}
 	(*g.rl.Load()).Take()
 	if g.stats.RunPaused.Load() {
 		return