|
49 | 49 | ) |
50 | 50 | from ldclient.migrations import OpTracker, Stage |
51 | 51 | from ldclient.plugin import EnvironmentMetadata |
52 | | -from ldclient.versioned_data_kind import FEATURES, SEGMENTS, VersionedDataKind |
53 | | - |
54 | | - |
55 | | -async def _get_store_item(store, kind: VersionedDataKind, key: str) -> Any: |
56 | | - # This decorator around store.get provides backward compatibility with any custom data |
57 | | - # store implementation that might still be returning a dict, instead of our data model |
58 | | - # classes like FeatureFlag. |
59 | | - item = await store.get(kind, key) |
60 | | - return kind.decode(item) if isinstance(item, dict) else item |
| 52 | +from ldclient.versioned_data_kind import FEATURES, SEGMENTS |
61 | 53 |
|
62 | 54 |
|
63 | 55 | class _NotStartedDataSystem: |
@@ -223,10 +215,10 @@ async def variation_eval_fn(key, context): |
223 | 215 | self.__big_segment_store_manager = big_segment_store_manager |
224 | 216 |
|
225 | 217 | async def get_flag_fn(key): |
226 | | - return await _get_store_item(self._data_system.store, FEATURES, key) |
| 218 | + return await self._data_system.store.get(FEATURES, key) |
227 | 219 |
|
228 | 220 | async def get_segment_fn(key): |
229 | | - return await _get_store_item(self._data_system.store, SEGMENTS, key) |
| 221 | + return await self._data_system.store.get(SEGMENTS, key) |
230 | 222 |
|
231 | 223 | async def get_membership_fn(key): |
232 | 224 | return await big_segment_store_manager.get_user_membership(key) |
@@ -523,7 +515,7 @@ async def _evaluate_internal(self, key: str, context: Context, default: Any, eve |
523 | 515 | return EvaluationDetail(default, None, error_reason('USER_NOT_SPECIFIED')), None |
524 | 516 |
|
525 | 517 | try: |
526 | | - flag = await _get_store_item(self._data_system.store, FEATURES, key) |
| 518 | + flag = await self._data_system.store.get(FEATURES, key) |
527 | 519 | except Exception as e: |
528 | 520 | log.error("Unexpected error while retrieving feature flag \"%s\": %s" % (key, repr(e))) |
529 | 521 | log.debug(traceback.format_exc()) |
|
0 commit comments