3
�<�e�
� @ sd d dl mZ d dlmZ ddlmZ ddlZddlZddlZddl Z ej
j � Z G dd� dej
�ZdS )� )�base)�command_custom� )�constsNc s� e Zd ZdZ� fdd�Zdd� Zdd� Zedd � �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zeddd�dd � �Z� ZS ) �IrqbalancePlugina�
`irqbalance`::
Plug-in for irqbalance settings management. The plug-in
configures CPUs which should be skipped when rebalancing IRQs in
`/etc/sysconfig/irqbalance`. It then restarts irqbalance if and
only if it was previously running.
+
The banned/skipped CPUs are specified as a CPU list via the
[option]`banned_cpus` option.
+
.Skip CPUs 2,4 and 9-13 when rebalancing IRQs
====
----
[irqbalance]
banned_cpus=2,4,9-13
----
====
c s t t| �j||� tj� | _d S )N)�superr �__init__�perfZcpu_map�_cpus)�self�args�kwargs)� __class__� �'/usr/lib/python3.6/plugin_irqbalance.pyr s zIrqbalancePlugin.__init__c C s d|_ d|_d S )NFT)Z_has_dynamic_tuningZ_has_static_tuning)r �instancer r r �_instance_init$ s z IrqbalancePlugin._instance_initc C s d S )Nr )r r r r r �_instance_cleanup( s z"IrqbalancePlugin._instance_cleanupc C s dd iS )N� banned_cpusr )�clsr r r �_get_config_options+ s z$IrqbalancePlugin._get_config_optionsc C sl y t tjd��
}|j� S Q R X W nF tk
rf } z*|jtjkrJtjd� ntj d| � d S d }~X nX d S )N�rz>irqbalance sysconfig file is missing. Is irqbalance installed?z,Failed to read irqbalance sysconfig file: %s)
�openr �IRQBALANCE_SYSCONFIG_FILE�read�IOError�errno�ENOENT�log�warn�error)r �f�er r r �_read_irqbalance_sysconfig1 s z+IrqbalancePlugin._read_irqbalance_sysconfigc C sZ y&t tjd��}|j|� W d Q R X dS tk
rT } ztjd| � dS d }~X nX d S )N�wTz-Failed to write irqbalance sysconfig file: %sF)r r r �writer r r )r �contentr! r" r r r �_write_irqbalance_sysconfig< s z,IrqbalancePlugin._write_irqbalance_sysconfigc C s |d| S )NzIRQBALANCE_BANNED_CPUS=%s
r )r � sysconfig�banned_cpumaskr r r �_write_banned_cpusE s z#IrqbalancePlugin._write_banned_cpusc C s8 g }x(|j d�D ]}tjd|�s|j|� qW dj|�S )N�
z\s*IRQBALANCE_BANNED_CPUS=)�split�re�match�append�join)r r( �lines�liner r r �_clear_banned_cpusH s
z#IrqbalancePlugin._clear_banned_cpusc C s2 | j jdddgdgd�\}}|dkr.tjd� d S )NZ systemctlz try-restartZ
irqbalance� )Z no_errorsr z.Failed to restart irqbalance. Is it installed?)�_cmdZexecuter r )r Zretcode�outr r r �_restart_irqbalanceO s
z$IrqbalancePlugin._restart_irqbalancec C s@ | j � }|d krd S | j|�}| j||�}| j|�r<| j� d S )N)r# r3 r* r' r7 )r r) r& r r r �_set_banned_cpusX s
z!IrqbalancePlugin._set_banned_cpusc C s4 | j � }|d krd S | j|�}| j|�r0| j� d S )N)r# r3 r' r7 )r r& r r r �_restore_banned_cpusa s
z%IrqbalancePlugin._restore_banned_cpusr F)Z
per_devicec C s� d }|d k rjt | jj|��}t | j�}|j|�rB| jjt|��}n(djdd� | jD ��}tj d||f � |sr|r~|d kr~d S |r�d S |r�| j
|� n| j � d S )N�,c S s g | ] }t |��qS r )�str)�.0�xr r r �
<listcomp>r s z1IrqbalancePlugin._banned_cpus.<locals>.<listcomp>zGInvalid banned_cpus specified, '%s' does not match available cores '%s') �setr5 Zcpulist_unpackr
�issubsetZ cpulist2hex�listr0 r r r8 r9 ) r Zenabling�valueZverifyZignore_missingr) ZbannedZpresentZstr_cpusr r r � _banned_cpusi s
z IrqbalancePlugin._banned_cpus)�__name__�
__module__� __qualname__�__doc__r r r � classmethodr r# r' r* r3 r7 r8 r9 r rC �
__classcell__r r )r r r s r )� r Z
decoratorsr Ztunedr Z
tuned.logsr r r- Zlogs�getr ZPluginr r r r r �<module> s