google_cloud_managed_kafka_cluster_creation_quickstart.py

1from google.api_core.exceptions import AlreadyExists
2from google.cloud import managedkafka_v1
3
4
5def create_cluster(
6    project_id: str,
7    region: str,
8    cluster_id: str,
9    subnet: str,
10    cpu: int,
11    memory_bytes: int,
12) -> None:
13    """
14    Creates a Managed Service for Apache Kafka cluster.
15
16    Args:
17        project_id: Google Cloud project ID.
18        region: Google Cloud region.
19        cluster_id: ID of the cluster to create.
20        subnet: The name of the subnet to which the cluster is connected.
21            Format: projects/{project_id}/regions/{region}/subnetworks/{subnet_id}
22        cpu: The number of CPUs for the cluster.
23        memory_bytes: The amount of memory in bytes for the cluster.
24    """
25    client = managedkafka_v1.ManagedKafkaClient()
26
27    cluster = managedkafka_v1.Cluster()
28    cluster.name = client.cluster_path(project_id, region, cluster_id)
29    cluster.gcp_config = managedkafka_v1.GcpConfig(
30        access_config=managedkafka_v1.AccessConfig(
31            network_configs=[managedkafka_v1.NetworkConfig(subnet=subnet)]
32        )
33    )
34    cluster.capacity_config = managedkafka_v1.CapacityConfig(
35        vcpu_count=cpu, memory_bytes=memory_bytes
36    )
37
38    request = managedkafka_v1.CreateClusterRequest(
39        parent=client.common_location_path(project_id, region),
40        cluster_id=cluster_id,
41        cluster=cluster,
42    )
43
44    try:
45        operation = client.create_cluster(request=request)
46        print("Waiting for operation to complete...")
47        response = operation.result()
48        print(f"Created cluster: {response.name}")
49    except AlreadyExists:
50        print(f"Cluster {cluster_id} already exists.")
51    except Exception as e:
52        print(f"Error creating cluster: {e}")
53
54
55if __name__ == "__main__":
56    # TODO(developer): Replace these variables before running the script.
57    PROJECT_ID = "your-project-id"
58    REGION = "us-central1"
59    CLUSTER_ID = "your-cluster-id"
60    # Subnet format: projects/{project_id}/regions/{region}/subnetworks/{subnet_id}
61    SUBNET = f"projects/{PROJECT_ID}/regions/{REGION}/subnetworks/default"
62    CPU = 3
63    MEMORY_BYTES = 3221225472  # 3 GiB
64
65    create_cluster(PROJECT_ID, REGION, CLUSTER_ID, SUBNET, CPU, MEMORY_BYTES)