Adding GetRehydrationToken to OperationInternalBase

Author: kinelski

sdk/core/Azure.Core/src/Shared/OperationInternalBase.cs

@@ -185,6 +185,12 @@ public Response WaitForCompletionResponse(CancellationToken cancellationToken)
         public Response WaitForCompletionResponse(TimeSpan pollingInterval, CancellationToken cancellationToken)
             => WaitForCompletionResponseAsync(async: false, pollingInterval, _waitForCompletionResponseScopeName, cancellationToken).EnsureCompleted();
 
+        /// <summary>
+        /// Get a token that can be used to rehydrate the operation.
+        /// </summary>
+        public RehydrationToken GetRehydrationToken(RequestMethod requestMethod, Uri startRequestUri, OperationFinalStateVia finalStateVia)
+            => NextLinkOperationImplementation.GetRehydrationToken(requestMethod, startRequestUri, RawResponse, finalStateVia);
+
         protected async ValueTask<Response> WaitForCompletionResponseAsync(bool async, TimeSpan? pollingInterval, string scopeName, CancellationToken cancellationToken)
         {
             // If _responseLock has the value, lockOrValue will contain that value, and no lock is acquired.

sdk/formrecognizer/Azure.AI.FormRecognizer/src/AnalyzeDocumentOperation.cs

@@ -206,6 +206,7 @@ async ValueTask<OperationState<AnalyzeResult>> IOperation<AnalyzeResult>.UpdateS
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<AnalyzeResult>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<AnalyzeResult>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/BuildDocumentClassifierOperation.cs

@@ -195,6 +195,7 @@ async ValueTask<OperationState<DocumentClassifierDetails>> IOperation<DocumentCl
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<DocumentClassifierDetails>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<DocumentClassifierDetails>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/BuildDocumentModelOperation.cs

@@ -195,6 +195,7 @@ async ValueTask<OperationState<DocumentModelDetails>> IOperation<DocumentModelDe
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<DocumentModelDetails>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<DocumentModelDetails>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/ClassifyDocumentOperation.cs

@@ -194,6 +194,7 @@ async ValueTask<OperationState<AnalyzeResult>> IOperation<AnalyzeResult>.UpdateS
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<AnalyzeResult>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<AnalyzeResult>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/ComposeDocumentModelOperation.cs

@@ -195,6 +195,7 @@ async ValueTask<OperationState<DocumentModelDetails>> IOperation<DocumentModelDe
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<DocumentModelDetails>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<DocumentModelDetails>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/CopyDocumentModelToOperation.cs

@@ -200,6 +200,7 @@ async ValueTask<OperationState<DocumentModelDetails>> IOperation<DocumentModelDe
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<DocumentModelDetails>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<DocumentModelDetails>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormRecognizerClient/RecognizeBusinessCardsOperation.cs

@@ -173,6 +173,7 @@ async ValueTask<OperationState<RecognizedFormCollection>> IOperation<RecognizedF
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormRecognizerClient/RecognizeContentOperation.cs

@@ -174,7 +174,8 @@ async ValueTask<OperationState<FormPageCollection>> IOperation<FormPageCollectio
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<FormPageCollection>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<FormPageCollection>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
 
         private static FormPageCollection ConvertValue(IReadOnlyList<PageResult> pageResults, IReadOnlyList<ReadResult> readResults)
         {

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormRecognizerClient/RecognizeCustomFormsOperation.cs

@@ -207,7 +207,8 @@ async ValueTask<OperationState<RecognizedFormCollection>> IOperation<RecognizedF
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
 
         private static RecognizedFormCollection ConvertToRecognizedForms(V2AnalyzeResult analyzeResult, string modelId)
         {

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormRecognizerClient/RecognizeIdentityDocumentsOperation.cs

@@ -172,6 +172,7 @@ async ValueTask<OperationState<RecognizedFormCollection>> IOperation<RecognizedF
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormRecognizerClient/RecognizeInvoicesOperation.cs

@@ -173,6 +173,7 @@ async ValueTask<OperationState<RecognizedFormCollection>> IOperation<RecognizedF
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormRecognizerClient/RecognizeReceiptsOperation.cs

@@ -172,6 +172,7 @@ async ValueTask<OperationState<RecognizedFormCollection>> IOperation<RecognizedF
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<RecognizedFormCollection>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormTrainingClient/CopyModelOperation.cs

@@ -206,7 +206,8 @@ async ValueTask<OperationState<CustomFormModelInfo>> IOperation<CustomFormModelI
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<CustomFormModelInfo>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<CustomFormModelInfo>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
 
         private static CustomFormModelInfo ConvertValue(CopyOperationResult result, string modelId, CustomFormModelStatus status)
         {

sdk/formrecognizer/Azure.AI.FormRecognizer/src/FormTrainingClient/CreateCustomFormModelOperation.cs

@@ -177,6 +177,7 @@ async ValueTask<OperationState<CustomFormModel>> IOperation<CustomFormModel>.Upd
         }
 
         // This method is never invoked since we don't override Operation<T>.GetRehydrationToken.
-        RehydrationToken IOperation<CustomFormModel>.GetRehydrationToken() => default;
+        RehydrationToken IOperation<CustomFormModel>.GetRehydrationToken() =>
+            throw new NotSupportedException($"{nameof(GetRehydrationToken)} is not supported.");
     }
 }