feat: Add support for multiplexed sessions#1383

Merged

rahul2393 merged 41 commits intogoogleapis:mainfrom

currantw:support-mux-session-2

Jun 10, 2025

Commits on Jun 4, 2025

Update `SessionOptions` to support `GOOGLE_CLOUD_SPANNER_FORCE_DISABLE_MULTIPLEXED_SESSIONS` and add unit tests.
currantw
committed
feat: Multiplexed sessions - Remove handling of `MethodNotImplemented` exception from `DatabaseSessionManager` and add unit tests.
currantw
committed
feat: Multiplexed sessions - Update `Connection` to use multiplexed sessions, add unit tests.
currantw
committed
cleanup: Rename `beforeNextRetry` to `before_next_retry`.
currantw
committed
cleanup: Fix a few unrelated typos.
currantw
committed
feat: Multiplexed sessions - Add ingest of precommit tokens to `_SnapshotBase` and update attributes and tests.
currantw
committed
feat: Multiplexed sessions - Deprecate `StreamedResultSet._source` (redundant as transaction ID is set via `_restart_on_unavailable`)
currantw
committed
feat: Multiplexed sessions - Move `_session_options` from `Database` to `Client` so that multiplexed are disabled for _all_ databases.
currantw
committed
feat: Multiplexed sessions - Deprecate `SessionCheckout` and update `Database.run_in_transaction` to not use it.
currantw
committed
feat: Multiplexed sessions - Deprecate `Database.session()` and minor cleanup.
currantw
committed
feat: Multiplexed sessions - Update `BatchSnapshot` to use database session manager.
currantw
committed
feat: Multiplexed sessions - Move `Batch` and `Transaction` attributes from class attributes to instance attributes.
currantw
committed
feat: Multiplexed sessions - Update pools so they don't use deprecated `database.session()`
currantw
committed
feat: Multiplexed sessions - Update session to remove class attributes, add TODOs, and make `Session._transaction` default to None. Plus add some `Optional` typing hints.
currantw
committed
feat: Multiplexed sessions - Move begin transaction logic from `Snapshot` to `_SnapshotBase` and update unit tests.
currantw
committed
feat: Multiplexed sessions - Remove begin transaction logic from `Transaction`, move to base class, update tests.
currantw
committed
feat: Multiplexed sessions - Add logic for beginning mutations-only transactions.
currantw
committed
feat: Multiplexed sessions - Cleanup and improve consistency of state checks, add `raises` documentation.
currantw
committed
feat: Multiplexed sessions - Cleanup documentation for `Batch.commit`, some minor cleanup.
currantw
committed
feat: Multiplexed sessions - Add logic for retrying commits if precommit token returned.
currantw
committed
feat: Multiplexed sessions - Remove `GOOGLE_CLOUD_SPANNER_FORCE_DISABLE_MULTIPLEXED_SESSIONS` and update tests.
currantw
committed
feat: Multiplexed sessions - Cleanup `TestDatabaseSessionManager` so that it doesn't depend on environment variable values.
currantw
committed
feat: Multiplexed sessions - Add type hints for `SessionOptions` and `DatabaseSessionManager`.
currantw
committed
feat: Multiplexed sessions - Fix `test_observability_options`
currantw
committed
feat: Multiplexed sessions - Update `_builders` to use mock scoped credentials.
currantw
committed
feat: Multiplexed sessions - Add helpers for mock scoped credentials for testing.
currantw
committed
feat: Multiplexed sessions - Fix failing `test_batch_insert_then_read`.
currantw
committed
feat: Multiplexed sessions - Fix failing `test_transaction_read_and_insert_then_rollback`.
currantw
committed
feat: Multiplexed sessions - Add test helper for multiplexed env vars.
currantw
committed
feat: Multiplexed sessions - Add unit tests for begin transaction base class, simplify `_SnapshotBase` tests, remove redundant tests.
currantw
committed
feat: Multiplexed sessions - Attempt to fix `test_transaction_read_and_insert_then_rollback` and add `build_request_id` helper method, fix `test_snapshot` and `test_transaction` failures.
currantw
committed
feat: Multiplexed sessions - Add test for log when new session created by maintenance thread.
currantw
committed
feat: Multiplexed sessions - Add additional multiplexed unit tests for `_SnapshotBase`.
currantw
committed
feat: Multiplexed sessions - Cleanup `Transaction` by extracting some constants for next step.
currantw
committed

Commits on Jun 9, 2025