Current File : //usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyo
�
���^c @@ s� d Z d d l m Z m Z m Z d d l m Z d d l Z y8 d d l Z d d l Z d d l
Z d d l m Z Wny e
k
r� yN d d l Z e j � d d l Z d d l Z d d l
Z d d l m Z Wq� e
k
r� d Z q� Xn Xd d l m Z d Z d Z d Z e j e � Z d a d a d a d
Z i d d 6Z d
� Z d d � Z d � Z! d � Z" d � Z# d � Z$ d d � Z% d � Z& d � Z' d S( u�
Minion data cache plugin for MySQL database.
.. versionadded:: develop
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.
The module requires the `salt_cache` database to exists 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 python-mysql
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 could 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/
i ( t absolute_importt print_functiont unicode_literals( t sleepN( t OperationalError( t SaltCacheErroru
salt_cacheu cacheg�������?u mysqlu listu lsc C@ s" t t � t d k r d n d f S( u:
Confirm that a python mysql client is installed.
u! No python mysql client installed.u N( t boolt MySQLdbt None( ( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt __virtual__U s i c C@ s� y) | j � } | j | � } | | f SWn� t t f k
r� } | d k rS � n t t � | d k ry t j d � n t j d | � t
j t � a
t t
| | d � St k
r� } t | � d k r� | d d } n t d j | | � � � n Xd S( u�
Get a cursor and run a query. Reconnect up to `retries` times if
needed.
Returns: cursor, affected rows counter
Raises: SaltCacheError, AttributeError, OperationalError
i u# mysql_cache: creating db connectionu0 mysql_cache: recreating db connection due to: %ri i� u <...>u Error running {0}: {1}N( t cursort executet AttributeErrorR R t _RECONNECT_INTERVAL_SECR t logt debugt infoR t connectt
_mysql_kwargst clientt run_queryt Exceptiont lenR t format( t connt queryt retriest curt outt e( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyR \ s"
c C@ s� d j t d t � } t t | � \ } } | j � } | j � | d d k rU d Sd j t � } t j d t � t t | � \ } } | j � d S( u
Create table if needed
uq SELECT COUNT(TABLE_NAME) FROM information_schema.tables
WHERE table_schema = '{0}' AND table_name = '{1}'u dbi i Nu� CREATE TABLE IF NOT EXISTS {0} (
bank CHAR(255),
etcd_key CHAR(255),
data MEDIUMBLOB,
PRIMARY KEY(bank, etcd_key)
);u mysql_cache: creating table %s( R R t _table_nameR R t fetchonet closeR R ( R R t _t r( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt
_create_tabley s
c C@ s t d k r d Si t j d d � d 6t j d d � d 6t j d d � d 6t j d t � d
6t j d d � d
6t j d d � d 6t j d d � d 6t d 6a t j d t � a x6 t j � D]( \ } } | d k r� t j | � q� q� Wt j
� } d | d <t j d | � t
� d S( u5 Initialize connection and create table if needed
Nu
mysql.hostu 127.0.0.1u hostu
mysql.useru useru mysql.passwordu passwdu mysql.databaseu dbu
mysql.porti� u portu mysql.unix_socketu unix_socketu mysql.connect_timeoutu connect_timeoutu
autocommitu mysql.table_nameu <hidden>u. mysql_cache: Setting up client with params: %r( R R t __opts__t gett _DEFAULT_DATABASE_NAMEt TrueR R t itemst popt copyR R R# ( t kt vt kwargs_copy( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt _init_client� s&
c C@ s t � t d j | � } d j t | | | � } t t | � \ } } | j � | d k r{ t d j | | | � � � n d S( u
Store a key value.
u serialsC REPLACE INTO {0} (bank, etcd_key, data) values('{1}', '{2}', '{3}')i i u" Error storing {0} {1} returned {2}N( i i ( R. t __context__t dumpsR R R R R R ( t bankt keyt dataR R t cnt( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt store� s
c C@ sl t � d j t | | � } t t | � \ } } | j � } | j � | d k rW i St d j | d � S( u
Fetch a key value.
u8 SELECT data FROM {0} WHERE bank='{1}' AND etcd_key='{2}'u seriali N(
R. R R R R R R R R/ t loads( R1 R2 R R R! R" ( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt fetch� s
c C@ s^ t � d j t | � } | d k r; | d j | � 7} n t t | � \ } } | j � d S( uF
Remove the key from the cache bank with all the key content.
u DELETE FROM {0} WHERE bank='{1}'u AND etcd_key='{0}'N( R. R R R R R R ( R1 R2 R R R! ( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt flush� s c C@ s_ t � d j t | � } t t | � \ } } g | j � D] } | d ^ q; } | j � | S( u\
Return an iterable object containing all entries stored in the specified
bank.
u) SELECT etcd_key FROM {0} WHERE bank='{1}'i ( R. R R R R t fetchallR ( R1 R R R! t rowR ( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt ls� s #
c C@ sU t � d j t | | � } t t | � \ } } | j � } | j � | d d k S( uB
Checks if the specified bank contains the specified key.
u? SELECT COUNT(data) FROM {0} WHERE bank='{1}' AND etcd_key='{2}'i i ( R. R R R R R R ( R1 R2 R R R! R" ( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt contains� s
(( t __doc__t
__future__R R R t timeR t loggingR t MySQLdb.cursorst MySQLdb.converterst MySQLdb.connectionsR t ImportErrort pymysqlt install_as_MySQLdbt MySQLdb.errR t salt.exceptionsR R& t _DEFAULT_CACHE_TABLE_NAMER
t getLoggert __name__R R R R t __virtualname__t __func_alias__R R R# R. R5 R7 R8 R; R< ( ( ( s: /usr/lib/python2.7/site-packages/salt/cache/mysql_cache.pyt <module>- sJ
Mini Shell