Merge pull request #13 from citrus-framework/add_map_method

mapに指定メソッドの実行処理を追加する
citrus-framework · Aug 27, 2020 · e08877b · e08877b
2 parents 83bc06e + 6ca2777
commit e08877b
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 0 deletions.
diff --git a/src/Collection.php b/src/Collection.php
@@ -74,6 +74,27 @@ public function map(callable $callable): self
 
 
 
+    /**
+     * 指定メソッドを実行した適用した内容を積んで返却する
+     *
+     * @param string $method_name
+     * @return $this
+     */
+    public function mapExecMethod(string $method_name): self
+    {
+        $this->source = Scanner::map($this->source, function ($vl) use ($method_name) {
+            // メソッドが存在する場合に実行する
+            if (true === method_exists($vl, $method_name))
+            {
+                $vl->$method_name();
+            }
+            return $vl;
+        });
+        return $this;
+    }
+
+
+
     /**
      * callable関数を適用した内容を積んで返却する
      * keyを指定する

diff --git a/tests/CollectionTest.php b/tests/CollectionTest.php
@@ -234,6 +234,31 @@ public function map_データ編集して配列生成()
 
 
 
+    /**
+     * @test
+     */
+    public function mapExecMethod_メソッド実行して配列生成()
+    {
+        $values = [
+            new CollectionObj(),
+            new CollectionObj(),
+            new CollectionObj(),
+            new CollectionObj(),
+        ];
+
+        // 全部1を足す
+        $expected = [
+            (new CollectionObj())->add(),
+            (new CollectionObj())->add(),
+            (new CollectionObj())->add(),
+            (new CollectionObj())->add(),
+        ];
+        // 検算
+        $this->assertEquals($expected, Collection::stream($values)->mapExecMethod('add')->toList());
+    }
+
+
+
     /**
      * @test
      */
@@ -434,3 +459,13 @@ public function one_一件だけ取得()
         $this->assertSame($expected, Collection::stream($values)->one());
     }
 }
+
+class CollectionObj
+{
+    public $count = 0;
+    public function add()
+    {
+        $this->count++;
+        return $this;
+    }
+}