Wait until workflows are finished for scale sample

Author: cschleiden

samples/scale/starter/main.go

@@ -3,6 +3,8 @@ package main
 import (
 	"context"
 	"log"
+	"sync"
+	"time"
 
 	"github.com/cschleiden/go-workflows/backend/mysql"
 	"github.com/cschleiden/go-workflows/client"
@@ -14,23 +16,37 @@ func main() {
 	ctx := context.Background()
 
 	//b := sqlite.NewSqliteBackend("../scale.sqlite")
-	b := mysql.NewMysqlBackend("localhost", 3306, "root", "SqlPassw0rd", "simple")
+	b := mysql.NewMysqlBackend("localhost", 3306, "root", "SqlPassw0rd", "scale")
 
 	// Start workflow via client
 	c := client.New(b)
 
+	wg := &sync.WaitGroup{}
+
+	now := time.Now()
+
 	for i := 0; i < 100; i++ {
-		startWorkflow(ctx, c)
+		wg.Add(1)
+
+		go startWorkflow(ctx, c, wg)
 	}
+
+	wg.Wait()
+
+	log.Println("Finished in", time.Since(now))
 }
 
-func startWorkflow(ctx context.Context, c client.Client) {
+func startWorkflow(ctx context.Context, c client.Client, wg *sync.WaitGroup) {
 	wf, err := c.CreateWorkflowInstance(ctx, client.WorkflowInstanceOptions{
 		InstanceID: uuid.NewString(),
 	}, scale.Workflow1, "Hello world "+uuid.NewString())
 	if err != nil {
 		panic("could not start workflow")
 	}
 
-	log.Println("Started workflow", wf.GetInstanceID())
+	// log.Println("Started workflow", wf.GetInstanceID())
+
+	c.WaitForWorkflowInstance(ctx, wf, time.Second*30)
+
+	wg.Done()
 }

samples/scale/worker/main.go

@@ -5,7 +5,6 @@ import (
 	"log"
 	"os"
 	"os/signal"
-	"time"
 
 	"github.com/cschleiden/go-workflows/backend"
 	"github.com/cschleiden/go-workflows/backend/mysql"
@@ -17,7 +16,7 @@ func main() {
 	ctx, cancel := context.WithCancel(context.Background())
 
 	//b := sqlite.NewSqliteBackend("../scale.sqlite?_busy_timeout=10000")
-	b := mysql.NewMysqlBackend("localhost", 3306, "root", "SqlPassw0rd", "simple")
+	b := mysql.NewMysqlBackend("localhost", 3306, "root", "SqlPassw0rd", "scale")
 
 	// Run worker
 	go RunWorker(ctx, b)
@@ -28,11 +27,16 @@ func main() {
 
 	log.Println("Shutting down")
 	cancel()
-	time.Sleep(5 * time.Second)
 }
 
 func RunWorker(ctx context.Context, mb backend.Backend) {
-	w := worker.New(mb, nil)
+	w := worker.New(mb, &worker.Options{
+		WorkflowPollers:          2,
+		MaxParallelWorkflowTasks: 100,
+		ActivityPollers:          2,
+		MaxParallelActivityTasks: 100,
+		HeartbeatWorkflowTasks:   false,
+	})
 
 	w.RegisterWorkflow(scale.Workflow1)