bugfix

src/main.go

@@ -168,14 +168,14 @@ func serverAPI(w http.ResponseWriter, r *http.Request) {
 
 	case "debug_enable":
 		log.Debug("enable schedule")
-		js, _ := json.Marshal(scheduler.Enable)
+		js, _ := json.Marshal(scheduler.Enable())
 		w.Header().Set("Content-Type", "application/json")
 		w.Write(js)
 		break
 
 	case "debug_disable":
 		log.Debug("disable schedule")
-		js, _ := json.Marshal(scheduler.Disable)
+		js, _ := json.Marshal(scheduler.Disable())
 		w.Header().Set("Content-Type", "application/json")
 		w.Write(js)
 		break

src/scheduler.go

@@ -29,7 +29,7 @@ type Scheduler interface {
 
 	Detach(GPU string, job string)
 
-	Enable()
+	Enable() bool
 
-	Disable()
+	Disable() bool
 }

src/scheduler_FCFS.go

@@ -19,6 +19,7 @@ type SchedulerFCFS struct {
 func (scheduler *SchedulerFCFS) Start() {
 	scheduler.jobs = map[string]*JobManager{}
 	scheduler.history = []*Job{}
+	scheduler.enabled = true
 
 	go func() {
 		for {
@@ -229,10 +230,12 @@ func (scheduler *SchedulerFCFS) Detach(GPU string, job string) {
 	pool.detach(GPU, job)
 }
 
-func (scheduler *SchedulerFCFS) Enable() {
+func (scheduler *SchedulerFCFS) Enable() bool {
 	scheduler.enabled = true
+	return true
 }
 
-func (scheduler *SchedulerFCFS) Disable() {
+func (scheduler *SchedulerFCFS) Disable() bool {
 	scheduler.enabled = false
+	return true
 }

src/scheduler_fair.go

@@ -44,6 +44,7 @@ func (scheduler *SchedulerFair) Start() {
 	scheduler.queues = map[string][]Job{}
 	scheduler.queues["default"] = []Job{}
 	scheduler.resourceAllocations = map[string]*ResourceCount{}
+	scheduler.enabled = true
 
 	go func() {
 		for {
@@ -386,10 +387,12 @@ func (scheduler *SchedulerFair) Detach(GPU string, job string) {
 	pool.detach(GPU, job)
 }
 
-func (scheduler *SchedulerFair) Enable() {
+func (scheduler *SchedulerFair) Enable() bool {
 	scheduler.enabled = true
+	return true
 }
 
-func (scheduler *SchedulerFair) Disable() {
+func (scheduler *SchedulerFair) Disable() bool {
 	scheduler.enabled = false
+	return true
 }

src/scheduler_priority.go

@@ -19,6 +19,7 @@ type SchedulerPriority struct {
 func (scheduler *SchedulerPriority) Start() {
 	scheduler.jobs = map[string]*JobManager{}
 	scheduler.history = []*Job{}
+	scheduler.enabled = true
 
 	go func() {
 		for {
@@ -253,10 +254,12 @@ func (scheduler *SchedulerPriority) Detach(GPU string, job string) {
 	pool.detach(GPU, job)
 }
 
-func (scheduler *SchedulerPriority) Enable() {
+func (scheduler *SchedulerPriority) Enable() bool {
 	scheduler.enabled = true
+	return true
 }
 
-func (scheduler *SchedulerPriority) Disable() {
+func (scheduler *SchedulerPriority) Disable() bool {
 	scheduler.enabled = false
+	return true
 }