Fix: Update

Author: vr-varad

cmd/dbos/migrate.go

@@ -33,8 +33,10 @@ func runMigrate(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	ctx := context.Background()
+
 	// Create DBOS context which will run migrations automatically for the system DB
-	_, err = createDBOSContext(dbURL)
+	_, err = createDBOSContext(dbURL, ctx)
 	if err != nil {
 		return fmt.Errorf("failed to create DBOS context: %w", err)
 	}

cmd/dbos/workflow.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"time"
 
@@ -89,8 +90,10 @@ func runWorkflowList(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	user_ctx := context.Background()
+
 	// Create DBOS context
-	ctx, err := createDBOSContext(dbURL)
+	ctx, err := createDBOSContext(dbURL, user_ctx)
 	if err != nil {
 		return err
 	}
@@ -194,8 +197,10 @@ func runWorkflowGet(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	user_ctx := context.Background()
+
 	// Create DBOS context
-	ctx, err := createDBOSContext(dbURL)
+	ctx, err := createDBOSContext(dbURL, user_ctx)
 	if err != nil {
 		return err
 	}
@@ -227,8 +232,10 @@ func runWorkflowSteps(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	user_ctx := context.Background()
+
 	// Create DBOS context
-	ctx, err := createDBOSContext(dbURL)
+	ctx, err := createDBOSContext(dbURL, user_ctx)
 	if err != nil {
 		return err
 	}
@@ -257,8 +264,10 @@ func runWorkflowCancel(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	user_ctx := context.Background()
+
 	// Create DBOS context
-	ctx, err := createDBOSContext(dbURL)
+	ctx, err := createDBOSContext(dbURL, user_ctx)
 	if err != nil {
 		return err
 	}
@@ -281,8 +290,10 @@ func runWorkflowResume(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	user_ctx := context.Background()
+
 	// Create DBOS context
-	ctx, err := createDBOSContext(dbURL)
+	ctx, err := createDBOSContext(dbURL, user_ctx)
 	if err != nil {
 		return err
 	}
@@ -312,8 +323,10 @@ func runWorkflowFork(cmd *cobra.Command, args []string) error {
 		return err
 	}
 
+	user_ctx := context.Background()
+
 	// Create DBOS context
-	ctx, err := createDBOSContext(dbURL)
+	ctx, err := createDBOSContext(dbURL, user_ctx)
 	if err != nil {
 		return err
 	}