remember drivers`s run time

Author: toplan

src/TaskBalancer/Driver.php

@@ -49,6 +49,15 @@ class Driver
      */
     protected $data = null;
 
+    /**
+     * run work time
+     * @var array
+     */
+    protected $time = [
+        'started_at' => 0,
+        'finished_at' => 0,
+    ];
+
     /**
      * constructor
      * @param            $task
@@ -82,16 +91,42 @@ public static function create(Task $task, $name, $weight = 1, $isBackUp = false,
         return $driver;
     }
 
+    /**
+     * before run hook
+     * @return bool
+     */
+    public function beforeRun()
+    {
+        $this->time['started_at'] = microtime();
+        return true;
+    }
+
     /**
      * run driver`s work
      * @return mixed|null
      */
     public function run()
     {
+        $this->beforeRun();
+        if (!$this->beforeRun()) {
+            return null;
+        }
         $result = null;
-        if ($this->work) {
+        if (is_callable($this->work)) {
             $result = call_user_func_array($this->work, [$this, $this->data]);
         }
+        return $this->afterRun($result);
+    }
+
+    /**
+     * after run hook
+     * @param $result
+     *
+     * @return mixed
+     */
+    public function afterRun($result)
+    {
+        $this->time['finished_at'] = microtime();
         return $result;
     }
 

src/TaskBalancer/Task.php

@@ -87,7 +87,7 @@ public static function create($name, \Closure $work = null)
      */
     public function runWork()
     {
-        if ($this->work) {
+        if (is_callable($this->work)) {
             call_user_func($this->work, $this);
         }
     }
@@ -119,6 +119,7 @@ public function runDriver($name)
         $success = $driver->success;
         $data = [
             'driver' => $driver->name,
+            'time' => $driver->time,
             'success' => $success,
             'result' => $result,
         ];

test/index.php

@@ -18,7 +18,7 @@
                     return ['test.driver1 working', $data];
                 });
     $task->driver('driver2')
-         ->weight(80)
+         ->weight(100)
          ->data(['this is data 2'])
          ->work(function($driver, $data){
                     $driver->failed();