Python与MySQL的长连接

Python与MySQL的长连接

在系统运行中，有时候会需要与MySQL保持一个长连接，可以在函数外将连接设置为一个全局变量：

import pymysql
db = pymysql.connect(localhost = '127.0.0.1', user = 'user', passwd = 'password', port = 3306, db = 'dbanme')
cursor = db.cursor()

然后在需要连接的时候直接在函数中使用，不需要再在函数中重复上述操作。

但是有一个问题是，这种连接会断开，当你再次使用的时候就会报错，所以为了不影响与数据库的正常连接并使用，需要在每一次使用之前对这种连接进行判断，而这里就可以用到 ping() 这个函数。

db.ping()

如果连接是正常的则可以正常使用，否则，就需要再次进行连接：

def test_connection():
	try:
		db.ping()
	except:
		db = pymysql.connect(localhost = '127.0.0.1', user = 'user', passwd = 'password', port = 3306, db = 'dbanme')
	return db

def foo():
	db = test_connection()
	cursor = db.cursor()
	sql = 'select id from table_name;'	
	cursor.execute(sql)
	balabala

当然，如果是使用 Django 来进行与数据库的连接，使用的方法就多了，比如直接使用自带的ORM模型，或者 connection 的模块。
Django 内封的这个模块其实是很好用的，使用方法也很简单，最主要的是一次调用之后，如果是 insert 或者 delete 操作，不需要进行 commit()，会自动提交并且关闭与 MySQL 的连接：

from django.db import connection
sql = "select id from table_name"
with connection.cursor() as cursor():
	cursor.execute(sql)