feat: support required_registration_fraction for secondary workers

Google APIs · copybara-github · commit a70dd17d1287 · 2023-11-01T22:36:59.000-07:00
PiperOrigin-RevId: 578743167
diff --git a/google/cloud/dataproc/v1/clusters.proto b/google/cloud/dataproc/v1/clusters.proto
@@ -20,9 +20,11 @@ import "google/api/annotations.proto";
 import "google/api/client.proto";
 import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
+import "google/cloud/dataproc/v1/operations.proto";
 import "google/cloud/dataproc/v1/shared.proto";
 import "google/longrunning/operations.proto";
 import "google/protobuf/duration.proto";
+import "google/protobuf/empty.proto";
 import "google/protobuf/field_mask.proto";
 import "google/protobuf/timestamp.proto";
 import "google/protobuf/wrappers.proto";
@@ -693,6 +695,23 @@ message InstanceGroupConfig {
   // provisioning models.
   InstanceFlexibilityPolicy instance_flexibility_policy = 13
       [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Configuration to handle the startup of instances during cluster
+  // create and update process.
+  StartupConfig startup_config = 14 [(google.api.field_behavior) = OPTIONAL];
+}
+
+// Configuration to handle the startup of instances during cluster create and
+// update process.
+message StartupConfig {
+  // Optional. The config setting to enable cluster creation/ updation to be
+  // successful only after required_registration_fraction of instances are up
+  // and running. This configuration is applicable to only secondary workers for
+  // now. The cluster will fail if required_registration_fraction of instances
+  // are not available. This will include instance creation, agent registration,
+  // and service registration (if enabled).
+  optional double required_registration_fraction = 1
+      [(google.api.field_behavior) = OPTIONAL];
 }
 
 // A reference to a Compute Engine instance.
