sparta-spanner

Sparta Google Spanner library.

Usage

import os

import sparta.spanner

instance_id = os.getenv("INSTANCE_ID")
database_id = os.getenv("DATABASE_ID")
db_service = sparta.spanner.DBService(
    instance_id=instance_id,
    database_id=database_id,
    pool_size=5,
    session_request_timeout=2,
)

# Blocking
results = sparta.spanner.zip_results(db_service.execute_sql("SELECT 1 as my_field"))
assert results is not None
assert len(results) == 1
assert results[0] == {"my_field": 1}


# Non-Blocking
def task(snapshot):
    result1 = sparta.spanner.zip_results(snapshot.execute_sql("SELECT 1 as my_field"))
    result2 = sparta.spanner.zip_results(snapshot.execute_sql("SELECT 2 as my_field"))
    return result1 + result2


results = await db_service.run_in_snapshot(task, multi_use=True)
assert results is not None
assert len(results) == 2
assert results[0] == {"my_field": 1}
assert results[1] == {"my_field": 2}