feat: [google-cloud-dataproc] Add cluster_tier to support creating premium clusters (#14047)

- [ ] Regenerate this pull request now.

PiperOrigin-RevId: 776861070

Source-Link:
googleapis/googleapis@ea37c68

Source-Link:
googleapis/googleapis-gen@cbf711c
Copy-Tag:
eyJwIjoicGFja2FnZXMvZ29vZ2xlLWNsb3VkLWRhdGFwcm9jLy5Pd2xCb3QueWFtbCIsImgiOiJjYmY3MTFjMDQ2NTIyNGRjNTBhNDBlMDczZGUzZGVkMjVkNGRlODAyIn0=

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>

Author: gcf-owl-bot[bot]

packages/google-cloud-dataproc/google/cloud/dataproc/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "5.20.0"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-dataproc/google/cloud/dataproc_v1/gapic_version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "5.20.0"  # {x-release-please-version}
+__version__ = "0.0.0"  # {x-release-please-version}

packages/google-cloud-dataproc/google/cloud/dataproc_v1/types/clusters.py

@@ -179,6 +179,8 @@ class ClusterConfig(proto.Message):
     r"""The cluster config.
 
     Attributes:
+        cluster_tier (google.cloud.dataproc_v1.types.ClusterConfig.ClusterTier):
+            Optional. The cluster tier.
         config_bucket (str):
             Optional. A Cloud Storage bucket used to stage job
             dependencies, config files, and job driver console output.
@@ -258,6 +260,26 @@ class ClusterConfig(proto.Message):
             Optional. The node group settings.
     """
 
+    class ClusterTier(proto.Enum):
+        r"""The cluster tier.
+
+        Values:
+            CLUSTER_TIER_UNSPECIFIED (0):
+                Not set. Works the same as CLUSTER_TIER_STANDARD.
+            CLUSTER_TIER_STANDARD (1):
+                Standard Dataproc cluster.
+            CLUSTER_TIER_PREMIUM (2):
+                Premium Dataproc cluster.
+        """
+        CLUSTER_TIER_UNSPECIFIED = 0
+        CLUSTER_TIER_STANDARD = 1
+        CLUSTER_TIER_PREMIUM = 2
+
+    cluster_tier: ClusterTier = proto.Field(
+        proto.ENUM,
+        number=29,
+        enum=ClusterTier,
+    )
     config_bucket: str = proto.Field(
         proto.STRING,
         number=1,

packages/google-cloud-dataproc/samples/generated_samples/snippet_metadata_google.cloud.dataproc.v1.json

@@ -8,7 +8,7 @@
     ],
     "language": "PYTHON",
     "name": "google-cloud-dataproc",
-    "version": "5.20.0"
+    "version": "0.1.0"
   },
   "snippets": [
     {

packages/google-cloud-dataproc/tests/unit/gapic/dataproc_v1/test_cluster_controller.py

@@ -6166,6 +6166,7 @@ def test_create_cluster_rest_call_success(request_type):
         "project_id": "project_id_value",
         "cluster_name": "cluster_name_value",
         "config": {
+            "cluster_tier": 1,
             "config_bucket": "config_bucket_value",
             "temp_bucket": "temp_bucket_value",
             "gce_cluster_config": {
@@ -6577,6 +6578,7 @@ def test_update_cluster_rest_call_success(request_type):
         "project_id": "project_id_value",
         "cluster_name": "cluster_name_value",
         "config": {
+            "cluster_tier": 1,
             "config_bucket": "config_bucket_value",
             "temp_bucket": "temp_bucket_value",
             "gce_cluster_config": {

packages/google-cloud-dataproc/tests/unit/gapic/dataproc_v1/test_workflow_template_service.py

@@ -5789,6 +5789,7 @@ def test_create_workflow_template_rest_call_success(request_type):
             "managed_cluster": {
                 "cluster_name": "cluster_name_value",
                 "config": {
+                    "cluster_tier": 1,
                     "config_bucket": "config_bucket_value",
                     "temp_bucket": "temp_bucket_value",
                     "gce_cluster_config": {
@@ -6569,6 +6570,7 @@ def test_instantiate_inline_workflow_template_rest_call_success(request_type):
             "managed_cluster": {
                 "cluster_name": "cluster_name_value",
                 "config": {
+                    "cluster_tier": 1,
                     "config_bucket": "config_bucket_value",
                     "temp_bucket": "temp_bucket_value",
                     "gce_cluster_config": {
@@ -7075,6 +7077,7 @@ def test_update_workflow_template_rest_call_success(request_type):
             "managed_cluster": {
                 "cluster_name": "cluster_name_value",
                 "config": {
+                    "cluster_tier": 1,
                     "config_bucket": "config_bucket_value",
                     "temp_bucket": "temp_bucket_value",
                     "gce_cluster_config": {