File: //lib/python3.6/site-packages/glances/exports/__pycache__/glances_cassandra.cpython-36.opt-1.pyc
3
Eo�c@ � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dd l
mZ dd
lmZ G d d � d e�ZdS )
z!Cassandra/Scylla interface class.� N)�datetime)�Number)�logger)�
GlancesExport)�PlainTextAuthProvider)�Cluster)�uuid_from_time)�InvalidRequestc s> e Zd ZdZd � fdd� Zdd� Zdd� Z� fd d
�Z� ZS ) �Exportz6This class manages the Cassandra/Scylla export module.Nc s~ t t| �j||d� d| _d| _d| _d| _d| _d| _| j ddddgd d
d d d
gd�| _
| j
sjt j d� | j
� \| _| _dS )z Init the Cassandra export IF.)�config�argsN� � � cassandra�host�port�keyspace�protocol_version�replication_factor�table�username�password)Z mandatoriesZoptions)�superr
�__init__r r r r r r Z load_conf�
export_enable�sys�exit�init�cluster�session)�selfr r )� __class__� �'/usr/lib/python3.6/glances_cassandra.pyr s
zExport.__init__c ! C sN | j s
dS t| j| jd�}y,t| jgt| j�t| j�|d�}|j � }W nD t
k
r� } z(t j d| j| j|f � t
jd� W Y dd}~X nX y|j| j� W nL tk
r� t jdj| j�� d| j| jf }|j|� |j| j� Y nX t jdj|jj|jj� | j�� y|jd | j � W n& t
k
�rD t jd
| j � Y nX ||fS ) z,Init the connection to the Cassandra server.N)r r )r r �
auth_providerz0Cannot connect to Cassandra cluster '%s:%s' (%s)r z+Create keyspace {} on the Cassandra clusterz_CREATE KEYSPACE %s WITH replication = { 'class': 'SimpleStrategy', 'replication_factor': '%s' }zBStats will be exported to Cassandra cluster {} ({}) in keyspace {}z�CREATE TABLE %s (plugin text, time timeuuid, stat map<text,float>, PRIMARY KEY (plugin, time)) WITH CLUSTERING ORDER BY (time DESC)z Cassandra table %s already exist)r r r r r r �intr r Zconnect� Exceptionr Zcriticalr r Z set_keyspacer r �info�formatr �executeZmetadataZ cluster_nameZ all_hostsr �debug)r r$ r r �e�cr" r" r# r 7 s@
z Export.initc C s� t jdj|�� dd� tt||��j� D �}y8dj| j�}| jj|�}| jj ||t
t j � �|f� W n4 t
k
r� } zt jdj||�� W Y dd}~X nX dS )z*Write the points to the Cassandra cluster.z Export {} stats to Cassandrac S s$ i | ] \}}t |t�rt|�|�qS r" )�
isinstancer �float)�.0�k�vr" r" r# �
<dictcomp>n s z!Export.export.<locals>.<dictcomp>z4INSERT INTO {} (plugin, time, stat) VALUES (?, ?, ?)z(Cannot export {} stats to Cassandra ({})N)r r* r( �dict�zipZ iteritemsr r Zpreparer) r r Znowr& �error)r �name�columnsZpoints�dataZstmtZqueryr+ r" r" r# �exporti s z
Export.exportc s& | j j� | jj� tt| �j� dS )z"Close the Cassandra export module.N)r Zshutdownr r r
r )r )r! r" r# r y s
z Export.exit)NN) �__name__�
__module__� __qualname__�__doc__r r r9 r �
__classcell__r" r" )r! r# r
s
2r
)r= r r Znumbersr Zglances.loggerr Z glances.exports.glances_exportr Zcassandra.authr Zcassandra.clusterr Zcassandra.utilr r r r
r" r" r" r# �<module>
s