Add Workflow::upsertMemo() implementation

src/Interceptor/WorkflowOutboundCallsInterceptor.php

@@ -26,6 +26,7 @@
 use Temporal\Interceptor\WorkflowOutboundCalls\SideEffectInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\SignalExternalWorkflowInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\TimerInput;
+use Temporal\Interceptor\WorkflowOutboundCalls\UpsertMemoInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\UpsertSearchAttributesInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\UpsertTypedSearchAttributesInput;
 use Temporal\Internal\Interceptor\Interceptor;
@@ -113,6 +114,11 @@ public function continueAsNew(ContinueAsNewInput $input, callable $next): Promis
      */
     public function getVersion(GetVersionInput $input, callable $next): PromiseInterface;
 
+    /**
+     * @param callable(UpsertMemoInput): PromiseInterface $next
+     */
+    public function upsertMemo(UpsertMemoInput $input, callable $next): PromiseInterface;
+
     /**
      * @param callable(UpsertSearchAttributesInput): PromiseInterface $next
      */

src/Internal/Workflow/WorkflowContext.php

@@ -35,6 +35,7 @@
 use Temporal\Interceptor\WorkflowOutboundCalls\PanicInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\SideEffectInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\TimerInput;
+use Temporal\Interceptor\WorkflowOutboundCalls\UpsertMemoInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\UpsertSearchAttributesInput;
 use Temporal\Interceptor\WorkflowOutboundCalls\UpsertTypedSearchAttributesInput;
 use Temporal\Interceptor\WorkflowOutboundCallsInterceptor;
@@ -55,6 +56,7 @@
 use Temporal\Internal\Transport\Request\NewTimer;
 use Temporal\Internal\Transport\Request\Panic;
 use Temporal\Internal\Transport\Request\SideEffect;
+use Temporal\Internal\Transport\Request\UpsertMemo;
 use Temporal\Internal\Transport\Request\UpsertSearchAttributes;
 use Temporal\Internal\Transport\Request\UpsertTypedSearchAttributes;
 use Temporal\Internal\Workflow\Process\HandlerState;
@@ -448,6 +450,31 @@ public function allHandlersFinished(): bool
         return !$this->handlers->hasRunningHandlers();
     }
 
+    public function upsertMemo(array $values): void
+    {
+        $this->callsInterceptor->with(
+            function (UpsertMemoInput $input): PromiseInterface {
+                $result = $this->request(new UpsertMemo($input->memo), false);
+
+                /** @psalm-suppress UnsupportedPropertyReferenceUsage $memo */
+                $memo = &$this->input->info->memo;
+                $memo ??= [];
+                foreach ($input->memo as $name => $value) {
+                    if ($value === null) {
+                        unset($memo[$name]);
+                        continue;
+                    }
+
+                    $memo[$name] = $value;
+                }
+
+                return $result;
+            },
+            /** @see WorkflowOutboundCallsInterceptor::upsertMemo() */
+            'upsertMemo',
+        )(new UpsertMemoInput($values));
+    }
+
     public function upsertSearchAttributes(array $searchAttributes): void
     {
         $this->callsInterceptor->with(

src/Workflow.php

@@ -903,6 +903,16 @@ public static function allHandlersFinished(): bool
         return $context->allHandlersFinished();
     }
 
+    /**
+     * Upsert Memo
+     *
+     * @param array<string, mixed> $values
+     */
+    public static function upsertMemo(array $values): void
+    {
+        self::getCurrentContext()->upsertMemo($values);
+    }
+
     /**
      * Upsert search attributes
      *

src/Workflow/WorkflowContextInterface.php

@@ -293,6 +293,43 @@ public function getStackTrace(): string;
      */
     public function allHandlersFinished(): bool;
 
+    /**
+     * Updates this Workflow's Memos by merging the provided memo with existing Memos.
+     *
+     * New Memo is merged by replacing properties of the same name at the first level only.
+     * Setting a property to {@see null} clears that key from the Memo.
+     *
+     * For example:
+     *
+     * ```php
+     *  Workflow::upsertMemo([
+     *      'key1' => 'value',
+     *      'key3' => ['subkey1' => 'value']
+     *      'key4' => 'value',
+     *  });
+     *
+     *  Workflow::upsertMemo([
+     *      'key2' => 'value',
+     *      'key3' => ['subkey2' => 'value']
+     *      'key4' => null,
+     *  ]);
+     * ```
+     *
+     * would result in the Workflow having these Memo:
+     *
+     * ```php
+     *  [
+     *      'key1' => 'value',
+     *      'key2' => 'value',
+     *      'key3' => ['subkey2' => 'value'], // Note this object was completely replaced
+     *      // Note that 'key4' was completely removed
+     *  ]
+     * ```
+     *
+     * @param array<non-empty-string, mixed> $values
+     */
+    public function upsertMemo(array $values): void;
+
     /**
      * Upsert search attributes
      *