File: //usr/lib/python3.6/site-packages/glances/exports/__pycache__/glances_kafka.cpython-36.opt-1.pyc
3
Eo�c) � @ sP d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z G dd� de�Z
dS ) zKafka interface class.� N)�logger)�
GlancesExport)�
json_dumps)�
KafkaProducerc s> e Zd ZdZd � fdd� Zdd� Zdd� Z� fd d
�Z� ZS ) �Exportz+This class manages the Kafka export module.Nc s^ t t| �j||d� d| _d| _d| _| jddddgddgd �| _| jsPtd
� | j � | _
dS ) zInit the Kafka export IF.)�config�argsN�kafka�host�port�topic� compression�tags)Z mandatoriesZoptionszMissing KAFKA config) �superr �__init__r r
r Z load_conf�
export_enable�exit�init�client)�selfr r )� __class__� �#/usr/lib/python3.6/glances_kafka.pyr s zExport.__init__c C s� | j s
dS dj| j| j�}yt|dd� | jd�}W n> tk
rp } z"tjd||f � t j
d� W Y dd}~X nX tj d| � |S ) z(Init the connection to the Kafka server.Nz{}:{}c S s t | �jd�S )Nzutf-8)r �encode)�vr r r �<lambda>9 s z Export.init.<locals>.<lambda>)Zbootstrap_serversZvalue_serializerZcompression_typez&Cannot connect to Kafka server %s (%s)� z Connected to the Kafka server %s) r �formatr
r r r
� Exceptionr Zcritical�sysr �info)r Z
server_uri�s�er r r r . s z Export.initc C s� t jdj|�� tt||��}| jdk r:|j| j| j�� y | jj | j
|j d�|d� W n4 t k
r� } zt j
dj||�� W Y dd}~X nX dS )z%Write the points to the kafka server.zExport {} stats to KafkaNzutf-8)�key�valuez$Cannot export {} stats to Kafka ({}))r �debugr �dict�zipr �updateZ
parse_tagsr �sendr r r �error)r �name�columnsZpoints�datar" r r r �exportD s
z
Export.exportc s& | j j� | j j� tt| �j� dS )z Close the Kafka export module.N)r �flush�closer r r )r )r r r r Z s
z Export.exit)NN) �__name__�
__module__� __qualname__�__doc__r r r. r �
__classcell__r r )r r r s
r ) r4 r Zglances.loggerr Z glances.exports.glances_exportr Zglances.globalsr r r r r r r r �<module>
s