update cancelation sample

Signed-off-by: Tihomir Surdilovic <[email protected]>

Author: tsurdilo

cancellation/activity.go

@@ -21,7 +21,7 @@ func (a *Activities) ActivityToBeCanceled(ctx context.Context) (string, error) {
 			activity.RecordHeartbeat(ctx, "")
 		case <-ctx.Done():
 			logger.Info("context is cancelled")
-			return "I am canceled by Done", nil
+			return "I am canceled by Done", ctx.Err()
 		}
 	}
 }

cancellation/workflow.go

@@ -31,10 +31,14 @@ func YourWorkflow(ctx workflow.Context) error {
 		if err != nil {
 			logger.Error("CleanupActivity failed", "Error", err)
 		}
+
 	}()
 
 	var result string
 	err := workflow.ExecuteActivity(ctx, a.ActivityToBeCanceled).Get(ctx, &result)
+	if err != nil {
+		return err
+	}
 	logger.Info(fmt.Sprintf("ActivityToBeCanceled returns %v, %v", result, err))
 
 	err = workflow.ExecuteActivity(ctx, a.ActivityToBeSkipped).Get(ctx, nil)