Allow only available workers

This commit is contained in:
Sergey Stepanov 2022-04-08 18:39:24 +03:00
parent 9ad3c98a7d
commit 76c66339aa
No known key found for this signature in database
GPG key ID: A56B4929BAA8556B

View file

@ -189,7 +189,7 @@ func (s *Server) WS(w http.ResponseWriter, r *http.Request) {
if workerId != "" {
if xid_, err := xid.FromString(workerId); err == nil {
if s.cfg.Coordinator.Debug {
for _, w := range s.workerClients {
for _, w := range s.getAvailableWorkers() {
if xid_ == w.Id {
wc = w
bc.Printf("[!] Worker found: %v", xid_)
@ -197,7 +197,7 @@ func (s *Server) WS(w http.ResponseWriter, r *http.Request) {
}
}
} else {
for _, w := range s.workerClients {
for _, w := range s.getAvailableWorkers() {
if bytes.Equal(xid_.Machine(), w.Id.Machine()) {
wc = w
bc.Printf("[!] Machine %v found: %v", xid_.Machine(), xid_)