Google
Edit File: plugin_systemd.cpython-39.pyc
a �]�iG � @ sp d dl mZ d dlT ddlZd dl mZ ddlmZ ddlm Z ddl Z ddl Z ej � � 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� Zejfdd�Zdd� Zed d d �d d � �Z� ZS )!� SystemdPlugina' 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 j�tj�s t�dtj ��tt| �j |i |�� 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.9/site-packages/tuned/plugins/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 �_instance_cleanup, s z SystemdPlugin._instance_cleanupc C s dd iS )N� cpu_affinityr )�clsr r r �_get_config_options/ s �z!SystemdPlugin._get_config_optionsc C sB |d ur>t jd| d |t jd�}|d ur>|jdkr>|�d�S d S )N�^\s*z \s*=\s*(.*)$��flagsr )�re�search� MULTILINE� lastindex�group)r �conf�key�mor r r � _get_keyval5 s zSystemdPlugin._get_keyvalc C s| t jd| d dt|� |t jd�\}}|dk rxz|d dkrF|d7 }W n tyZ Y n0 ||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 )Nr z\s*=.*\n� r )r" �subr$ )r r' r( r r r �_del_keyI s zSystemdPlugin._del_keyc C s, | j jtjd d�}|d u r(t�d� 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_confL s z'SystemdPlugin._read_systemd_system_confc C sp t jt j }| j�||�s8t�d� | jj|dd� dS | j�|t j�slt�dt 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_filer8 r9 �unlink�rename)r r' Ztmpfiler r r �_write_systemd_system_confS s z(SystemdPlugin._write_systemd_system_confc C s t j�tj| j�S r )r r �joinr ZPERSISTENT_STORAGE_DIR�name)r r r r �_get_storage_filenamea s z#SystemdPlugin._get_storage_filenamec C sl | � � }|d urh| �� }| jj|d dd�}| j�|� |d u rN| �|tj�}n| �|tj|�}| � |� d S )NT�r6 r; ) r: rB r r7 r= r5 r �SYSTEMD_CPUAFFINITY_VARr2 r? )r r' �fname�cpu_affinity_savedr r r �_remove_systemd_tuningd s z$SystemdPlugin._remove_systemd_tuningc C s0 |t jkr,t�dt j � | �� t�d� 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)r Z ROLLBACK_FULLr8 �inforD rG Zconsole)r r Zrollbackr r r �_instance_unapply_staticp s z&SystemdPlugin._instance_unapply_staticc C s< |d u r dS d� dd� | j�t�ddt�dd|���D ��S )Nr3 � c s s | ]}t |�V qd S r �r/ ��.0�vr r r � <genexpr>z � z8SystemdPlugin._cpulist_convert_unpack.<locals>.<genexpr>z\s+�,z,\s+)r@ r �cpulist_unpackr" r4 )r Zcpulistr r r �_cpulist_convert_unpackw s z%SystemdPlugin._cpulist_convert_unpackr F)Z per_devicec C s� d }d }| j �| j�| j �|���}d�dd� | j �|�D ��} | �� } | d urh| �| t j �}| � |�}|r|| � d| ||�S |r�| � � } | j j| d dd�} |d ur�| d u r�| |kr�| j j| |dd� t�dt j | t jf � | �| �| t j | �� dS d S ) NrJ c s s | ]}t |�V qd S r rK rL r r r rO � rP z)SystemdPlugin._cmdline.<locals>.<genexpr>r TrC )Zmakedirz setting '%s' to '%s' in the '%s')r ZunescapeZ _variables�expand�unquoter@ rR r: r* r rD rS Z _verify_valuerB r7 r< r8 rH r r? r2 ) r Zenabling�valueZverifyZignore_missingr Z conf_affinityZconf_affinity_unpackedrN Z v_unpackedr' rE rF r r r �_cmdline| s& zSystemdPlugin._cmdline)�__name__� __module__� __qualname__�__doc__r r r � classmethodr r* r2 r5 r: r? rB rG r Z ROLLBACK_SOFTrI rS Zcommand_customrW � __classcell__r r r r r s" r )r3 r Z decoratorsZ tuned.logsZtunedr Ztuned.utils.commandsr Z tuned.constsr r r" Zlogs�getr8 ZPluginr r r r r �<module> s