diff --git a/pythonbpf/functions/func_registry_handlers.py b/pythonbpf/functions/func_registry_handlers.py new file mode 100644 index 0000000..010e3ed --- /dev/null +++ b/pythonbpf/functions/func_registry_handlers.py @@ -0,0 +1,22 @@ +from typing import Dict + + +class StatementHandlerRegistry: + """Registry for statement handlers.""" + + _handlers: Dict = {} + + @classmethod + def register(cls, stmt_type): + """Register a handler for a specific statement type.""" + + def decorator(handler): + cls._handlers[stmt_type] = handler + return handler + + return decorator + + @classmethod + def get_handler(cls, stmt_type): + """Get the handler for a specific statement type.""" + return cls._handlers.get(stmt_type, None)