File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/cache/__pycache__/mysql_cache.cpython-310.pyc
Back
o ;j�' � @ sF d Z ddlZddlZddlZddlZddlZddlmZ zddl Z ddl Z ddl Z ddl m Z W n1 eya z ddlZe�� ddl Z ddl Z ddl Z ddlm Z W n ey^ dZ Y nw Y nw dZdZdZe�e�ZdZd d iZd d � Zd d� Zd!dd�Z dd� Z dd� Z dd� Z dd� Z d"dd�Z!d d � Z"d d � Z#d d � Z$dS )#a[ Minion data cache plugin for MySQL database. .. versionadded:: 2018.3.0 It is up to the system administrator to set up and configure the MySQL infrastructure. All is needed for this plugin is a working MySQL server. .. warning:: The mysql.database and mysql.table_name will be directly added into certain queries. Salt treats these as trusted input. The module requires the database (default ``salt_cache``) to exist but creates its own table if needed. The keys are indexed using the ``bank`` and ``etcd_key`` columns. To enable this cache plugin, the master will need the python client for MySQL installed. This can be easily installed with pip: .. code-block:: bash pip install pymysql Optionally, depending on the MySQL agent configuration, the following values could be set in the master config. These are the defaults: .. code-block:: yaml mysql.host: 127.0.0.1 mysql.port: 2379 mysql.user: None mysql.password: None mysql.database: salt_cache mysql.table_name: cache Related docs can be found in the `python-mysql documentation`_. To use the mysql as a minion data cache backend, set the master ``cache`` config value to ``mysql``: .. code-block:: yaml cache: mysql .. _`MySQL documentation`: https://github.com/coreos/mysql .. _`python-mysql documentation`: http://python-mysql.readthedocs.io/en/latest/ � N)�SaltCacheError)�OperationalErrorZ salt_cache�cacheg�������?Zmysql�ls�listc C s t t�tdu r dfS dfS )z: Confirm that a python mysql client is installed. Nz!No python mysql client installed.� )�bool�MySQLdb� r r �J/opt/saltstack/salt/lib/python3.10/site-packages/salt/cache/mysql_cache.py� __virtual__\ s r c C s t �dd� dS )zi Force a reconnection to the MySQL database, by removing the client from Salt's __context__. � mysql_clientN)� __context__�popr r r r �force_reconnectc s r � c C sF | du r t �d�} z&| �� }|s t�d|� |�|�}nt�d|t|�� |�||�}||fW S ttfyv } z9|dkr>� t � t � | du rMt�d� nt� d|� t jdi t d ��t d< tt �d�|||d d �W Y d}~S d}~w ty� } z!t|�d kr�|dd � d }td �||r�d|� �|���d|���d}~ww )z� Get a cursor and run a query. Reconnect up to ``retries`` times if needed. Returns: cursor, affected rows counter Raises: SaltCacheError, AttributeError, OperationalError Nr zDoing query: %szDoing query: %s args: %s r z#mysql_cache: creating db connectionz0mysql_cache: recreating db connection due to: %r� mysql_kwargs� )�conn�query�args�retries� z<...>zError running {}{}: {}z- args: r r )r �getZcursor�log�debugZexecute�repr�AttributeErrorr �time�sleep�_RECONNECT_INTERVAL_SEC�infor �connect� run_query� Exception�lenr �format)r r r r �cur�out�er r r r# k sF �� ����r# c C s d} t t�d�| td d td fd�\}}|�� }|�� |d dkrdd } t td | td d td fd�\}}|�� }|�� |d dkrId S d �td d td �} t td | �\}}|�� d S d �td �} t�d td � t t�d�| �\}}|�� d S )z Create table if needed zkSELECT COUNT(TABLE_NAME) FROM information_schema.tables WHERE table_schema = %s AND table_name = %sr r �db�mysql_table_name�r r r z� SELECT COUNT(TABLE_NAME) FROM information_schema.columns WHERE table_schema = %s AND table_name = %s AND column_name = 'last_update' Nz� ALTER TABLE {}.{} ADD COLUMN last_update TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP a CREATE TABLE IF NOT EXISTS {} ( bank CHAR(255), etcd_key CHAR(255), data MEDIUMBLOB, last_update TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, PRIMARY KEY(bank, etcd_key) );z mysql_cache: creating table %s)r# r r �fetchone�closer& r r! )r r' �_�rr r r � _create_table� sF � � ���� r1 c C s t �d�dur dS t�t�} d| �dd�| �dd�| �dd�| �dt�| �d d �| �d d�| �d d�d �}d|d< | �dd�t d< | D ]}|�d�rX|�d�d }| �|�||< qCt�|�� � D ] \}}|du rm|�|� q`|�� }d|d< t � d|� |t d< t � dS )z0Initialize connection and create table if neededr NTz mysql.hostz 127.0.0.1z mysql.userzmysql.passwordzmysql.databasez mysql.porti� zmysql.unix_socketzmysql.connect_timeout)� autocommit�host�user�passwdr* �portZ unix_socketZconnect_timeoutr2 zmysql.table_name�saltr+ zmysql.�.r z<hidden>r5 z.mysql_cache: Setting up client with params: %rr ) r r �copy�deepcopyZ__opts__r �_DEFAULT_DATABASE_NAME� startswith�split�itemsr r! r1 )�optsr �kZ_key�vZ kwargs_copyr r r � _init_client� s8 � � � rB c C sp t � tj�|�}d�td �}| ||f}tt�d�||d�\}}|�� |dvr6t d| � d|� d|� ���d S ) z Store a key value. z7REPLACE INTO {} (bank, etcd_key, data) values(%s,%s,%s)r+ r r, )r � zError storing � z returned N) rB r7 �payload�dumpsr& r r# r r. r )�bank�key�datar r r' Zcntr r r �store� s � �rJ c C s\ t � d�td �}tt�d�|| |fd�\}}|�� }|�� |du r&i S tj� |d �S )z Fetch a key value. z1SELECT data FROM {} WHERE bank=%s AND etcd_key=%sr+ r r, Nr ) rB r& r r# r r- r. r7 rE �loads)rG rH r r' r/ r0 r r r �fetch s � rL c C sV t � d�td �}|du r| f}n| |f}|d7 }ttd ||d�\}}|�� dS )zF Remove the key from the cache bank with all the key content. z DELETE FROM {} WHERE bank=%sr+ Nz AND etcd_key=%sr r, )rB r& r r# r. )rG rH r rI r' r/ r r r �flush s rM c C sL t � d�td �}tt�d�|| fd�\}}dd� |�� D �}|�� |S )z\ Return an iterable object containing all entries stored in the specified bank. z%SELECT etcd_key FROM {} WHERE bank=%sr+ r r, c S s g | ]}|d �qS )r r )�.0�rowr r r � <listcomp>8 s zls.<locals>.<listcomp>)rB r& r r# r Zfetchallr. )rG r r' r/ r( r r r r . s �c C sn t � |du r| f}d�td �}n | |f}d�td �}tt�d�||d�\}}|�� }|�� |d dkS ) zB Checks if the specified bank contains the specified key. Nz(SELECT COUNT(data) FROM {} WHERE bank=%sr+ z8SELECT COUNT(data) FROM {} WHERE bank=%s AND etcd_key=%sr r, r r )rB r& r r# r r- r. )rG rH rI r r' r/ r0 r r r �contains= s �� rQ c C sV t � d�td �}| |f}ttd ||d�\}}|�� }|�� |r)t|d �S |S )z[ Return the integer Unix epoch update timestamp of the specified bank and key. zHSELECT UNIX_TIMESTAMP(last_update) FROM {} WHERE bank=%s AND etcd_key=%sr+ r )r r r )rB r& r r# r- r. �int)rG rH r rI r' r/ r0 r r r �updatedR s �rS )Nr )N)%�__doc__r9 �loggingr Z salt.payloadr7 Zsalt.utils.stringutilsZsalt.exceptionsr r ZMySQLdb.convertersZMySQLdb.cursorsZMySQLdb.connectionsr � ImportErrorZpymysqlZinstall_as_MySQLdbZ MySQLdb.errr; Z_DEFAULT_CACHE_TABLE_NAMEr � getLogger�__name__r Z__virtualname__Z__func_alias__r r r# r1 rB rJ rL rM r rQ rS r r r r �<module> sT 3 ��� ,?'
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings