from . import base
import tuned.consts as consts
import tuned.logs
log = tuned.logs.get()
class Plugin(base.Plugin):
"""
Base class for plugins with device hotpluging support.
"""
def __init__(self, *args, **kwargs):
super(Plugin, self).__init__(*args, **kwargs)
def cleanup(self):
super(Plugin, self).cleanup()
self._hardware_events_cleanup()
def _hardware_events_init(self):
raise NotImplementedError()
def _hardware_events_cleanup(self):
raise NotImplementedError()
def _init_devices(self):
self._hardware_events_init()
def _hardware_events_callback(self, event, device):
if event == "add":
log.info("device '%s' added" % device.sys_name)
self._add_device(device)
elif event == "remove":
log.info("device '%s' removed" % device.sys_name)
self._remove_device(device)
def _add_device(self, device):
device_name = device.sys_name
if device_name in (self._assigned_devices | self._free_devices):
return
for instance_name, instance in list(self._instances.items()):
if len(self._get_matching_devices(instance, [device_name])) == 1:
log.info("instance %s: adding new device %s" % (instance_name, device_name))
self._assigned_devices.add(device_name)
self._call_device_script(instance, instance.script_pre, "apply", [device_name])
self._added_device_apply_tuning(instance, device_name)
self._call_device_script(instance, instance.script_post, "apply", [device_name])
instance.processed_devices.add(device_name)
break
else:
log.debug("no instance wants %s" % device_name)
self._free_devices.add(device_name)
def _remove_device(self, device):
device_name = device.sys_name
if device_name not in (self._assigned_devices | self._free_devices):
return
for instance in list(self._instances.values()):
if device_name in instance.processed_devices:
self._call_device_script(instance, instance.script_post, "unapply", [device_name])
self._removed_device_unapply_tuning(instance, device_name)
self._call_device_script(instance, instance.script_pre, "unapply", [device_name])
instance.processed_devices.remove(device_name)
# This can be a bit racy (we can overcount),
# but it shouldn't affect the boolean result
instance.active = len(instance.processed_devices) \
+ len(instance.assigned_devices) > 0
self._assigned_devices.remove(device_name)
break
else:
self._free_devices.remove(device_name)
def _added_device_apply_tuning(self, instance, device_name):
self._execute_all_device_commands(instance, [device_name])
if instance.has_dynamic_tuning and self._global_cfg.get(consts.CFG_DYNAMIC_TUNING, consts.CFG_DEF_DYNAMIC_TUNING):
self._instance_apply_dynamic(instance, device_name)
def _removed_device_unapply_tuning(self, instance, device_name):
if instance.has_dynamic_tuning and self._global_cfg.get(consts.CFG_DYNAMIC_TUNING, consts.CFG_DEF_DYNAMIC_TUNING):
self._instance_unapply_dynamic(instance, device_name)
self._cleanup_all_device_commands(instance, [device_name])