mirror of
https://pagure.io/fm-orchestrator.git
synced 2026-04-23 18:21:42 +08:00
4a30847bea
No implementations of MBS are using Greenwave, and there are no current plans to do so. Koji Resolver will be sufficient for any usecase dependent on gating.
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
"""
|
|
This module defines constant for events emitted by external services that work
|
|
with MBS together to complete a module build.
|
|
|
|
The event name is defined in general as much as possible, especially for the
|
|
events from Koji. Because some instance based on Koji, like Brew, might send
|
|
messages to different topics on different message bus. For example, when a
|
|
build is complete, Koji sends a message to topic buildsys.build.state.change,
|
|
however Brew sends to topic brew.build.complete, etc.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
from functools import wraps
|
|
import sched
|
|
import time
|
|
|
|
from module_build_service.common import log
|
|
|
|
|
|
KOJI_BUILD_CHANGE = "koji_build_change"
|
|
KOJI_TAG_CHANGE = "koji_tag_change"
|
|
KOJI_REPO_CHANGE = "koji_repo_change"
|
|
MBS_MODULE_STATE_CHANGE = "mbs_module_state_change"
|
|
|
|
|
|
class Scheduler(sched.scheduler):
|
|
"""
|
|
Subclass of `sched.scheduler` allowing to schedule handlers calls.
|
|
|
|
If one of the MBS handler functions need to call another handler, they need to do it in a safe
|
|
way - such another handler call should not be done in the middle of another handler's
|
|
execution.
|
|
|
|
This class provides an solution for that. Handler can schedule run of other handler using
|
|
the `add` method. The handlers should use `mbs_event_handler` decorator which ensures that
|
|
the `run` method is called at the end of handler's execution and other scheduler handlers
|
|
are executed.
|
|
"""
|
|
|
|
def add(self, handler, arguments=()):
|
|
"""
|
|
Schedule execution of `handler` with `arguments`.
|
|
"""
|
|
self.enter(0, 0, handler.delay, arguments)
|
|
|
|
def run(self):
|
|
"""
|
|
Runs scheduled handlers.
|
|
"""
|
|
log.debug("Running event scheduler with following events:")
|
|
for event in self.queue:
|
|
log.debug(" %r", event)
|
|
sched.scheduler.run(self)
|
|
|
|
def reset(self):
|
|
"""
|
|
Resets the Scheduler to initial state.
|
|
"""
|
|
while not self.empty():
|
|
self.cancel(self.queue[0])
|
|
|
|
|
|
scheduler = Scheduler(time.time, delayfunc=lambda x: x)
|
|
|
|
|
|
def mbs_event_handler(func):
|
|
"""
|
|
A decorator for MBS event handlers. It implements common tasks which should otherwise
|
|
be repeated in every MBS event handler, for example:
|
|
|
|
- at the end of handler, call events.scheduler.run().
|
|
"""
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
return func(*args, **kwargs)
|
|
finally:
|
|
scheduler.run()
|
|
# save origin function as functools.wraps from python2 doesn't preserve the signature
|
|
if not hasattr(wrapper, "__wrapped__"):
|
|
wrapper.__wrapped__ = func
|
|
return wrapper
|