Your IP : 52.15.49.90
3
�<�e�3 � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dl j
Z
d dl m Z d dlm
Z
d dlmZ d dlZejj� ZG dd� de�ZdS ) � N)�TunedException)�InvalidProfileException)�commands)�exports)�ProfileRecommenderc @ s e Zd Zd;dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z d<dd�Z
e dd� �Z e dd� �Z
e dd� �Ze dd� �Ze dd � �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d=d9d:�Z dS )>�DaemonNc C s� t jd� tj| _ttj�| _ttj�| _ tj
| _ d| _ tj
| _|d k r�|jtjtj�| _t|jtjtj��| _t|jtjtj��| _ |jtjtj
�| _ |jtjtj�| _ |jtjtj
�| _|| _| jdkr�ttj�| _| j dkr�d| _ n| j | jk r�| j| _ | j | j | _t jd| j � | j �rJt jd� t jd| j| j | jf � t| j d�| _ || _ || _ | j � t � | _!y| j"|� W n2 t#k
�r� } zt j$d | � W Y d d }~X nX d S )
Nzinitializing daemonTr Fz$using sleep interval of %d second(s)z8dynamic tuning is enabled (can be overridden by plugins)zFusing update interval of %d second(s) (%d times of the sleep interval))Z is_hardcodedz:Cannot set initial profile. No tunings will be enabled: %s)%�log�debug�constsZCFG_DEF_DAEMON�_daemon�intZCFG_DEF_SLEEP_INTERVAL�_sleep_intervalZCFG_DEF_UPDATE_INTERVALZ_update_intervalZCFG_DEF_DYNAMIC_TUNING�_dynamic_tuningZ_recommend_commandZCFG_DEF_ROLLBACK� _rollbackZget_boolZ
CFG_DAEMON�getZCFG_SLEEP_INTERVALZCFG_UPDATE_INTERVALZCFG_DYNAMIC_TUNINGZCFG_RECOMMEND_COMMANDZCFG_DEF_RECOMMEND_COMMAND� CFG_ROLLBACK� _application�
_sleep_cycles�infor �_profile_recommender�
_unit_manager�_profile_loader�
_init_threadsr �_cmd�
_init_profiler �error)�selfZ unit_manager�profile_loader�
profile_names�configZ application�e� r! � /usr/lib/python3.6/daemon.py�__init__ sF
zDaemon.__init__c C sF d | _ tj� | _tj� | _tj� | _tj� | _| jj� tj� | _d S )N) �_thread� threadingZEvent�
_terminate�_terminate_profile_switch� _not_used�_sighup_processing�set�_profile_applied)r r! r! r" r 7 s
zDaemon._init_threadsc C s | j d� dS )zARead configuration files again and load profile according to themN)r )r r! r! r" �reload_profile_configC s z Daemon.reload_profile_configc C s� d}| j j� }|d krP| j� \}}|d krrd}|r<|d7 }n|d7 }tj|� n"|dkrr|rhtjd� n
tjd� d | _d | _g | _d | _| j |||� d S )NTz.No profile is preset, running in manual mode. z(Only post-loaded profile will be enabledzNo profile will be enabled.� )
r Zget_post_loaded_profile�_get_startup_profiler r �_profile�_manual�_active_profiles�_post_loaded_profile�set_all_profiles)r r �manual�post_loaded_profile�msgr! r! r" r G s&
zDaemon._init_profilec C s� |pd}|j � }| jr2tjd| j � || jg }x:|D ]2}|| jjj� kr8d| }| j|d|� t|��q8W y.|r�| j j
|�| _ nd | _ || _ |j � | _
W nJ tk
r� } z.ddj|�|f }| j|d|� t|��W Y d d }~X nX d S )Nr- z Using post-loaded profile '%s'z%Requested profile '%s' doesn't exist.Fz Cannot load profile(s) '%s': %s� )�splitr2 r r r Zprofile_locatorZget_known_names�_notify_profile_changedr r �loadr/ r0 r1 r �join)r r r4 Z profile_list�profile�errstrr r! r! r" �_load_profiles` s*
zDaemon._load_profilesc C s2 | j � r"d}| j|d|� t|��| j||� d S )Nz/Cannot set profile while the daemon is running.F)�
is_runningr9 r r> )r r r4 r= r! r! r" � set_profilez s zDaemon.set_profilec C s4 |s d | _ n$t|j� �dkr*d}t|��n|| _ d S )N� zYWhitespace is not allowed in profile names; only a single post-loaded profile is allowed.)r2 �lenr8 r )r � profile_namer= r! r! r" �_set_post_loaded_profile� s
z Daemon._set_post_loaded_profileFc C sV | j � r"d}| j|d|� t|��| j|� | j||� |rR| j||� | j|� d S )Nz/Cannot set profile while the daemon is running.F)r? r9 r rD r> �_save_active_profile�_save_post_loaded_profile)r Zactive_profilesr4 r5 Zsave_instantlyr= r! r! r" r3 � s
zDaemon.set_all_profilesc C s | j S )N)r/ )r r! r! r" r< � s zDaemon.profilec C s | j S )N)r0 )r r! r! r" r4 � s z
Daemon.manualc C s | j r | jS d S )N)r/ r2 )r r! r! r" r5 � s zDaemon.post_loaded_profilec C s | j S )N)r )r r! r! r" �profile_recommender� s zDaemon.profile_recommenderc C s | j S )N)r )r r! r! r" r � s zDaemon.profile_loaderc C s | j d k r tjtj|||� |S )N)r r Z send_signalr
ZSIGNAL_PROFILE_CHANGED)r r �resultr= r! r! r" r9 � s
z Daemon._notify_profile_changedc C sj | j jddgdgd�\}}|dk r&dS |d d� dkr:dS | j jddgdgd�\}}tjd |�d koh| S )
NZ systemctlzis-system-runningr )Z no_errorsF� Zstoppingz list-jobsz0\b(shutdown|reboot|halt|poweroff)\.target.*start)r Zexecute�re�search)r Zretcode�outr! r! r" �_full_rollback_required� s z Daemon._full_rollback_requiredc C s� | j d krtd��| jj| j j� | jdj| j�| j� | j | j
� | jj � | j j
� tjd| j j � | jrxtj� dj| j�}| j|dd� | jj� | j�r| j}x\| jj| j| j��s| j r�|d8 }|dkr�| j}tj d� | jj � tj d � | jj � q�W | j j� d}x.| jj| j | j� �rD|d
k �rD|d7 }�qW | j!j"� �rZt#j$}njt#j%}| j&� �svtjd � nN| j'd k�r�t#j(}tjd
t#j)t#j*f � n$| j�r�tjd� t#j$}n
tjd� | j�r�| jj+|� | jj,� d S )Nz2Cannot start the daemon without setting a profile.r7 z'static tuning from profile '%s' appliedTZOKrA r zupdating monitorszperforming tunings� z1terminating TuneD due to system shutdown / rebootZ not_on_exitzMterminating TuneD and not rolling back any changes due to '%s' option in '%s'z+terminating TuneD, rolling back all changesz"terminating TuneD in one-shot mode)-r/ r r ZcreateZunitsrE r; r1 r0 rF r2 Z start_tuningr+ r* r r �namer r �startr9 r) �clearr r �waitr& r
r r Zupdate_monitorsZ
update_tuningr( r' �is_setr
Z
ROLLBACK_FULLZ
ROLLBACK_SOFTrM r Z
ROLLBACK_NONEr ZGLOBAL_CONFIG_FILEZ stop_tuningZ destroy_all)r r Z
_sleep_cnt�iZrollbackr! r! r" � _thread_code� sX
"
zDaemon._thread_codec C sH y| j j||� W n0 tk
rB } ztjt|�� W Y d d }~X nX d S )N)r Zsave_active_profiler r r �str)r r r4 r r! r! r" rE s zDaemon._save_active_profilec C sF y| j j|� W n0 tk
r@ } ztjt|�� W Y d d }~X nX d S )N)r Zsave_post_loaded_profiler r r rV )r rC r r! r! r" rF s z Daemon._save_post_loaded_profilec C s& t jd� | jj� }t jd| � |S )NzWRunning in automatic mode, checking what profile is recommended for your configuration.zUsing '%s' profile)r r r Z recommend)r r< r! r! r" �_get_recommended_profile s
z Daemon._get_recommended_profilec C s2 | j j� \}}|d kr |d k }|s*| j� }||fS )N)r Zget_active_profilerW )r r< r4 r! r! r" r. s zDaemon._get_startup_profilec C s | j jj� S )z$Return all accessible plugin classes)r �plugins_repositoryZload_all_plugins)r r! r! r" �get_all_plugins" s zDaemon.get_all_pluginsc C s. y| j jj|�}W n tk
r& dS X |jS )z Return plugin class docstringr- )r rX � load_plugin� ImportError�__doc__)r � plugin_name� plugin_classr! r! r" �get_plugin_documentation& s
z Daemon.get_plugin_documentationc C s0 y| j jj|�}W n tk
r& i S X |j� S )z�Return plugin's parameters and their hints
Parameters:
plugin_name -- plugins name
Return:
dictionary -- {parameter_name: hint}
)r rX rZ r[ Zget_config_options_hints)r r] r^ r! r! r" �get_plugin_hints0 s
zDaemon.get_plugin_hintsc C s
| j d k S )N)r/ )r r! r! r" �
is_enabledA s zDaemon.is_enabledc C s | j d k o| j j� S )N)r$ Zis_alive)r r! r! r" r? D s zDaemon.is_runningc C s` | j � r dS | jd krdS tjd� | jj� tj| jd�| _ | j
j � | j j � | j j
� dS )NFzstarting tuning)�targetT)r? r/ r r r( r* r% ZThreadrU r$ r' rQ r& rP )r r! r! r" rP G s
z Daemon.startc C s| | j � stjd� dS | jd kr.tjd� dS | jj� sFtjd� dS | jj� tjd| jj � | j
j |�}| jj � |S )NzTuneD is not runningFzno profile is setzprofile is not appliedzverifying profile(s): %s)
r? r r r/ r+ rS r( rQ r rO r Z
verify_tuningr* )r Zignore_missing�retr! r! r" �verify_profileV s
zDaemon.verify_profilec C sB | j � s dS tjd� |r$| jj� | jj� | jj� d | _dS )NFzstopping tuningT)r? r r r' r* r& r$ r; )r Zprofile_switchr! r! r" �stopl s
z Daemon.stop)NNN)F)F) �__name__�
__module__� __qualname__r# r r, r r> r@ rD r3 �propertyr< r4 r5 rG r r9 rM rU rE rF rW r. rY r_ r` ra r? rP rd re r! r! r! r" r s8
&
G
r )�os�errnor% Z
tuned.logsZtunedZtuned.exceptionsr Ztuned.profiles.exceptionsr Z tuned.constsr
Ztuned.utils.commandsr r Z tuned.utils.profile_recommenderr rJ Zlogsr r �objectr r! r! r! r" �<module> s