print step name in DBOSContext RunAsStep errors

Author: maxdml

dbos/workflow.go

@@ -935,25 +935,25 @@ func RunAsStep[R any](ctx DBOSContext, fn GenericStepFunc[R]) (R, error) {
 }
 
 func (c *dbosContext) RunAsStep(_ DBOSContext, fn StepFunc) (any, error) {
+	// Look up for step parameters in the context and set defaults
+	params, ok := c.Value(StepParamsKey).(*StepParams)
+	if !ok {
+		params = nil
+	}
+	params = setStepParamDefaults(params, runtime.FuncForPC(reflect.ValueOf(fn).Pointer()).Name())
+
 	// Get workflow state from context
 	wfState, ok := c.Value(workflowStateKey).(*workflowState)
 	if !ok || wfState == nil {
 		// TODO: try to print step name
-		return nil, newStepExecutionError("", "", "workflow state not found in context: are you running this step within a workflow?")
+		return nil, newStepExecutionError("", params.StepName, "workflow state not found in context: are you running this step within a workflow?")
 	}
 
 	// This should not happen when called from the package-level RunAsStep
 	if fn == nil {
-		return nil, newStepExecutionError(wfState.workflowID, "", "step function cannot be nil")
+		return nil, newStepExecutionError(wfState.workflowID, params.StepName, "step function cannot be nil")
 	}
 
-	// Look up for step parameters in the context and set defaults
-	params, ok := c.Value(StepParamsKey).(*StepParams)
-	if !ok {
-		params = nil
-	}
-	params = setStepParamDefaults(params, runtime.FuncForPC(reflect.ValueOf(fn).Pointer()).Name())
-
 	// If within a step, just run the function directly
 	if wfState.isWithinStep {
 		return fn(c)