override driver data

Author: toplan

src/TaskBalancer/Balancer.php

@@ -21,16 +21,21 @@ class Balancer {
 
     /**
      * create a task instance
-     * @param               $name
-     * @param \Closure|null $fn
+     * @param      $name
+     * @param      $data
+     * @param \Closure|null $callback
      *
-     * @return Task
+     * @return null|Task
      */
-    public static function task($name, \Closure $fn = null)
+    public static function task($name, $data, \Closure $callback = null)
     {
         $task = self::getTask($name);
         if (!$task) {
-            $task = Task::create($name, $fn);
+            if (is_callable($data)) {
+                $callback = $data;
+                $data = null;
+            }
+            $task = Task::create($name, $data, $callback);
             self::$tasks[$name] = $task;
         }
         return $task;
@@ -39,16 +44,20 @@ public static function task($name, \Closure $fn = null)
     /**
      * run a task instance
      * @param string $name
+     * @param string $data
      *
      * @return mixed
      * @throws \Exception
      */
-    public static function run($name = '')
+    public static function run($name = '', $data = null)
     {
         $task = self::getTask($name);
         if (!$task) {
             throw new \Exception("run task $name failed, not find this task");
         }
+        if ($data) {
+            $task->data($data);
+        }
         $task->run();
         return $task->results;
     }

src/TaskBalancer/Driver.php

@@ -87,7 +87,7 @@ public function __construct(Task $task, $name, $weight = 1, $isBackUp = false, \
      */
     public static function create(Task $task, $name, $weight = 1, $isBackUp = false, \Closure $work = null)
     {
-        $driver = new static($task, $name, $weight, $isBackUp, $work);
+        $driver = new self($task, $name, $weight, $isBackUp, $work);
         return $driver;
     }
 
@@ -113,7 +113,7 @@ public function run()
         }
         $result = null;
         if (is_callable($this->work)) {
-            $result = call_user_func_array($this->work, [$this, $this->data]);
+            $result = call_user_func_array($this->work, [$this, $this->getData()]);
         }
         return $this->afterRun($result);
     }
@@ -198,8 +198,45 @@ public function work(\Closure $work)
         return $this;
     }
 
+    /**
+     * get data
+     * @return null
+     */
+    public function getData()
+    {
+        return $this->getDriverData() ?:
+               $this->getTaskData();
+    }
+
+    /**
+     * get driver data
+     * @return null
+     */
+    public function getDriverData()
+    {
+        return $this->data;
+    }
+
+    /**
+     * get task data
+     * @return null
+     */
+    public function getTaskData()
+    {
+        return $this->task->data;
+    }
+
+    /**
+     * override
+     * @param $name
+     *
+     * @return null
+     */
     public function __get($name)
     {
+        if ($name == "data") {
+            return $this->getData();
+        }
         if (isset($this->$name)) {
             return $this->$name;
         }

src/TaskBalancer/Task.php

@@ -61,6 +61,12 @@ class Task {
         'finished_at' => 0
     ];
 
+    /**
+     * data for driver
+     * @var null
+     */
+    protected $data = null;
+
     /**
      * drivers` results
      * @var array
@@ -70,23 +76,26 @@ class Task {
     /**
      * constructor
      * @param               $name
+     * @param               $data
      * @param \Closure|null $work
      */
-    public function __construct($name, \Closure $work = null)
+    public function __construct($name, $data = null, \Closure $work = null)
     {
         $this->name = $name;
+        $this->data = $data;
         $this->work = $work;
     }
 
     /**
      * create a new task
      * @param               $name
+     * @param               $data
      * @param \Closure|null $work
      * @return Task
      */
-    public static function create($name, \Closure $work = null)
+    public static function create($name, $data = null, \Closure $work = null)
     {
-        $task = new static($name, $work);
+        $task = new self($name, $data, $work);
         $task->runWork();
         return $task;
     }
@@ -131,9 +140,12 @@ public function run($driverName = '')
      */
     private function beforeRun()
     {
+        $pass = true;
         $this->time['started_at'] = microtime();
-        $this->status = static::RUNNING;
-        return true;
+        if ($pass) {
+            $this->status = static::RUNNING;
+        }
+        return $pass;
     }
 
     /**
@@ -377,6 +389,18 @@ public function removeFromBackupDrivers($driverName)
         }
     }
 
+    /**
+     * set data
+     * @param $data
+     *
+     * @return $this
+     */
+    public function data($data)
+    {
+        $this->data = $data;
+        return $this;
+    }
+
     /**
      * override
      * @param $name

test/index.php

@@ -8,15 +8,20 @@
 use Toplan\TaskBalance\Driver;
 
 //define task:
-$t = Balancer::task('test1', function($task){
+$data = [
+    'name' => 'top$',
+    'age'  => '25!'
+];
+$t = Balancer::task('test1', $data, function($task){
     $task->driver('driver1')
          ->weight(100)->backUp()
-         ->data(['this is data 1'])
          ->work(function($driver, $data){
+                    $person = new Person($data['name'], $data['age']);
                     $driver->failed();
                     print_r('run work! by '.$driver->name.'<br>');
-                    return ['test.driver1 working', $data];
+                    return ['test.driver1 working', $person->toString()];
                 });
+
     $task->driver('driver2')
          ->weight(100)
          ->data(['this is data 2'])
@@ -25,6 +30,7 @@
                     print_r('run work! by '.$driver->name.'<br>');
                     return ['test.driver2 working', $data];
                 });
+
     $task->driver('driver3')
          ->weight(0)->backUp()
          ->data(['this is data 3'])
@@ -36,9 +42,28 @@
 });
 
 //run task:
-$result = Balancer::run('test1');
+$data['age'] = '25###';
+$result = Balancer::run('test1', $data);
 
 print_r('<br>resuts data:<br>');
 var_dump($result);
 print_r('<hr>task data:<br>');
 var_dump($t);
+
+class Person {
+
+    protected  $name;
+
+    protected  $age;
+
+    public function __construct($name, $age)
+    {
+        $this->name = $name;
+        $this->age = $age;
+    }
+
+    public function toString()
+    {
+        return "hi, I am $this->name, and $this->age year old";
+    }
+}