1 files changed, 5 insertions, 2 deletions
diff --git a/app/client/socket/socket.task.js b/app/client/socket/socket.task.js
index 0c8429b..ba074c0 100644
--- a/app/client/socket/socket.task.js
+++ b/app/client/socket/socket.task.js
@@ -3,6 +3,8 @@ import types from '../types'
 
 import { socket } from './socket.connection'
 
+let finishTimeout;
+
 socket.on('task_res', (data) => {
   console.log('system response', data)
   switch (data.type) {
@@ -10,12 +12,13 @@ socket.on('task_res', (data) => {
       // return dispatch({ type: types.system.rpc_connected, runner: data.runner })
       break
     case 'task_begin':
-      return dispatch({ type: types.task.task_begin, data: data.data })
+      clearTimeout(finishTimeout)
+      return dispatch({ type: types.task.task_begin, task: data.task })
       break
     case 'stop':
       break
     case 'task_finish':
-      return dispatch({ type: types.task.task_finish, data: data.data })
+      return finishTimeout = setTimeout(() => dispatch({ type: types.task.task_finish, task: data.task }), 100)
       break
     case 'kill':
       break