feat(BigQuery): Adding ConfigurationModifier action to CreateModelExtractJobOptions

Author: cy-yun

apis/Google.Cloud.BigQuery.V2/Google.Cloud.BigQuery.V2.Tests/CreateModelExtractJobOptionsTest.cs

@@ -1,4 +1,4 @@
-// Copyright 2021 Google LLC
+// Copyright 2021 Google LLC
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -24,11 +24,13 @@ public void ModifyRequest()
         {
             var options = new CreateModelExtractJobOptions
             {
-                DestinationFormat = ModelFormat.XGBoostBooster
+                DestinationFormat = ModelFormat.XGBoostBooster,
+                ConfigurationModifier = job => job.UseAvroLogicalTypes = true
             };
             JobConfigurationExtract extract = new JobConfigurationExtract();
             options.ModifyRequest(extract);
             Assert.Equal("ML_XGBOOST_BOOSTER", extract.DestinationFormat);
+            Assert.True(extract.UseAvroLogicalTypes);
         }
     }
-}
+}

apis/Google.Cloud.BigQuery.V2/Google.Cloud.BigQuery.V2/CreateModelExtractJobOptions.cs

@@ -27,12 +27,27 @@ public sealed class CreateModelExtractJobOptions : JobCreationOptions
         /// </summary>
         public ModelFormat? DestinationFormat { get; set; }
 
+        /// <summary>
+        /// Optional action to perform after preparing the request. If this property is non-null,
+        /// the <see cref="JobConfigurationExtract"/> used for a request will be passed to the delegate
+        /// before the request is executed. This allows for fine-grained modifications which aren't
+        /// otherwise directly supported by the properties in this options type.
+        /// </summary>
+        /// <remarks>
+        /// Prefer the properties on this type over this modifier to prepare the request.
+        /// Only use this modifier to configure aspects for which there are no properties available.
+        /// This modifier is applied to the request after all properties on this type have been applied.
+        /// The delegate is only called once per operation, even if the request is automatically retried.
+        /// </remarks>
+        public System.Action<JobConfigurationExtract> ConfigurationModifier { get; set; }
+
         internal void ModifyRequest(JobConfigurationExtract extract)
         {
             if (DestinationFormat != null)
             {
                 extract.DestinationFormat = EnumMap.ToApiValue(DestinationFormat.Value);
             }
+            ConfigurationModifier?.Invoke(extract);
         }
     }
 }