Skip to content

feat: Add async big segment store manager and async Redis adapter#462

Open
jsonbailey wants to merge 6 commits into
mainfrom
jb/sdk-2729/async-big-segments
Open

feat: Add async big segment store manager and async Redis adapter#462
jsonbailey wants to merge 6 commits into
mainfrom
jb/sdk-2729/async-big-segments

Conversation

@jsonbailey

@jsonbailey jsonbailey commented Jul 20, 2026

Copy link
Copy Markdown
Contributor

What

The async big-segments layer for the async Python SDK.

  • impl/async_big_segments.pyAsyncBigSegmentStoreManager (async twin of the sync manager: membership caching via ExpiringDict, status polling via AsyncRepeatingTask) + BigSegmentStoreStatusProviderImpl.
  • impl/integrations/redis/async_redis_big_segment_store.py — async Redis-backed AsyncBigSegmentStore (redis.asyncio).
  • integrations/__init__.py — new Redis.async_big_segment_store(...) factory (local import of the adapter, so redis stays optional).
  • async_config.pyseeded with AsyncBigSegmentsConfig only (see below).
  • Tests: testing/impl/test_async_big_segments.py (unit) + testing/integrations/test_async_redis.py (live-Redis integration, guarded by skip_database_tests).
  • pyproject.toml — bumps redis to >=4.2.0 (the redis.asyncio API requires redis-py 4.2+).

Note on async_config.py

AsyncBigSegmentStoreManager needs AsyncBigSegmentsConfig, which lives in async_config.py. The full AsyncConfig isn't extractable yet (it imports AsyncHook/AsyncPlugin/AsyncEventProcessor, which land in later slices), so this PR seeds async_config.py with just the self-contained AsyncBigSegmentsConfig (it depends only on the already-merged AsyncBigSegmentStore). A later slice grows the file with AsyncConfig — mirroring how the sync SDK keeps BigSegmentsConfig alongside Config in config.py.

Why

Part of the async Python SDK (epic SDK-60). Depends on the async support classes (#451) and async interfaces/store (#457), both merged.

Validation

  • uv run pytest ldclient/testing --ignore=ldclient/testing/integrations — 1033 passed, 2 skipped
  • uv run pytest ldclient/testing/impl/test_async_big_segments.py — 14 passed
  • uv run mypy ldclient — clean; isort / pycodestyle — clean
  • Redis integration tests (test_async_redis.py) require a live Redis (run in CI); properly guarded by skip_database_tests.

SDK-2729 (epic SDK-60).


Note

Medium Risk
Sync big-segment code moved to shared helpers (behavior should be unchanged), and the redis optional dependency minimum increases; the async path is experimental but will affect segment membership when wired into the async client.

Overview
Adds the experimental async Big Segments stack: AsyncBigSegmentsConfig, AsyncBigSegmentStoreManager (membership cache, background status polling via AsyncRepeatingTask, and an inline poll on the first membership lookup if the poller has not run yet), plus Redis.async_big_segment_store() backed by redis.asyncio.

Sync BigSegmentStoreManager is refactored to share big_segments_common (EMPTY_MEMBERSHIP, user-key hashing, staleness, and BigSegmentStoreStatusProviderImpl) with the async manager so behavior stays aligned without duplicating logic.

pyproject.toml raises the optional/dev redis floor to >=4.2.0 for redis.asyncio. Unit and live-Redis integration tests cover the new manager and adapter.

Reviewed by Cursor Bugbot for commit e428549. Bugbot is set up for automated code reviews on this repo. Configure here.

Adds AsyncBigSegmentStoreManager, the async Redis-backed AsyncBigSegmentStore adapter, and Redis.async_big_segment_store(). Seeds async_config.py with the self-contained AsyncBigSegmentsConfig (the full AsyncConfig lands later). Bumps the redis dependency to >=4.2.0 (required by redis.asyncio).
Drop the max_connections=10 default (and the now-unneeded redis_opts copy) so the async store passes redis_opts straight to from_url, matching the sync store. The sync side removed this cap in #387 (it throttled connections under load); redis.asyncio otherwise defaults to an effectively-unlimited pool.
…egments_common

Both big_segments.py and async_big_segments.py defined the class identically (bar one docstring word); move it to a shared module imported by both, matching the evaluator_common/client_common pattern.
Fold _hash_for_user_key, EMPTY_MEMBERSHIP, and is_stale (as a free function) into big_segments_common, imported by both big_segments.py and async_big_segments.py.
Drop the redundant EMPTY_MEMBERSHIP class attribute and reference the
big_segments_common module constant directly instead of via self.
@jsonbailey
jsonbailey marked this pull request as ready for review July 21, 2026 16:24
@jsonbailey
jsonbailey requested a review from a team as a code owner July 21, 2026 16:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant