Fix: Handle max retries in awaitWorkflowResult

Author: vr-varad

dbos/errors.go

@@ -118,6 +118,14 @@ func newAwaitedWorkflowCancelledError(workflowID string) *DBOSError {
 	}
 }
 
+func newAwaitedWorkflowMaxStepRetriesExceeded(workflowID string) *DBOSError {
+	return &DBOSError{
+		Message:    fmt.Sprintf("Awaited workflow %s has exceeded the maximum number of step retries", workflowID),
+		Code:       MaxStepRetriesExceeded,
+		WorkflowID: workflowID,
+	}
+}
+
 func newWorkflowCancelledError(workflowID string) *DBOSError {
 	return &DBOSError{
 		Message: fmt.Sprintf("Workflow %s was cancelled", workflowID),

dbos/system_database.go

@@ -1221,6 +1221,8 @@ func (s *sysDB) awaitWorkflowResult(ctx context.Context, workflowID string, poll
 			return outputString, errors.New(*errorStr)
 		case WorkflowStatusCancelled:
 			return outputString, newAwaitedWorkflowCancelledError(workflowID)
+		case WorkflowStatusMaxRecoveryAttemptsExceeded:
+			return outputString, newAwaitedWorkflowMaxStepRetriesExceeded(workflowID)
 		default:
 			time.Sleep(pollInterval)
 		}