Always lock when getting workers (#245)

beam-cloud · May 22, 2024 · 48c2ac5 · 48c2ac5
1 parent a22bcc9
commit 48c2ac5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/pkg/repository/worker_redis.go b/pkg/repository/worker_redis.go
@@ -225,7 +225,7 @@ func (r *WorkerRedisRepository) GetAllWorkers() ([]*types.Worker, error) {
 }
 
 func (r *WorkerRedisRepository) GetAllWorkersInPool(poolName string) ([]*types.Worker, error) {
-	workers, err := r.getWorkers(false)
+	workers, err := r.getWorkers(true)
 	if err != nil {
 		return nil, err
 	}
@@ -241,7 +241,7 @@ func (r *WorkerRedisRepository) GetAllWorkersInPool(poolName string) ([]*types.W
 }
 
 func (r *WorkerRedisRepository) GetAllWorkersOnMachine(machineId string) ([]*types.Worker, error) {
-	workers, err := r.getWorkers(false)
+	workers, err := r.getWorkers(true)
 	if err != nil {
 		return nil, err
 	}

diff --git a/pkg/scheduler/pool_sizing.go b/pkg/scheduler/pool_sizing.go
@@ -1,17 +1,15 @@
 package scheduler
 
 import (
-	"context"
 	"log"
 	"time"
 
 	"github.com/beam-cloud/beta9/pkg/types"
 )
 
-const poolMonitoringInterval = 1 * time.Second
+const poolMonitoringInterval = 5 * time.Second
 
 type WorkerPoolSizer struct {
-	ctx        context.Context
 	controller WorkerPoolController
 	config     *types.WorkerPoolSizingConfig
 }