Edit File: plugin_irqbalance.cpython-36.pyc
3 ��c� � @ 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