场景：做性能测试，向Mysql数据库，student表中插入10w条测试数据，每条数据不重复。
解决方案：
这里采用Python自动生成SQL，再执行已生成好的SQL往Mysql数据库插入数据。
使用语言：python 3.8
准备SQL 语句，需每条 id 不重复 ,创建student表
create table student(
s_no int primary key auto_increment,
s_name varchar(20) not null,
s_sex varchar(10) not null,
s_birthday datetime,
s_class VARCHAR(20)
);
插入单条数据SQL:
INSERT into student(s_name,s_sex,s_birthday,s_class) VALUES('利玉2','男','1994-08-02','95033');
插入多条数据SQL:
insert into student (s_name,s_sex,s_birthday,s_class) values ('褚饯新', '女', '1978-11-27 00:00:00','95032'), ('蒋斗越', '女', '1971-10-02 00:00:00','95034');
插入多条需要生成多个insert 语句，这里用 Python 语言快速生成插入的SQL脚本，每条SQL后面分号隔开,每次写入数据，最后面加\n 换行,然后拼接执行的sql语句，拼接的时候需注意，最后的字符 ，需改成 ；,用 %s 替换需要变的字段值，如果有多个值都需要变，可以用多个%s替换对应值。用for 循环控制插入的数据数目，因为ID是自动生成，所以不会重复。
学生的姓名、性别、出生日期、班级编号全部自动生成。这里使用Python封装成函数，在拼接SQL时直接引用即可。
在执行代码前先获取当前的时间戳，代码执行完成后再次获取一次时间戳。两次的时间间隔，就是执行的时间了，时间单位是s.
在cmd命令行执行命令安装pymysql：
pip3 install pymysql -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
具体代码如下：
from pymysql import *
import time
import random
dbinfo = {
"host": "localhost",
"user": "root",
"password": "123123",
"port": 3306}
class DbConnect():
def __init__(self,db_cof,database=""):
self.db_cof = db_cof
self.db = connect(database=database,cursorclass=cursors.DictCursor,**db_cof,charset='utf8')
self.cursor = self.db.cursor()
def select(self,sql):
self.cursor.execute(sql)
results = self.cursor.fetchall()
return results
def execute(self,sql):
try:
self.cursor.execute(sql)
self.db.commit()
except:
self.db.rollback()
print("发生回滚")
def close(self):
self.db.close()
# 随机生成出生日期
def get_birthday(self):
# 随机生成年月日
year = random.randint(1960, 2000)
month = random.randint(1, 12)
# 判断每个月有多少天随机生成日
if year % 4 == 0:
if month in (1, 3, 5, 7, 8, 10, 12):
day = random.randint(1, 31)
elif month in (4, 6, 9, 11):
day = random.randint(1, 30)
else:
day = random.randint(1, 29)
else:
if month in (1, 3, 5, 7, 8, 10, 12):
day = random.randint(1, 31)
elif month in (4, 6, 9, 11):
day = random.randint(1, 30)
else:
day = random.randint(1, 28)
# 小于10的月份前面加0
if month < 10:
month = '0' + str(month)
if day < 10:
day = '0' + str(day)
birthday = str(year) +'-'+ str(month) +'-'+ str(day)+' '+'00:00:00'
return birthday
def get_name(self):
list_Xing = ['赵', '钱', '孙', '李', '周', '吴', '郑', '王', '冯', '陈', '褚', '卫', '蒋', '沈', '韩', '杨', '张', '李']
list_Ming = ['豫', '章', '故', '郡', '洪', '都', '新', '府', '星', '分', '翼', '轸', '地', '接', '衡', '庐', '襟', '三', '江', '',
'而', '带', '五', '湖', '控', '蛮', '荆', '而', '引', '瓯', '越', '物', '华', '天', '宝', '龙', '光', '射', '牛',
'斗', '之', '墟', '人', '杰', '地', '灵', '徐', '孺', '饯', '子']
name = random.choice(list_Xing)+random.choice(list_Ming)+random.choice(list_Ming)
return name
def get_sex(self):
list_Sex = ['男', '女']
sex = random.choice(list_Sex)
return sex
def get_class(self):
list_Class = ['95033', '95032', '95034', '95038']
classs = random.choice(list_Class)
return classs
def test(self):
db = DbConnect(dbinfo,database='selecttest')
insert_sql = "insert into student (s_name,s_sex,s_birthday,s_class) values "
insert_values ="".join(["('"+db.get_name()+ "', "
"'"+db.get_sex()+"', "
"'"+db.get_birthday()+"',"
"'"+db.get_class()+"'),\n"
for i in range(100000)])
print(insert_values[:-2])
sql = insert_sql+insert_values[:-2]+";"
print(sql)
time1 = time.time()
db.execute(sql)
db.close()
time2 = time.time()
print("总过耗时：%s" % (time2-time1))
if __name__ == '__main__':
DbConnect.test(DbConnect)
以下是插入10条的执行结果：
insert into student (s_name,s_sex,s_birthday,s_class) values ('蒋星', '男', '1971-09-21 00:00:00','95038'),
('韩接斗', '男', '1962-02-08 00:00:00','95038'),
('卫瓯接', '男', '1994-08-03 00:00:00','95034'),
('卫子孺', '女', '1983-12-16 00:00:00','95032'),
('李府饯', '女', '1990-06-29 00:00:00','95032'),
('沈华徐', '女', '1998-09-15 00:00:00','95033'),
('赵孺郡', '女', '1990-07-12 00:00:00','95034'),
('韩牛分', '女', '1982-09-27 00:00:00','95038'),
('钱之湖', '女', '1981-02-17 00:00:00','95038'),
('李分天', '男', '1969-08-15 00:00:00','95034');
总过耗时：0.18653035163879395
进入Navicat，查询student表,已正确插入
image.png
