o va@sdZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZdd lmZGd d d eZdS)z!Cassandra/Scylla interface class.N)datetime)Number)logger) GlancesExport) iteritems)PlainTextAuthProvider)Cluster)uuid_from_time)InvalidRequestcs>eZdZdZd fdd ZddZddZfd d ZZS) Exportz6This class manages the Cassandra/Scylla export module.Ncsvtt|j||dd|_d|_d|_d|_d|_d|_|j dgdgdd|_ |j s1t d| \|_|_dS) zInit the Cassandra export IF.)configargsN cassandra)hostportkeyspace)protocol_versionreplication_factortableusernamepassword)Z mandatoriesoptions)superr __init__rrrrrrZ load_conf export_enablesysexitinitclustersession)selfr r  __class__C/usr/lib/python3/dist-packages/glances/exports/glances_cassandra.pyr(s zExport.__init__c Cs`|jsdSt|j|jd}zt|jgt|jt|j|d}| }Wn#t yF}zt d|j|j|ft dWYd}~nd}~wwz||jWn.ty}}z"t d|jd|j|jf}||||jWYd}~nd}~wwt d|jj|j|jz |d |jW||fSt yt d |jY||fSw) z,Init the connection to the Cassandra server.N)rr)rr auth_providerz0Cannot connect to Cassandra cluster '%s:%s' (%s)rz+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 {}zCREATE TABLE %s (plugin text, time timeuuid, stat map, PRIMARY KEY (plugin, time)) WITH CLUSTERING ORDER BY (time DESC)z Cassandra table %s already exist)rrrrrrintrrZconnect ExceptionrZcriticalrrZ set_keyspacerr infoformatrexecuteZmetadataZ cluster_nameZ all_hostsrdebug)r"r'r r!ecr%r%r&rDsL   z Export.initc Cstd|ddtt||D}zd|j}|j|}|j ||t t |fWdSt yO}ztd||WYd}~dSd}~ww)z*Write the points to the Cassandra cluster.zExport {} stats to CassandracSs$i|]\}}t|tr|t|qSr%) isinstancerfloat).0kvr%r%r& rs$z!Export.export..z4INSERT INTO {} (plugin, time, stat) VALUES (?, ?, ?)z(Cannot export {} stats to Cassandra ({})N)rr-r+dictziprrr!Zpreparer,r rZnowr)error)r"namecolumnsZpointsdataZstmtZqueryr.r%r%r&exportms    z Export.exportcs&|j|jtt|dS)z"Close the Cassandra export module.N)r!Zshutdownr rr r)r"r#r%r&rs  z Export.exit)NN) __name__ __module__ __qualname____doc__rrr<r __classcell__r%r%r#r&r $s )r )r@rrZnumbersrZglances.loggerrZglances.exports.glances_exportrZglances.compatrZcassandra.authrZcassandra.clusterrZcassandra.utilr rr r r%r%r%r&s