utility function

Author: henderkes

scaling.go

@@ -130,6 +130,19 @@ func scaleRegularThread() {
 	autoScaledThreads = append(autoScaledThreads, thread)
 }
 
+func getWorker(fc *frankenPHPContext) *worker {
+	// if the request has been stalled long enough, scale
+	if workersList, ok := workers[fc.scriptFilename]; ok {
+		// Look for a worker with matching moduleID or a global worker (moduleID == 0)
+		for _, worker := range workersList {
+			if worker.moduleID == 0 || worker.moduleID == fc.moduleID {
+				return worker
+			}
+		}
+	}
+	return nil
+}
+
 func startUpscalingThreads(maxScaledThreads int, scale chan *frankenPHPContext, done chan struct{}) {
 	for {
 		scalingMu.Lock()
@@ -160,19 +173,8 @@ func startUpscalingThreads(maxScaledThreads int, scale chan *frankenPHPContext,
 			}
 
 			// if the request has been stalled long enough, scale
-			if workersList, ok := workers[fc.scriptFilename]; ok {
-				// Look for a worker with matching moduleID or a global worker (moduleID == 0)
-				workerFound := false
-				for _, worker := range workersList {
-					if worker.moduleID == 0 || worker.moduleID == fc.moduleID {
-						scaleWorkerThread(worker)
-						workerFound = true
-						break
-					}
-				}
-				if !workerFound {
-					scaleRegularThread()
-				}
+			if worker := getWorker(fc); worker != nil {
+				scaleWorkerThread(worker)
 			} else {
 				scaleRegularThread()
 			}