mirror/vincentbernat.i3wm-configuration
1
0
Fork 0
mirror of https://github.com/vincentbernat/i3wm-configuration.git synced 2025-07-13 11:44:21 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

i3-companion: add back ability to trigger urgent work despite of damping

Browse source
This commit is contained in:
Vincent Bernat 2021-07-13 00:12:54 +02:00
parent f56b995d5a
commit b62f747837
1 changed files with 24 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

32
bin/i3-companion
View file

@ -99,18 +99,23 @@ def on(*events):
def dampen(sleep, *, unless=None, retry=0): def dampen(sleep, *, unless=None, retry=0):
"""Dampen a function call. Optional retry on failure. Ensure only one """Dampen a function call. Optional immediate execution. Optional
instance is executed. It is assumed the arguments provided to the retry on failure. Ensure only one instance is executed. It is
dampened function have no effect on its execution. assumed the arguments provided to the dampened function have no
effect on its execution.
""" """
def decorator(fn): def decorator(fn):
async def worker(): async def worker(urgent):
while True: while True:
await asyncio.sleep(sleep) try:
# Wait for an urgent work or until sleep is elapsed
await asyncio.wait_for(urgent.wait(), timeout=sleep)
except asyncio.TimeoutError:
pass
retry, args, kwargs = fn.queue retry, args, kwargs = fn.queue
fn.queue = None fn.queue = None
urgent.clear()
# Execute the work # Execute the work
logger.debug(f"execute work for {fn}") logger.debug(f"execute work for {fn}")
@ -129,8 +134,14 @@ def dampen(sleep, *, unless=None, retry=0):
# Retry, unless we have something already scheduled # Retry, unless we have something already scheduled
if fn.queue is not None: if fn.queue is not None:
return logger.debug(f"retry now with queued event for {fn}")
urgent.set()
continue
logger.debug(f"reschedule retry for {fn}")
fn.queue = (retry, args, kwargs) fn.queue = (retry, args, kwargs)
if unless is not None and unless(*args, **kwargs):
logger.debug(f"wake up now for retry of {fn}")
urgent.set()
# Do we still have something to do? # Do we still have something to do?
if fn.queue is None: if fn.queue is None:
break break
@ -144,9 +155,14 @@ def dampen(sleep, *, unless=None, retry=0):
fn.queue = (retry, args, kwargs) fn.queue = (retry, args, kwargs)
if fn.worker is None: if fn.worker is None:
logger.debug(f"create new worker for {fn}") logger.debug(f"create new worker for {fn}")
fn.worker = asyncio.create_task(worker()) urgent = asyncio.Event()
fn.worker = asyncio.create_task(worker(urgent))
fn.worker.urgent = urgent
else: else:
logger.debug(f"enqueue new work for {fn}") logger.debug(f"enqueue new work for {fn}")
if unless is not None and unless(*args, **kwargs):
logger.debug(f"wake up now for {fn}")
fn.worker.urgent.set()
fn.worker = None fn.worker = None
return wrapper return wrapper