Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

docs: add sample for managed autoscaler #1111

Merged
merged 4 commits into from
Mar 7, 2024
Merged

docs: add sample for managed autoscaler #1111

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
4ef20f0
docs: add sample for managed autoscaler
rahul2393 Mar 6, 2024
e4dafec
incorporate suggestions
rahul2393 Mar 7, 2024
1a67cf8
Merge branch 'main' into managed_autoscalar
rahul2393 Mar 7, 2024
76d0515
Merge branch 'main' into managed_autoscalar
harshachinta Mar 7, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
Next Next commit
docs: add sample for managed autoscaler
  • Loading branch information
@rahul2393
rahul2393 committed Mar 6, 2024
commit 4ef20f01b23d9ce7d624e15d33d826dd58f35697
54 changes: 54 additions & 0 deletions samples/samples/snippets.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3017,6 +3017,58 @@ def directed_read_options(
# [END spanner_directed_read]


# [START spanner_create_instance_with_autoscaling_config]
def create_instance_with_autoscaling_config(instance_id):
"""Creates a Cloud Spanner instance with an autoscaling configuration."""
from google.cloud.spanner_admin_instance_v1.types import \
spanner_instance_admin

spanner_client = spanner.Client()

config_name = "{}/instanceConfigs/regional-us-central1".format(
spanner_client.project_name
)

autoscaling_config = spanner_instance_admin.AutoscalingConfig(
autoscaling_limits=spanner_instance_admin.AutoscalingConfig.AutoscalingLimits(
rahul2393 marked this conversation as resolved.
Show resolved Hide resolved
min_nodes=1,
max_nodes=2,
),
autoscaling_targets=spanner_instance_admin.AutoscalingConfig.AutoscalingTargets(
high_priority_cpu_utilization_percent=65,
rahul2393 marked this conversation as resolved.
Show resolved Hide resolved
storage_utilization_percent=95,
),
)

request = spanner_instance_admin.CreateInstanceRequest(
rahul2393 marked this conversation as resolved.
Show resolved Hide resolved
parent=spanner_client.project_name,
instance_id=instance_id,
instance=spanner_instance_admin.Instance(
config=config_name,
display_name="This is a display name.",
autoscaling_config=autoscaling_config,
labels={
"cloud_spanner_samples": "true",
"sample_name": "snippets-create_instance_with_autoscaling_config",
"created": str(int(time.time())),
},
),
)

operation = spanner_client.instance_admin_api.create_instance(request=request)

print("Waiting for operation to complete...")
instance = operation.result(OPERATION_TIMEOUT_SECONDS)

print(
"Created instance {} with {} autoscaling config".format(
instance_id, instance.autoscaling_config
)
)


# [END spanner_create_instance_with_autoscaling_config]

if __name__ == "__main__": # noqa: C901
parser = argparse.ArgumentParser(
description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
Expand Down Expand Up @@ -3290,3 +3342,5 @@ def directed_read_options(
)
elif args.command == "directed_read_options":
directed_read_options(args.instance_id, args.database_id)
elif args.command == "create_instance_with_autoscaling_config":
create_instance_with_autoscaling_config(args.instance_id)
12 changes: 12 additions & 0 deletions samples/samples/snippets_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -154,6 +154,18 @@ def test_create_instance_with_processing_units(capsys, lci_instance_id):
retry_429(instance.delete)()


def test_create_instance_with_autoscaling_config(capsys, lci_instance_id):
retry_429(snippets.create_instance_with_autoscaling_config)(
lci_instance_id,
)
out, _ = capsys.readouterr()
assert lci_instance_id in out
assert "autoscaling config" in out
spanner_client = spanner.Client()
instance = spanner_client.instance(lci_instance_id)
retry_429(instance.delete)()


def test_update_database(capsys, instance_id, sample_database):
snippets.update_database(instance_id, sample_database.database_id)
out, _ = capsys.readouterr()
Expand Down