因需要对saltstack的所有动作进行入库采集,网上采集脚本mysql连接会因超时而断开,导致守护进程在下一次采集数据时提示:Traceback (most recent call last):
File "./salt_event_to_mysql.py", line 39, in
ret[‘success‘], json.dumps(ret)))
File "build/bdist.linux-x86_64/egg/MySQLdb/cursors.py", line 173, in execute
File "build/bdist.linux-x86_64/egg/MySQLdb/connections.py", line 36, in defaulterrorhandler
_mysql_exceptions.OperationalError: (2006, ‘MySQL server has gone away‘)
后退出,解决方法是采用类似mysql_ping()的方法在每次数据入库前做一次检测,唤醒连接,以下是完整代码:
#!/bin/env python
#coding=utf8
# Import python libs
import json
# Import salt modules
import salt.config
import salt.utils.event
# Import third party libs
import MySQLdb
__opts__ = salt.config.client_config(‘/etc/salt/master‘)
# Create MySQL connect
conn = MySQLdb.connect(host=__opts__[‘mysql‘][‘host‘], user=__opts__[‘mysql‘][‘user‘], passwd=__opts__[‘mysql‘][‘pass‘], db=__opts__[‘mysql‘][‘db‘], port=__opts__[‘mysql‘][‘port‘])
cursor = conn.cursor()
# Listen Salt Master Event System
event = salt.utils.event.MasterEvent(__opts__[‘sock_dir‘])
for eachevent in event.iter_events(full=True):
####保持连接的部分###########
if conn is None:
conn = MySQLdb.connect(host=__opts__[‘mysql‘][‘host‘], user=__opts__[‘mysql‘][‘user‘], passwd=__opts__[‘mysql‘][‘pass‘], db=__opts__[‘mysql‘][‘db‘], port=__opts__[‘mysql‘][‘port‘])
cursor = conn.cursor()
else:
conn.ping(True)
###########################
ret = eachevent[‘data‘]
if "salt/job/" in eachevent[‘tag‘]:
# Return Event
if ret.has_key(‘id‘) and ret.has_key(‘return‘):
# Igonre saltutil.find_job event
if ret[‘fun‘] == "saltutil.find_job":
continue
if ret[‘fun‘] == "test.ping":
continue
sql = ‘‘‘INSERT INTO `salt_returns`
(`fun`, `jid`, `return`, `id`, `success`, `full_ret` )
VALUES (%s, %s, %s, %s, %s, %s)‘‘‘
cursor.execute(sql, (ret[‘fun‘], ret[‘jid‘],
json.dumps(ret[‘return‘]), ret[‘id‘],
ret[‘success‘], json.dumps(ret)))
cursor.execute("COMMIT")
# Other Event
else:
pass