pikers
/
piker
3
0
Fork 0
Code Issues 4 Pull Requests 6 Packages Projects Releases Wiki Activity

Drop feed refs

ordermodepps_backup
Tyler Goodlet 2021-08-10 09:32:59 -04:00
parent 2340a1666b
commit cab1cf4a00
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
piker/_cacheables.py
View File

@ -22,7 +22,13 @@ Cacheing apis and toolz.
# https://gist.github.com/njsmith/cf6fc0a97f53865f2c671659c88c1798#file-cache-py-L8 # https://gist.github.com/njsmith/cf6fc0a97f53865f2c671659c88c1798#file-cache-py-L8
from collections import OrderedDict from collections import OrderedDict
from typing import Optional, Hashable, TypeVar, AsyncContextManager from typing import (
Optional,
Hashable,
TypeVar,
AsyncContextManager,
AsyncIterable,
)
from contextlib import ( from contextlib import (
asynccontextmanager, asynccontextmanager,
AsyncExitStack, AsyncExitStack,
@ -33,7 +39,6 @@ import trio
from .brokers import get_brokermod from .brokers import get_brokermod
from .log import get_logger from .log import get_logger
from .data.feed import Feed
T = TypeVar('T') T = TypeVar('T')
@ -132,7 +137,7 @@ class cache:
''' '''
lock = trio.Lock() lock = trio.Lock()
users: int = 0 users: int = 0
ctxs: dict[tuple[str, str], Feed] = {} ctxs: dict[tuple[str, str], AsyncIterable] = {}
no_more_users: Optional[trio.Event] = None no_more_users: Optional[trio.Event] = None
@ -150,7 +155,7 @@ async def maybe_open_ctx(
''' '''
@contextmanager @contextmanager
def get_and_use() -> Feed: def get_and_use() -> AsyncIterable[T]:
# key error must bubble here # key error must bubble here
feed = cache.ctxs[key] feed = cache.ctxs[key]
log.info(f'Reusing cached feed for {key}') log.info(f'Reusing cached feed for {key}')