fix(update): fixes <Update>Async for those that exceed default grpc poll timeout

Author: cludden

README.md

@@ -15,6 +15,7 @@ A protoc plugin for generating typed Temporal clients and workers in Go from pro
     - [Service Options](#service-options)
     - [Method Options](#method-options)
     - [Bloblang Expressions](#bloblang-expressions)
+  - [Workflow Update](#workflow-update)
   - [CLI](#cli)
   - [Test Client](#test-client)
   - [Cross-Namespace (XNS)](#cross-namespace-xns)
@@ -608,6 +609,13 @@ run, _ := example.ExecuteSayGreeting(context.Background(), &examplev1.SayGreetin
 require.Regexp(`^say-greeting/Howdy/Stranger/[a-f0-9-]{32}$`, run.ID())
 ```
 
+## Workflow Update
+
+This plugin has experimental support for Temporal's experimental [Workflow Update](https://docs.temporal.io/workflows#update) capability. Note that this requires cluster support enabled with both of the following dynamic config values set to `true`
+
+- `frontend.enableUpdateWorkflowExecution`
+- `frontend.enableUpdateWorkflowExecutionAsyncAccepted`
+  
 ## CLI
 
 This plugin can optionally generate a configurable CLI using [github.com/urfave/cli/v2](https://github.com/urfave/cli/v2). To enable this functionality, use the corresponding [plugin option](#plugin-options). When enabled, this plugin will generate a CLI command for each workflow, start-workflow-with-signal, query, and signal. Each command provides typed flags for configuring the corresponding inputs and options.

example/README.md

@@ -3,7 +3,9 @@
 In an initial terminal:
 1. Start temporal
     ```shell
-    temporal server start-dev --dynamic-config-value "frontend.enableUpdateWorkflowExecution=true"
+    temporal server start-dev \
+        --dynamic-config-value "frontend.enableUpdateWorkflowExecution=true" \
+        --dynamic-config-value "frontend.enableUpdateWorkflowExecutionAsyncAccepted=true"
     ```
 
 In a second terminal:

example/main.go

@@ -9,6 +9,7 @@ import (
 	"os"
 	"os/signal"
 	"syscall"
+	"time"
 
 	"github.com/cludden/protoc-gen-go-temporal/example/external"
 	examplev1 "github.com/cludden/protoc-gen-go-temporal/gen/example/v1"
@@ -52,7 +53,7 @@ func (wf *CreateFooWorkflow) Execute(ctx workflow.Context) (*examplev1.CreateFoo
 	workflow.Go(ctx, func(ctx workflow.Context) {
 		for {
 			signal, _ := wf.SetFooProgress.Receive(ctx)
-			wf.UpdateFooProgress(ctx, &examplev1.SetFooProgressRequest{Progress: signal.GetProgress()})
+			wf.setProgress(signal)
 		}
 	})
 
@@ -84,6 +85,13 @@ func (wf *CreateFooWorkflow) GetFooProgress() (*examplev1.GetFooProgressResponse
 
 // UpdateFooProgress defines the handler for an UpdateFooProgress update
 func (wf *CreateFooWorkflow) UpdateFooProgress(ctx workflow.Context, req *examplev1.SetFooProgressRequest) (*examplev1.GetFooProgressResponse, error) {
+	if err := workflow.Sleep(ctx, time.Second*45); err != nil {
+		return nil, err
+	}
+	return wf.setProgress(req)
+}
+
+func (wf *CreateFooWorkflow) setProgress(req *examplev1.SetFooProgressRequest) (*examplev1.GetFooProgressResponse, error) {
 	wf.progress = req.GetProgress()
 	switch {
 	case wf.progress < 0:

example/main_test.go

@@ -12,7 +12,11 @@ import (
 	"github.com/cludden/protoc-gen-go-temporal/mocks/go.temporal.io/sdk/clientutils"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/require"
+	"go.temporal.io/api/enums/v1"
+	"go.temporal.io/api/operatorservice/v1"
 	"go.temporal.io/sdk/client"
+	"go.temporal.io/sdk/testsuite"
+	"go.temporal.io/sdk/worker"
 )
 
 func TestCreateFooStartWorkflowOptions(t *testing.T) {
@@ -54,3 +58,67 @@ func TestCreateFooStartWorkflowOptions(t *testing.T) {
 	_, err := example.CreateFooAsync(ctx, &examplev1.CreateFooInput{RequestName: "bar"})
 	require.NoError(err)
 }
+
+func TestUpdate(t *testing.T) {
+	if testing.Short() {
+		t.SkipNow()
+	}
+
+	require, ctx := require.New(t), context.Background()
+
+	srv, err := testsuite.StartDevServer(ctx, testsuite.DevServerOptions{
+		LogLevel: "fatal",
+		ExtraArgs: []string{
+			"--dynamic-config-value", "frontend.enableUpdateWorkflowExecution=true",
+			"--dynamic-config-value", "frontend.enableUpdateWorkflowExecutionAsyncAccepted=true",
+		},
+	})
+	require.NoError(err)
+	defer srv.Stop()
+
+	c := srv.Client()
+	example := examplev1.NewExampleClient(c)
+
+	_, err = c.OperatorService().AddSearchAttributes(ctx, &operatorservice.AddSearchAttributesRequest{
+		Namespace: "default",
+		SearchAttributes: map[string]enums.IndexedValueType{
+			"foo":        enums.INDEXED_VALUE_TYPE_TEXT,
+			"created_at": enums.INDEXED_VALUE_TYPE_DATETIME,
+		},
+	})
+	require.NoError(err)
+
+	w := worker.New(c, examplev1.ExampleTaskQueue, worker.Options{})
+	examplev1.RegisterExampleActivities(w, &Activities{})
+	examplev1.RegisterExampleWorkflows(w, &Workflows{})
+	require.NoError(w.Start())
+	defer w.Stop()
+	defer c.Close()
+
+	run, err := example.CreateFooAsync(ctx, &examplev1.CreateFooInput{RequestName: "test"})
+	require.NoError(err)
+
+	require.NoError(run.SetFooProgress(ctx, &examplev1.SetFooProgressRequest{Progress: 5.7}))
+
+	query, err := run.GetFooProgress(ctx)
+	require.NoError(err)
+	require.Equal(float32(5.7), query.GetProgress())
+	require.Equal(examplev1.Foo_FOO_STATUS_CREATING.String(), query.GetStatus().String())
+
+	handle, err := run.UpdateFooProgressAsync(ctx, &examplev1.SetFooProgressRequest{Progress: 100})
+	require.NoError(err)
+
+	update, err := handle.Get(ctx)
+	require.NoError(err)
+	require.Equal(float32(100), update.GetProgress())
+	require.Equal(examplev1.Foo_FOO_STATUS_READY.String(), update.GetStatus().String())
+
+	// update, err := run.UpdateFooProgress(ctx, &examplev1.SetFooProgressRequest{Progress: 100})
+	// require.NoError(err)
+	// require.Equal(float32(100), update.GetProgress())
+	// require.Equal(examplev1.Foo_FOO_STATUS_READY.String(), update.GetStatus().String())
+
+	resp, err := run.Get(ctx)
+	require.NoError(err)
+	require.Equal(examplev1.Foo_FOO_STATUS_READY.String(), resp.GetFoo().GetStatus().String())
+}