smart parse arguments for task`s driver method

Author: toplan

README.md

@@ -4,39 +4,32 @@ task load balancer for php (like the nginx load balancing)
 # Usage
 
 ```php
-//define tas
+//define task
 Balancer::task('task1', function($task){
-    //define driver
-    $task->driver('driver1')
-         ->weight(3)->backUp()
-         ->data(['this is data 1'])
-         ->work(function($driver, $data){
+    //define a driver
+    $task->driver('driver_1 100 backup', function($driver, $data){
+                    $person = new Person($data['name'], $data['age']);
                     $driver->failed();
-                    $msg = 'working! by '.$driver->name.'<br>';
-                    print_r($msg);
-                    return [$msg, $data];
+                    print_r('run work! by '.$driver->name.'<br>');
+                    return ['test.driver1 working', $person->toString()];
                 });
 
-    //define driver
-    $task->driver('driver2')
-         ->weight(3)->backUp(false)
-         ->data(['this is data 2'])
-         ->work(function($driver, $data){
-                    $driver->success();
-                    $msg = 'working! by '.$driver->name.'<br>';
-                    print_r($msg);
-                    return [$msg, $data];
-                });
+    //define a driver
+    $task->driver('driver_2', 100, function($driver, $data){
+             $driver->failed();
+             print_r('run work! by '.$driver->name.'<br>');
+             return ['test.driver2 working', $data];
+         })
+         ->data(['this is data 2']);
 
-    //define driver
-    $task->driver('driver3')
+    //define a driver
+    $task->driver('driver_3')
          ->weight(0)->backUp()
          ->data(['this is data 3'])
          ->work(function($driver, $data){
                     $driver->failed();
-                    $msg = 'working! by '.$driver->name.'<br>';
-                    print_r($msg);
-                    return [$msg, $data];
+                    print_r('run work! by '.$driver->name.'<br>');
+                    return ['test.driver3 working', $data];
                 });
 });
 
@@ -47,6 +40,6 @@ $result = Balancer::run('task1');
 
 - [x] remember every tasks` start time and end time.
 - [x] remember every drivers` start time and end time.
-- [ ] smart parse driver`s create arguments in task class
+- [x] smart parse driver`s create arguments in task class
 - [ ] define task`s lifecycle and hook
 - [ ] pause and continue task

src/TaskBalancer/Task.php

@@ -273,17 +273,22 @@ public function driverNameRand()
         return array_rand(array_keys($this->drivers));
     }
 
+
     /**
      * create a new driver instance for current task
-     * @param               $name
-     * @param int           $weight
-     * @param bool|false    $isBackup
-     * @param \Closure|null $work
-     *
-     * @return mixed
+     * @return null|static
+     * @throws \Exception
      */
-    public function driver($name, $weight = 1, $isBackup = false, \Closure $work = null)
+    public function driver()
     {
+        $args = func_get_args();
+        if (!count($args)) {
+            throw new \Exception('please give task`s method `driver` some args');
+        }
+        extract($this->parseDriverArgs($args));
+        if (!$name) {
+            throw new \Exception('please set driver`s name!');
+        }
         $driver = $this->getDriver($name);
         if (!$driver) {
             $driver = Driver::create($this, $name, $weight, $isBackup, $work);
@@ -296,7 +301,44 @@ public function driver($name, $weight = 1, $isBackup = false, \Closure $work = n
     }
 
     /**
-     * current task has driver?
+     * parse arguments for method `driver()`
+     * @param $args
+     *
+     * @return array
+     */
+    private function parseDriverArgs($args)
+    {
+        $result = [
+            'name' => '',
+            'work' => null,
+            'weight' => 1,
+            'isBackup' => false,
+        ];
+        foreach ($args as $arg) {
+            //find work
+            if (is_callable($arg)) {
+                $result['work'] = $arg;
+            }
+            //find weight, backup, name
+            if (is_string($arg) || is_numeric($arg)) {
+                $arg = preg_replace('/\s+/', ' ', "$arg");
+                $subArgs = explode(' ', trim($arg));
+                foreach ($subArgs as $subArg) {
+                    if (preg_match('/^[0-9]+$/', $subArg)) {
+                        $result['weight'] = $subArg;
+                    } elseif (preg_match('/(backup)/', strtolower($subArg))) {
+                        $result['isBackup'] = true;
+                    } else {
+                        $result['name'] = $subArg;
+                    }
+                }
+            }
+        }
+        return $result;
+    }
+
+    /**
+     * current task has character driver?
      * @param $name
      *
      * @return bool

test/index.php

@@ -13,25 +13,21 @@
     'age'  => '25!'
 ];
 $t = Balancer::task('test1', $data, function($task){
-    $task->driver('driver1')
-         ->weight(100)->backUp()
-         ->work(function($driver, $data){
+    $task->driver('driver_1 100 backup', function($driver, $data){
                     $person = new Person($data['name'], $data['age']);
                     $driver->failed();
                     print_r('run work! by '.$driver->name.'<br>');
                     return ['test.driver1 working', $person->toString()];
                 });
 
-    $task->driver('driver2')
-         ->weight(100)
-         ->data(['this is data 2'])
-         ->work(function($driver, $data){
-                    $driver->failed();
-                    print_r('run work! by '.$driver->name.'<br>');
-                    return ['test.driver2 working', $data];
-                });
+    $task->driver('driver_2', 90, function($driver, $data){
+             $driver->failed();
+             print_r('run work! by '.$driver->name.'<br>');
+             return ['test.driver2 working', $data];
+         })
+         ->data(['this is data 2']);
 
-    $task->driver('driver3')
+    $task->driver('driver_3 ')
          ->weight(0)->backUp()
          ->data(['this is data 3'])
          ->work(function($driver, $data){