mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
a84ebd0e51
Drop most of the functionality provided by EventManager. Add the functionality that allows saving the working directory after a period of inactivity.
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import time
|
|
from enum import StrEnum, auto
|
|
from functools import wraps
|
|
from threading import Thread
|
|
|
|
from revng.api import Manager
|
|
|
|
|
|
class EventType(StrEnum):
|
|
BEGIN = auto()
|
|
CONTEXT = auto()
|
|
|
|
|
|
def emit_event(event_type: EventType):
|
|
"""Decorator for allowing a GraphQL handler to emit an event after handling.
|
|
This is expected to wrap handler-like function where the second argument is
|
|
GraphQL's info object"""
|
|
|
|
def decorator(f):
|
|
@wraps(f)
|
|
async def inner(*args, **kwargs):
|
|
result = await f(*args, **kwargs)
|
|
context = args[1].context
|
|
if "event_manager" in context:
|
|
context["event_manager"].handle_event(event_type)
|
|
return result
|
|
|
|
return inner
|
|
|
|
return decorator
|
|
|
|
|
|
class EventManager(Thread):
|
|
def __init__(self, manager: Manager):
|
|
super().__init__()
|
|
self.manager = manager
|
|
self.running = True
|
|
self.next_save: float | None = None
|
|
|
|
def run(self):
|
|
while self.running:
|
|
# This loop will call manager.save after 5 minutes of inactivity. This is to prevent
|
|
# the manager from lagging on every change. Some changes related to speeding up the
|
|
# pipeline manager from saving (e.g. detecting dirtiness in containers) need to be
|
|
# implemented before this can be dropped
|
|
if self.next_save is not None and self.next_save < time.time():
|
|
self.manager.save()
|
|
self.next_save = None
|
|
time.sleep(10)
|
|
|
|
def handle_event(self, type_: EventType):
|
|
self.next_save = time.time() + 300
|