테이블 Truncate 함수 추가

main
jaehwang 2025-11-10 11:02:56 +09:00
parent a951d0d1b6
commit 9196ed08b4
2 changed files with 14 additions and 0 deletions

View File

@ -22,6 +22,7 @@ class UserFactory:
return (cls.generate_user_name(), cls.generate_phone_number()) return (cls.generate_user_name(), cls.generate_phone_number())
async def main(): async def main():
await sql.truncate_user_table() # clear user table
for i in range(INSERT_NUM): for i in range(INSERT_NUM):
await sql.create_user(*UserFactory.create()) await sql.create_user(*UserFactory.create())
# await asyncio.gather(*[sql.create_user(*UserFactory.create()) for i in range(INSERT_NUM)]) # 커넥션 INSERT NUM개수만큼 생기는 문제 Semaphore 등으로 해결해야 쓸수있음 # await asyncio.gather(*[sql.create_user(*UserFactory.create()) for i in range(INSERT_NUM)]) # 커넥션 INSERT NUM개수만큼 생기는 문제 Semaphore 등으로 해결해야 쓸수있음

View File

@ -227,3 +227,16 @@ async def delete_blog_from_id(blog_id:int):
return return
async def truncate_user_table():
async with await get_cnx() as cnx:
async with await cnx.cursor() as cur:
query = '''
TRUNCATE TABLE user_table
'''
await cur.execute(query)
await cnx.commit()
return