Check slots for direct workers

giongto35 · Nov 17, 2024 · f09500f · f09500f
1 parent 1147aed
commit f09500f
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/pkg/coordinator/hub.go b/pkg/coordinator/hub.go
@@ -1,7 +1,6 @@
 package coordinator
 
 import (
-	"bytes"
 	"encoding/base64"
 	"fmt"
 	"net/http"
@@ -299,16 +298,20 @@ func (h *Hub) findWorkerById(id string, useAllWorkers bool) *Worker {
 		if w.Id() == com.NilUid {
 			continue
 		}
-		if useAllWorkers {
-			if uid == w.Id() {
-				return w
-			}
-		} else {
-			// select any worker on the same machine when workers are grouped on the client
-			if bytes.Equal(uid.Machine(), w.Id().Machine()) {
-				return w
-			}
+
+		if uid == w.Id() && w.HasSlot() {
+			return w
 		}
+		//if useAllWorkers {
+		//	if uid == w.Id() {
+		//		return w
+		//	}
+		//} else {
+		//	// select any worker on the same machine when workers are grouped on the client
+		//	if bytes.Equal(uid.Machine(), w.Id().Machine()) {
+		//		return w
+		//	}
+		//}
 	}
 
 	return nil