Sample utilizing workflow registration by name

Author: cschleiden

samples/workflow-registration/workflow.go

@@ -0,0 +1,34 @@
+package main
+
+import (
+	"context"
+
+	"github.com/cschleiden/go-workflows/activity"
+	"github.com/cschleiden/go-workflows/workflow"
+)
+
+func Workflow1(ctx workflow.Context, msg string) (int, error) {
+	logger := workflow.Logger(ctx)
+	logger.Debug("Entering Workflow1", "msg", msg)
+	defer logger.Debug("Leaving Workflow1")
+
+	r1, err := workflow.ExecuteActivity[int](ctx, workflow.DefaultActivityOptions, Activity1, 35, 12).Get(ctx)
+	if err != nil {
+		panic("error getting activity 1 result")
+	}
+	logger.Debug("R1 result", "r1", r1)
+
+	return r1, nil
+}
+
+func SubWorkflow1(ctx workflow.Context, msg string) (int, error) {
+	return 42, nil
+}
+
+func Activity1(ctx context.Context, a, b int) (int, error) {
+	logger := activity.Logger(ctx)
+	logger.Debug("Entering Activity1")
+	defer logger.Debug("Leaving Activity1")
+
+	return a + b, nil
+}

samples/workflow-registration/workflow_registration.go

@@ -0,0 +1,67 @@
+package main
+
+import (
+	"context"
+	"log"
+	"time"
+
+	"github.com/cschleiden/go-workflows/backend"
+	"github.com/cschleiden/go-workflows/client"
+	"github.com/cschleiden/go-workflows/internal/workflow"
+	"github.com/cschleiden/go-workflows/samples"
+	"github.com/cschleiden/go-workflows/worker"
+
+	"github.com/google/uuid"
+)
+
+func main() {
+	ctx, cancel := context.WithCancel(context.Background())
+
+	b := samples.GetBackend("simple")
+
+	// Run worker
+	w := RunWorker(ctx, b)
+
+	// Start workflow via client
+	c := client.New(b)
+
+	runWorkflow(ctx, c)
+
+	cancel()
+
+	if err := w.WaitForCompletion(); err != nil {
+		panic("could not stop worker" + err.Error())
+	}
+}
+
+func runWorkflow(ctx context.Context, c *client.Client) {
+	wf, err := c.CreateWorkflowInstance(ctx, client.WorkflowInstanceOptions{
+		InstanceID: uuid.NewString(),
+	}, "WorkflowName", "Hello world")
+	if err != nil {
+		log.Fatal(err)
+		panic("could not start workflow")
+	}
+
+	result, err := client.GetWorkflowResult[int](ctx, c, wf, time.Second*10)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	log.Println("Workflow finished. Result:", result)
+}
+
+func RunWorker(ctx context.Context, mb backend.Backend) *worker.Worker {
+	w := worker.New(mb, nil)
+
+	w.RegisterWorkflow(Workflow1, workflow.WithName("WorkflowName"))
+	w.RegisterWorkflow(SubWorkflow1, workflow.WithName("SubWorkflowName"))
+
+	w.RegisterActivity(Activity1)
+
+	if err := w.Start(ctx); err != nil {
+		panic("could not start worker")
+	}
+
+	return w
+}

samples/workflow-registration/workflow_registration_test.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"context"
+	"testing"
+
+	"github.com/cschleiden/go-workflows/tester"
+	"github.com/stretchr/testify/mock"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_Workflow(t *testing.T) {
+	tester := tester.NewWorkflowTester[int](Workflow1)
+
+	tester.OnSubWorkflowByName("SubWorkflowName", mock.Anything, "Hello world").Return(42, nil)
+
+	tester.OnActivity(Activity1, mock.Anything, 35, 12).Return(47, nil)
+
+	tester.Execute(context.Background(), "Hello world")
+
+	require.True(t, tester.WorkflowFinished())
+
+	wr, werr := tester.WorkflowResult()
+	require.Equal(t, 47, wr)
+	require.Empty(t, werr)
+	tester.AssertExpectations(t)
+}