add ifPresent consumer for Option (#98)

munusphp · May 18, 2024 · 134f1bc · 134f1bc
1 parent 07409b7
commit 134f1bc
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/Control/Option.php b/src/Control/Option.php
@@ -70,6 +70,16 @@ public function isPresent(): bool
         return !$this->isEmpty();
     }
 
+    /**
+     * @param callable(T): void $consumer
+     */
+    public function ifPresent(callable $consumer): void
+    {
+        if ($this->isPresent()) {
+            $consumer($this->get());
+        }
+    }
+
     /**
      * @template U
      *

diff --git a/tests/Control/OptionTest.php b/tests/Control/OptionTest.php
@@ -182,4 +182,10 @@ public function testOptionIsPresent(): void
         self::assertTrue(Option::some(null)->isPresent());
         self::assertFalse(Option::none()->isPresent());
     }
+
+    public function testOptionIfPresent(): void
+    {
+        Option::none()->ifPresent(fn ($v) => throw new \RuntimeException('impossible is nothing'));
+        Option::of('a')->ifPresent(fn ($v) => self::assertSame('a', $v));
+    }
 }