Vasanko/cista-storage
Vasanko/cista-storage
2
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Watching code cleanup

Browse Source
This commit is contained in:
Leo Vasanko 2023-11-12 19:58:37 +00:00
parent 0be72827db
commit 540e825cc3
2 changed files with 31 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cista/api.py
View File

@ -107,8 +107,8 @@ async def watch(req, ws):
with watching.state.lock:
q = watching.pubsub[uuid] = asyncio.Queue()
# Init with disk usage and full tree
await ws.send(watching.format_du(watching.state.space))
await ws.send(watching.format_tree(watching.state.root))
await ws.send(watching.format_space(watching.state.space))
await ws.send(watching.format_root(watching.state.root))
# Send updates
while True:
await ws.send(await q.get())

40
cista/watching.py
View File

@ -86,7 +86,6 @@ class State:
def __getitem__(self, index: PurePosixPath | tuple[PurePosixPath, int]):
with self.lock:
print(self._slice(index))
return self._listing[self._slice(index)]
def __setitem__(
@ -153,8 +152,7 @@ def watcher_thread(loop):
if old != new:
with state.lock:
state.root = new
msg = format_tree(new)
asyncio.run_coroutine_threadsafe(broadcast(msg), loop).result()
broadcast(format_root(new), loop)
# The watching is not entirely reliable, so do a full refresh every minute
refreshdl = time.monotonic() + 60.0
@ -167,9 +165,7 @@ def watcher_thread(loop):
space = Space(*du, storage=state.root[0].size)
if space != state.space:
state.space = space
asyncio.run_coroutine_threadsafe(
broadcast(format_du(space)), loop
).result()
broadcast(format_space(space), loop)
break
# Do a full refresh?
if time.monotonic() > refreshdl:
@ -197,27 +193,20 @@ def watcher_thread_poll(loop):
old = state.root
new = walk()
if old != new:
with state.lock:
state.root = new
asyncio.run_coroutine_threadsafe(broadcast(format_tree(new)), loop).result()
broadcast(format_update(old, new), loop)
# Disk usage update
du = shutil.disk_usage(rootpath)
space = Space(*du, storage=state.root[0].size)
if space != state.space:
state.space = space
asyncio.run_coroutine_threadsafe(broadcast(format_du(space)), loop).result()
broadcast(format_space(space), loop)
time.sleep(2.0)
def format_du(usage):
return msgspec.json.encode({"space": usage}).decode()
def format_tree(root):
return msgspec.json.encode({"root": root}).decode()
def walk(rel=PurePosixPath()) -> list[FileEntry]: # noqa: B008
path = rootpath / rel
try:
@ -265,6 +254,7 @@ def update(relpath: PurePosixPath, loop):
if rootpath is None or relpath is None:
print("ERROR", rootpath, relpath)
new = walk(relpath)
with state.lock:
old = state[relpath]
if old == new:
return
@ -273,9 +263,7 @@ def update(relpath: PurePosixPath, loop):
state[relpath, new[0].isfile] = new
else:
del state[relpath]
# FIXME: broadcast format_update()
msg = format_update(old, state.root)
asyncio.run_coroutine_threadsafe(broadcast(msg), loop).result()
broadcast(format_update(old, state.root), loop)
def format_update(old, new):
@ -322,7 +310,19 @@ def format_update(old, new):
return msgspec.json.encode({"update": update}).decode()
async def broadcast(msg):
def format_space(usage):
return msgspec.json.encode({"space": usage}).decode()
def format_root(root):
return msgspec.json.encode({"root": root}).decode()
def broadcast(msg, loop):
return asyncio.run_coroutine_threadsafe(abroadcast(msg), loop).result()
async def abroadcast(msg):
try:
for queue in pubsub.values():
queue.put_nowait(msg)