start/stop tasks

3 files changed, 4 insertions, 8 deletions

diff --git a/app/relay/modules/test.js b/app/relay/modules/test.js
index 1d7042e..b0753f8 100644
--- a/app/relay/modules/test.js
+++ b/app/relay/modules/test.js
@@ -16,6 +16,7 @@ const gpu = {
 const live = {
 	type: 'python',
   script: 'test.py',
+  live: true,
 }
 
 export default {
diff --git a/app/relay/remote.js b/app/relay/remote.js
index c66929a..ff2d32b 100644
--- a/app/relay/remote.js
+++ b/app/relay/remote.js
@@ -5,7 +5,6 @@ const runner = require('./runner')
 
 export const remote = io.connect(process.env.SOCKETIO_REMOTE)
 remote.on('cmd', (data) => {
-  console.log('cmd data', data)
   if (! data.cmd) return console.log('malformed param...?')
   console.log('got', data.cmd)
   switch (data.cmd) {
@@ -60,13 +59,6 @@ remote.on('task', (data) => {
       break
     case 'set_priority':
       break
-    // case 'get_status':
-    //   remote.emit('task_res', {
-    //     type: 'relay_status',
-    //     rpc_connected: rpc_connected,
-    //     runner: runner.status(),
-    //   })
-    //   break
     default:
       response = { type: 'error', error: 'unknown task command' }
       break
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)
 	})
 }