wehub-resource-sync
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import abc
|
|
from typing import Callable
|
|
|
|
from ray.core.generated.events_base_event_pb2 import RayEvent
|
|
|
|
|
|
class PlatformEventProvider(abc.ABC):
|
|
"""
|
|
Base interface for all platform-specific event providers (e.g. Kubernetes).
|
|
"""
|
|
|
|
def __init__(self, callback: Callable[[RayEvent], None]):
|
|
"""
|
|
Initialize the provider.
|
|
Implementations of run() may invoke the callback from background worker
|
|
threads, so consumers are responsible for passing a callback that is
|
|
fully thread-safe.
|
|
|
|
Args:
|
|
callback: A callback function to deliver new or updated RayEvent
|
|
protobuf objects to the central manager.
|
|
"""
|
|
self._callback = callback
|
|
|
|
@abc.abstractmethod
|
|
async def run(self) -> None:
|
|
"""
|
|
Start the provider. This method should run the event watch/poll loop.
|
|
"""
|
|
pass
|
|
|
|
@abc.abstractmethod
|
|
async def cleanup(self) -> None:
|
|
"""
|
|
Stop the provider and release all resources gracefully.
|
|
"""
|
|
pass
|