Move HelperHandlerRegistry to helper_registry.py

This commit is contained in:
Pragyansh Chaturvedi
2025-10-13 18:21:50 +05:30
parent 5d0a888542
commit 74d8014ade
4 changed files with 31 additions and 29 deletions

View File

@ -0,0 +1,27 @@
from typing import Callable
class HelperHandlerRegistry:
"""Registry for BPF helpers"""
_handlers: dict[str, Callable] = {}
@classmethod
def register(cls, helper_name):
"""Decorator to register a handler function for a helper"""
def decorator(func):
cls._handlers[helper_name] = func
return func
return decorator
@classmethod
def get_handler(cls, helper_name):
"""Get the handler function for a helper"""
return cls._handlers.get(helper_name)
@classmethod
def has_handler(cls, helper_name):
"""Check if a handler function is registered for a helper"""
return helper_name in cls._handlers