\jBdZddlmZmZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZGdd ZGd d e e ZGd d Zy)aD This submodule contains basic classes related to the feature store. The feature store is the SDK component that holds the last known state of all feature flags, as received from LaunchDarkly. This submodule does not include specific integrations with external storage systems; those are in :class:`ldclient.integrations`. )CallableAny) OrderedDict defaultdict) ReadWriteLock)log)DiagnosticDescription FeatureStore)VersionedDataKindceZdZdZdZdZeefdedefdZe ddZ e dd Z e de fd Ze defd Ze defd Zy ) CacheConfigzbEncapsulates caching parameters for feature store implementations that support local caching. g.@i expirationcapacityc ||_||_y)aConstructs an instance of CacheConfig. :param expiration: the cache TTL, in seconds. Items will be evicted from the cache after this amount of time from the time when they were originally cached. If the time is less than or equal to zero, caching is disabled. :param capacity: the maximum number of items that can be in the cache at a time N) _expiration _capacity)selfrrs @/root/env/lib/python3.12/site-packages/ldclient/feature_store.py__init__zCacheConfig.__init__s&!returnctS)zReturns an instance of CacheConfig with default properties. By default, caching is enabled. This is the same as calling the constructor with no parameters. r rrdefaultzCacheConfig.default&s }rctdS)zWReturns an instance of CacheConfig specifying that caching should be disabled. r)rrrrrdisabledzCacheConfig.disabled-s**rc |jdkDS)zBReturns True if caching is enabled in this configuration. rrrs renabledzCacheConfig.enabled3s!##rc|jS)z6Returns the configured cache TTL, in seconds. rr s rrzCacheConfig.expiration9src|jS)zBReturns the configured maximum number of cacheable items. )rr s rrzCacheConfig.capacity?s~~rN)rr )__name__ __module__ __qualname____doc__DEFAULT_EXPIRATIONDEFAULT_CAPACITYfloatintr staticmethodrrpropertyboolr!rrrrrr r s&8!1 "" " " ++ $$$  E  #rr c ~eZdZdZdZdfdededeegefdefdZ d Z d Z ded e fd Z d ZedefdZdZy)InMemoryFeatureStoreznThe default feature store implementation, which holds all data in a thread-safe data structure in memory. cXt|_d|_tt|_y)z8Constructs an instance of InMemoryFeatureStore. FN)r_lock _initializedrdict_itemsr s rrzInMemoryFeatureStore.__init__Js!#_ !!$' rc|S)Nr)xs rzInMemoryFeatureStore.Qs^_rkindkeycallbackrc |jj|j|}|j|}|Ct j d||j |d|jjSd|vrH|drCt j d||j |d|jjS|||jjS#|jjwxYw) Nz7Attempted to get missing key %s in '%s', returning Nonedeletedz7Attempted to get deleted key %s in '%s', returning None)r2rlockr5getrdebug namespacerunlock)rr9r:r; itemsOfKinditems rr@zInMemoryFeatureStore.getQs ! JJ   ++d+K??3'D| SUXZ^ZhZhi~ JJ   D T)_ SUXZ^ZhZhi~ JJ   D> JJ   DJJ   sA$C/1C/ C//D c |jj|j|}|td|j D|jj S#|jj wxYw)r=c3>K|]\}}d|vs|dr||fyw)r>Nr).0kis r z+InMemoryFeatureStore.all..hs) tDAqI]^L^ghirgs!Q ts )r2r?r5r4itemsrC)rr9r;rDs rallzInMemoryFeatureStore.allbsb ! JJ   ++d+KD tK4E4E4G ttu JJ   DJJ   AA++Bc (i}|jD]8\}}i}|jD]\}}|j|||<|||<: |jj|jj |jj |d|_|D]/}tjd|jt||1 |jjy#|jjwxYw)r=Tz$Initialized '%s' store with %d itemsN) rLdecoder2r?r5clearupdater3rrArBlenrC) rall_data all_decodedr9rL items_decodedr:rErIs rinitzInMemoryFeatureStore.initls #>>+ .KD%M"[[] 7 T%)[[%6 c" 7 -K   .  ! JJ    KK    KK  { + $D  a @!++sS[\]S^O_` a JJ   DJJ   s B C55Dversionc |jj|j|}|j|}||d|kr d|d}|||<|jj y#|jj wxYw)r=NrXT)r>rX)r2r?r5r@rC)rr9r:rXrDrJs rdeletezInMemoryFeatureStore.deletesx ! JJ   ++d+K$AyAiL72 $9#$ C JJ   DJJ   rNc|j|}|d} |jj|j|}|j |}| |d|dkr*|||<t j d||j|d|jjy#|jjwxYw)r=r:NrXz Updated %s in '%s' to version %d) rPr2r?r5r@rrArBrC)rr9rE decoded_itemr:rDrJs rupsertzInMemoryFeatureStore.upserts{{4( 5k ! JJ   ++d+K$AyAiL4 ?:#/ C  >SWXaSbc JJ   DJJ   s A1B$$Cc |jj|j|jjS#|jjwxYw)r=)r2r?r3rCr s r initializedz InMemoryFeatureStore.initializedsA ! JJ   $$ JJ   DJJ   s %AAcy)Nmemoryr)rconfigs rdescribe_configurationz+InMemoryFeatureStore.describe_configurationsrN)r$r%r&r'rr strrrr@rMrWr+rZr]r-r.r_rcrrrr0r0Fs(U`!)!!xs ?S!dg!"!!( ! !c ! !!T!!rr0c@eZdZdZedZedZedZy)_FeatureStoreDataSetSorterz Implements a dependency graph ordering for data to be stored in a feature store. We must use this on every data set that will be passed to the feature store's init() method. ct}t|j}d}|j||D] }||}tj ||||<"|S)a Returns a copy of the input data that has the following guarantees: the iteration order of the outer dictionary will be in ascending order by the VersionDataKind's :priority property (if any), and for each data kind that has a "get_dependency_keys" function, the inner dictionary will have an iteration order where B is before A if A has a dependency on B. c\t|dr |jSt|jS)Npriority)hasattrrirSrB)r9s rpriority_orderzG_FeatureStoreDataSetSorter.sort_all_collections..priority_orders%tZ(}}$t~~& &r)r:)rlistkeyssortrf_sort_collection)rT outer_hashkindsrkr9rLs rsort_all_collectionsz/_FeatureStoreDataSetSorter.sort_all_collectionssj!] X]]_% '  ~ & XDTNE9JJ4QVWJt  XrcJt|dk(s t|ds|S|j}|t|dk(r|S|j}t }t|dkDr@|j D]\}}t j||||nt|dkDr@|S)Nrget_dependency_keys)rSrjrtcopyrrLrf_add_with_dependencies_first)r9input dependency_fnremaining_items items_outr:rEs rroz+_FeatureStoreDataSetSorter._sort_collections u:?'$0E"FL00  CJ!OL**,M /"Q&,224  T*GGm]lnwx /"Q& rc|jd}||=||D].}|j|}|tj||||0|||<y)Nr:)r@rfrv)rErxryrzr:dep_keydep_items rrvz7_FeatureStoreDataSetSorter._add_with_dependencies_firstsehhuo C $T* }G&**73H#*GGR_apr{| } #rN)r$r%r&r'r,rrrorvrrrrfrfsC$  rrfN)r'typingrr collectionsrrldclient.impl.rwlockrldclient.impl.utilrldclient.interfacesr r ldclient.versioned_data_kindr r r0rfrrrrsE!0."C:11ha<)>aH00r