mirror of https://gitlab.com/litecord/litecord.git
litecord.jobs: add docstrings
This commit is contained in:
parent
69723437a6
commit
f1b8f7b0a0
|
|
@ -24,18 +24,27 @@ log = Logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class JobManager:
|
class JobManager:
|
||||||
"""Manage background jobs"""
|
"""Background job manager.
|
||||||
|
|
||||||
|
Handles closing all existing jobs when going on a shutdown. This does not
|
||||||
|
use helpers such as asyncio.gather and asyncio.Task.all_tasks. It only uses
|
||||||
|
its own internal list of jobs.
|
||||||
|
"""
|
||||||
def __init__(self, loop=None):
|
def __init__(self, loop=None):
|
||||||
self.loop = loop or asyncio.get_event_loop()
|
self.loop = loop or asyncio.get_event_loop()
|
||||||
self.jobs = []
|
self.jobs = []
|
||||||
|
|
||||||
async def _wrapper(self, coro):
|
async def _wrapper(self, coro):
|
||||||
|
"""Wrapper coroutine for other coroutines. This adds a simple
|
||||||
|
try/except for general exceptions to be logged.
|
||||||
|
"""
|
||||||
try:
|
try:
|
||||||
await coro
|
await coro
|
||||||
except Exception:
|
except Exception:
|
||||||
log.exception('Error while running job')
|
log.exception('Error while running job')
|
||||||
|
|
||||||
def spawn(self, coro):
|
def spawn(self, coro):
|
||||||
|
"""Spawn a given future or coroutine in the background."""
|
||||||
task = self.loop.create_task(
|
task = self.loop.create_task(
|
||||||
self._wrapper(coro)
|
self._wrapper(coro)
|
||||||
)
|
)
|
||||||
|
|
@ -43,5 +52,10 @@ class JobManager:
|
||||||
self.jobs.append(task)
|
self.jobs.append(task)
|
||||||
|
|
||||||
def close(self):
|
def close(self):
|
||||||
|
"""Close the job manager, cancelling all existing jobs.
|
||||||
|
|
||||||
|
It is the job's responsibility to handle the given CancelledError
|
||||||
|
and release any acquired resources.
|
||||||
|
"""
|
||||||
for job in self.jobs:
|
for job in self.jobs:
|
||||||
job.cancel()
|
job.cancel()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue