Skip to content

Commit

Permalink
Close HappyBase connection in bigtable/hello sample.
Browse files Browse the repository at this point in the history
I noticed that the `bigtable/hello` sample was not quitting when I sent
a `Ctrl-C` this should fix that problem.
  • Loading branch information
tswast committed Jun 16, 2016
1 parent 0976714 commit d9f544a
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions bigtable/hello/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,13 +37,11 @@ def main(project_id, cluster_id, zone, table_name):
# The client must be created with admin=True because it will create a
# table.
client = bigtable.Client(project=project_id, admin=True)
cluster = client.cluster(zone, cluster_id)
connection = happybase.Connection(cluster=cluster)
# [END connecting_to_bigtable]

with client:
cluster = client.cluster(zone, cluster_id)
cluster.reload()
connection = happybase.Connection(cluster=cluster)
# [END connecting_to_bigtable]

try:
# [START creating_a_table]
print('Creating the {} table.'.format(table_name))
column_family_name = 'cf1'
Expand Down Expand Up @@ -95,6 +93,8 @@ def main(project_id, cluster_id, zone, table_name):
print('Deleting the {} table.'.format(table_name))
connection.delete_table(table_name)
# [END deleting_a_table]
finally:
connection.close()


if __name__ == '__main__':
Expand Down

0 comments on commit d9f544a

Please sign in to comment.