From 45793534911dce31d753206dd22810dccfeeaa61 Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Tue, 5 May 2020 19:12:11 +0800
Subject: [PATCH] update

---
 src/scheduler_fair.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/scheduler_fair.go b/src/scheduler_fair.go
index 90aad2a..ff8a8b0 100644
--- a/src/scheduler_fair.go
+++ b/src/scheduler_fair.go
@@ -48,6 +48,8 @@ type SchedulerFair struct {
 	queuesSchedulingCnt map[string]int
 	queueUsingGPU       map[string]int
 	queuesUsingGPUMu    sync.Mutex
+
+	mu sync.Mutex
 }
 
 type FairJobSorter []Job
@@ -342,6 +344,8 @@ func (scheduler *SchedulerFair) Schedule(job Job) {
 }
 
 func (scheduler *SchedulerFair) AcquireResource(job Job, task Task, nodes []NodeStatus) NodeStatus {
+	scheduler.mu.Lock()
+	defer scheduler.mu.Unlock()
 	segID := rand.Intn(pool.poolsCount)
 	res := NodeStatus{}
 	start := &pool.pools[segID]