X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/419db47f0e97123cb3ff491d189b5607468101da..49bb6cb300e9f6a3862c4e17112e34da1226a3c7:/services/keep-balance/balance.go

diff --git a/services/keep-balance/balance.go b/services/keep-balance/balance.go
index 7b25d78852..d6a2dde9d7 100644
--- a/services/keep-balance/balance.go
+++ b/services/keep-balance/balance.go
@@ -365,27 +365,29 @@ func (bal *Balancer) ComputeChangeSets() {
 		blkid arvados.SizedDigest
 		blk   *BlockState
 	}
-	nWorkers := 1 + runtime.NumCPU()
-	todo := make(chan balanceTask, nWorkers)
-	results := make(chan balanceResult, 16)
-	var wg sync.WaitGroup
-	for i := 0; i < nWorkers; i++ {
-		wg.Add(1)
-		go func() {
-			for work := range todo {
-				results <- bal.balanceBlock(work.blkid, work.blk)
+	workers := runtime.GOMAXPROCS(-1)
+	todo := make(chan balanceTask, workers)
+	go func() {
+		bal.BlockStateMap.Apply(func(blkid arvados.SizedDigest, blk *BlockState) {
+			todo <- balanceTask{
+				blkid: blkid,
+				blk:   blk,
 			}
-			wg.Done()
-		}()
-	}
-	bal.BlockStateMap.Apply(func(blkid arvados.SizedDigest, blk *BlockState) {
-		todo <- balanceTask{
-			blkid: blkid,
-			blk:   blk,
-		}
-	})
-	close(todo)
+		})
+		close(todo)
+	}()
+	results := make(chan balanceResult, workers)
 	go func() {
+		var wg sync.WaitGroup
+		for i := 0; i < workers; i++ {
+			wg.Add(1)
+			go func() {
+				for work := range todo {
+					results <- bal.balanceBlock(work.blkid, work.blk)
+				}
+				wg.Done()
+			}()
+		}
 		wg.Wait()
 		close(results)
 	}()