Skip to content

Commit

Permalink
fix(samples): batch_update() results processing error (#484)
Browse files Browse the repository at this point in the history
* fix(samples): batch_update() results processing error

* fix the comment

* minor fix

Co-authored-by: larkee <31196561+larkee@users.noreply.github.com>
  • Loading branch information
Ilya Gurov and larkee authored Aug 15, 2021
1 parent d557a8d commit bdd5f8b
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion samples/samples/snippets.py
Original file line number Diff line number Diff line change
Expand Up @@ -1511,6 +1511,8 @@ def delete_data_with_partitioned_dml(instance_id, database_id):
def update_with_batch_dml(instance_id, database_id):
"""Updates sample data in the database using Batch DML. """
# [START spanner_dml_batch_update]
from google.rpc.code_pb2 import OK

# instance_id = "your-spanner-instance"
# database_id = "your-spanner-db-id"

Expand All @@ -1531,7 +1533,13 @@ def update_with_batch_dml(instance_id, database_id):
)

def update_albums(transaction):
row_cts = transaction.batch_update([insert_statement, update_statement])
status, row_cts = transaction.batch_update([insert_statement, update_statement])

if status.code != OK:
# Do handling here.
# Note: the exception will still be raised when
# `commit` is called by `run_in_transaction`.
return

print("Executed {} SQL statements using Batch DML.".format(len(row_cts)))

Expand Down

0 comments on commit bdd5f8b

Please sign in to comment.