파이썬을 이용해 mysql에 dummy 데이터 추가하기

xeonu·2023년 3월 6일
0

부하테스트 목적으로 dummy 데이터를 생성해야할 일이 종종 생긴다. 하나하나 수작업으로 insert 쿼리를 생성하는 것은 컴퓨터공학도 답지 못한 생각이다. 파이썬으로 쿼리에 필요한 랜덤 데이터를 생성하고 mysql과 연동한 후 쿼리를 전송하면된다.

import pymysql
import uuid
import secrets

# MySQL Connection 연결
con = pymysql.connect(host='', user='', password='', database='')

# Connection으로부터 Cursor 생성
cur = con.cursor()

# 랜덤 데이터 생성 및 INSERT 쿼리 실행
for i in range(1000):
    login_id = str(uuid.uuid4())[:10]
    password = secrets.token_urlsafe(16)
    sql = "INSERT INTO member (login_id, password) VALUES (%s, %s)"
    cur.execute(sql, (login_id, password))

# DB 연결 종료
con.commit()
con.close()

1000개의 쿼리를 날려주면된다. 주의할 점으로는 connect할 때 host에는 port 번호 없이 ip 주소만 작성해주면된다. 또 너무 많은 양의 쿼리를 한번에 날릴 경우 오류가 발생할 수 있으니 적절한 숫자를 조절해가며 dummy 데이터를 추가하면된다.

profile
백엔드 개발자가 되기위한 여정

0개의 댓글