python模拟mysql多客户端并发操作

为测试系统多线程并发操作时的响应，今天写了个简单的python脚本往mysql中插入、取出数据，用到了MySQLdb、threading模块。

其中threading模块是对thread模块的一个面向对象的封装，比较好用。

import MySQLdb
import threading
import sys
def db_op_thread_func( i, num_of_op ):
conn=MySQLdb.connect(host="x.x.x.x",port=x, user="xxxx",passwd="",db="xxxx")
cursor = conn.cursor()
sql = "select * from xxxx"
for j in range(0, int(num_of_op) ):
cursor.execute( sql )
print "thread", i, ":", " num:", j

conn.close()
if __name__ == "__main__":
args = sys.argv
num_of_thd  = args[1]
num_of_op   = args[2]
threads = []
for i in range( 0, int(num_of_thd) ):
threads.append( threading.Thread( target=db_op_thread_func,args=(i, num_of_op ) ) )

for t in threads:
t.start()

for t in threads:
t.join()

注意：

1，为防止主线程先于其它线程退出造成程序异常，需要调用 join() 函数，等待进程结束；

2，print语句的作用是为了了解程序运行情况，实时关注程序是否卡在了什么地方；

3，python程序获得命令行参数；

改进：

1，如果想模拟不同的操作，可以将若干sql语句写到一个字符串数组中，随机获取，放到线程中执行；