chore: add samples for transaction timeout configuration

Author: alkatrivedi

samples/samples/snippets.py

@@ -2509,6 +2509,35 @@ def update_venues(transaction):
 
     # [END spanner_set_transaction_tag]
 
+def set_transaction_timeout(instance_id, database_id):
+    """Executes a transaction with a transaction timeout."""
+    # [START spanner_transaction_timeout]
+    # instance_id = "your-spanner-instance"
+    # database_id = "your-spanner-db-id"
+    spanner_client = spanner.Client()
+    instance = spanner_client.instance(instance_id)
+    database = instance.database(database_id)
+
+    def read_then_write(transaction):
+        # Read records.
+        results = transaction.execute_sql(
+            "SELECT SingerId, FirstName, LastName FROM Singers ORDER BY LastName, FirstName"
+        )
+        for result in results:
+            print("SingerId: {}, FirstName: {}, LastName: {}".format(*result))
+        
+        # Insert a record.
+        row_ct = transaction.execute_update(
+            "INSERT INTO Singers (SingerId, FirstName, LastName) "
+            " VALUES (100, 'George', 'Washington')"
+        )
+        print("{} record(s) inserted.".format(row_ct))
+
+        
+
+    database.run_in_transaction(read_then_write, timeout_secs=60)
+
+    # [END spanner_transaction_timeout]
 
 def set_request_tag(instance_id, database_id):
     """Executes a snapshot read with a request tag."""
@@ -3617,6 +3646,7 @@ def add_split_points(instance_id, database_id):
     subparsers.add_parser("add_column", help=add_column.__doc__)
     subparsers.add_parser("update_data", help=update_data.__doc__)
     subparsers.add_parser("set_max_commit_delay", help=set_max_commit_delay.__doc__)
+    subparsers.add_parser("set_transaction_timeout", help=set_transaction_timeout.__doc__)
     subparsers.add_parser(
         "query_data_with_new_column", help=query_data_with_new_column.__doc__
     )
@@ -3783,6 +3813,8 @@ def add_split_points(instance_id, database_id):
         update_data(args.instance_id, args.database_id)
     elif args.command == "set_max_commit_delay":
         set_max_commit_delay(args.instance_id, args.database_id)
+    elif args.command == "set_transaction_timeout":
+        set_transaction_timeout(args.instance_id, args.database_id)
     elif args.command == "query_data_with_new_column":
         query_data_with_new_column(args.instance_id, args.database_id)
     elif args.command == "read_write_transaction":

samples/samples/snippets_test.py

@@ -854,6 +854,11 @@ def test_set_transaction_tag(capsys, instance_id, sample_database):
     assert "Venue capacities updated." in out
     assert "New venue inserted." in out
 
+@pytest.mark.dependency(depends=["insert_datatypes_data"])
+def test_set_transaction_timeout(capsys, instance_id, sample_database):
+    snippets.set_transaction_timeout(instance_id, sample_database.database_id)
+    out, _ = capsys.readouterr()
+    assert "1 record(s) inserted." in out
 
 @pytest.mark.dependency(depends=["insert_data"])
 def test_set_request_tag(capsys, instance_id, sample_database):