accept an entity id

bottledcode · Jan 14, 2024 · 3ea0f2b · 3ea0f2b
1 parent 272a93e
commit 3ea0f2b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/OrchestrationContext.php b/src/OrchestrationContext.php
@@ -466,7 +466,7 @@ public static function reset(){
         };
     }
 
-    public function entityOp(string $id, \Closure $operation): mixed
+    public function entityOp(string|EntityId $id, \Closure $operation): mixed
     {
         $func = new \ReflectionFunction($operation);
         if($func->getNumberOfParameters() !== 1) {
@@ -498,7 +498,7 @@ public function entityOp(string $id, \Closure $operation): mixed
             throw new LogicException('Did not call an operation');
         }
 
-        $entityId = new EntityId($name, $id);
+        $entityId = $id instanceof EntityId ? $id : new EntityId($name, $id);
 
         if($returns) {
             return $this->waitOne($this->callEntity($entityId, $operationName, $arguments));

diff --git a/src/OrchestrationContextInterface.php b/src/OrchestrationContextInterface.php
@@ -70,11 +70,11 @@ public function signalEntity(EntityId $entityId, string $operation, array $args
     /**
      * @template T
      * @template V
-     * @param string $id
+     * @param string|EntityId $id
      * @param \Closure(T): V $operation
      * @return V
      */
-    public function entityOp(string $id, \Closure $operation): mixed;
+    public function entityOp(string|EntityId $id, \Closure $operation): mixed;
 
     /**
      * Determines if an entity is locked. Returns true if the entity is locked.