Edit File: plugin_systemd.cpython-36.pyc
3 ��c � @ sn d dl mZ d dlT ddlZd dl mZ ddlmZ ddlj Z ddl Z ddl Z ej j � ZG dd� dej�ZdS ) � )�base)�*� N)� exceptions)�commandsc s� e Zd ZdZ� fdd�Zdd� Zdd� Zedd � �Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zd!dd�Zdd � Zed dd �d d � �Z� ZS )"� SystemdPlugina< `systemd`:: Plug-in for tuning systemd options. + The [option]`cpu_affinity` option allows setting CPUAffinity in `/etc/systemd/system.conf`. This configures the CPU affinity for the service manager as well as the default CPU affinity for all forked off processes. The option takes a comma-separated list of CPUs with optional CPU ranges specified by the minus sign (`-`). + .Set the CPUAffinity for `systemd` to `0 1 2 3` ==== ---- [systemd] cpu_affinity=0-3 ---- ==== + NOTE: These tunings are unloaded only on profile change followed by a reboot. c s< t jjtj�s tjdtj ��tt| �j ||� t � | _ d S )NzERequired systemd '%s' configuration file not found, disabling plugin.) �os�path�isfile�consts�SYSTEMD_SYSTEM_CONF_FILEr ZNotSupportedPluginException�superr �__init__r �_cmd)�self�args�kwargs)� __class__� �$/usr/lib/python3.6/plugin_systemd.pyr $ s zSystemdPlugin.__init__c C s d|_ d|_d S )NFT)Z_has_dynamic_tuningZ_has_static_tuning)r �instancer r r �_instance_init* s z SystemdPlugin._instance_initc C s d S )Nr )r r r r r �_instance_cleanup. s z SystemdPlugin._instance_cleanupc C s dd iS )N� cpu_affinityr )�clsr r r �_get_config_options1 s z!SystemdPlugin._get_config_optionsc C sB |d k r>t jd| d |t jd�}|d k r>|jdkr>|jd�S d S )Nz^\s*z \s*=\s*(.*)$)�flagsr )�re�search� MULTILINE� lastindex�group)r �conf�key�mor r r � _get_keyval7 s zSystemdPlugin._get_keyvalc C s~ t jd| d dt|� |t jd�\}}|dk rzy|d dkrF|d7 }W n tk r\ Y nX ||d t|� d 7 }|S |S ) Nz^(\s*z\s*=).*$z\g<1>)r r � �=���)r �subn�strr � IndexError)r r" r# �valZconf_newZnsubsr r r � _add_keyval? s ( zSystemdPlugin._add_keyvalc C s t jd| d d|t jd�S )Nz^\s*z\s*=.*\n� )r )r �subr )r r" r# r r r �_del_keyK s zSystemdPlugin._del_keyc C s, | j jtjd d�}|d kr(tjd� d S |S )N)�err_retz(error reading systemd configuration file)r � read_filer r �log�error)r Zsystemd_system_confr r r �_read_systemd_system_confN s z'SystemdPlugin._read_systemd_system_confc C sp t jt j }| jj||�s8tjd� | jj|dd� dS | jj|t j�sltjdt j � | jj|dd� dS dS )Nz(error writing systemd configuration fileT)�no_errorFz/error replacing systemd configuration file '%s') r r ZTMP_FILE_SUFFIXr � write_to_filer3 r4 �unlink�rename)r r" Ztmpfiler r r �_write_systemd_system_confU s z(SystemdPlugin._write_systemd_system_confc C s t jjtj| j�S )N)r r �joinr ZPERSISTENT_STORAGE_DIR�name)r r r r �_get_storage_filenamec s z#SystemdPlugin._get_storage_filenamec C sl | j � }|d k rh| j� }| jj|d dd�}| jj|� |d krN| j|tj�}n| j|tj|�}| j |� d S )NT)r1 r6 ) r5 r= r r2 r8 r0 r �SYSTEMD_CPUAFFINITY_VARr- r: )r r" �fname�cpu_affinity_savedr r r �_remove_systemd_tuningf s z$SystemdPlugin._remove_systemd_tuningFc C s* |r&t jdtj � | j� t jd� d S )Nz6removing '%s' systemd tuning previously added by TuneDz[you may need to manualy run 'dracut -f' to update the systemd configuration in initrd image)r3 �infor r> rA Zconsole)r r Z full_rollbackr r r �_instance_unapply_staticr s z&SystemdPlugin._instance_unapply_staticc C s< |d kr dS dj dd� | jjtjddtjdd|���D ��S )Nr. � c s s | ]}t |�V qd S )N)r* )�.0�vr r r � <genexpr>| s z8SystemdPlugin._cpulist_convert_unpack.<locals>.<genexpr>z\s+�,z,\s+)r; r �cpulist_unpackr r/ )r Zcpulistr r r �_cpulist_convert_unpacky s z%SystemdPlugin._cpulist_convert_unpackr )Z per_devicec C s� d }d }| j j| jj| j j|���}djdd� | j j|�D ��}| j� } | d k rh| j| t j �}| j |�}|r|| j d|||�S |r�| j � } | j j| d dd�} |d k r�| d kr�||kr�| j j| |dd� tjdt j |t jf � | j| j| t j |�� d S ) NrD c s s | ]}t |�V qd S )N)r* )rE rF r r r rG � s z)SystemdPlugin._cmdline.<locals>.<genexpr>r T)r1 r6 )Zmakedirz setting '%s' to '%s' in the '%s')r ZunescapeZ _variables�expandZunquoter; rI r5 r% r r> rJ Z _verify_valuer= r2 r7 r3 rB r r: r- ) r Zenabling�valueZverifyZignore_missingZ conf_affinityZconf_affinity_unpackedrF Z v_unpackedr" r? r@ r r r �_cmdline~ s" zSystemdPlugin._cmdline)F)�__name__� __module__� __qualname__�__doc__r r r � classmethodr r% r- r0 r5 r: r= rA rC rJ Zcommand_customrM � __classcell__r r )r r r s r )r. r Z decoratorsZ tuned.logsZtunedr Ztuned.utils.commandsr Z tuned.constsr r r Zlogs�getr3 ZPluginr r r r r �<module> s