mirror of
https://github.com/varun-r-mallya/Python-BPF.git
synced 2025-12-31 21:06:25 +00:00
24 lines
615 B
Python
24 lines
615 B
Python
from collections.abc import Callable
|
|
from typing import Any
|
|
|
|
|
|
class MapProcessorRegistry:
|
|
"""Registry for map processor functions"""
|
|
|
|
_processors: dict[str, Callable[..., Any]] = {}
|
|
|
|
@classmethod
|
|
def register(cls, map_type_name):
|
|
"""Decorator to register a processor function for a map type"""
|
|
|
|
def decorator(func):
|
|
cls._processors[map_type_name] = func
|
|
return func
|
|
|
|
return decorator
|
|
|
|
@classmethod
|
|
def get_processor(cls, map_type_name):
|
|
"""Get the processor function for a map type"""
|
|
return cls._processors.get(map_type_name)
|