start/stop tasks

1 files changed, 3 insertions, 0 deletions

diff --git a/app/relay/runner.js b/app/relay/runner.js
index ff32e81..906ddbf 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -3,6 +3,7 @@ import interpreters from './interpreters'
 import modules from './modules'
 import kill from 'tree-kill'
 import { remote } from './remote'
+import { set_connected } from './rpc'
 import uuidv1 from 'uuid/v1'
 
 const idle_state = { status: 'IDLE', task: {} }
@@ -171,6 +172,7 @@ export function run_task(task, preempt, watch){
 		console.log('task error', subprocess.exitCode, err)
     clear_task(interpreter.gpu, task)
     remote.emit('task_res', { type: 'task_error', task, err })
+    set_connected(false)
 	})
 
 	subprocess.on('close', () => {
@@ -179,6 +181,7 @@ export function run_task(task, preempt, watch){
 		console.log('task ended', subprocess.exitCode || '')
     clear_task(interpreter.gpu, task)
     remote.emit('task_res', { type: 'task_finish', task })
+    set_connected(false)
 	})
 }