python下保持mysql连接，避免“MySQL server has gone away“方法
#!/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
页:
[1]