(pool, records, batch_size=1000)
| 54 | |
| 55 | |
| 56 | async def batch_insert(pool, records, batch_size=1000): |
| 57 | async with pool.acquire() as conn: |
| 58 | for i in range(0, len(records), batch_size): |
| 59 | batch = records[i:i + batch_size] |
| 60 | |
| 61 | # Execute batch insert |
| 62 | await conn.executemany( |
| 63 | 'INSERT INTO user_data(name, email, age, created_at) VALUES ($1, $2, $3, $4)', |
| 64 | batch |
| 65 | ) |
| 66 | |
| 67 | if i % 100_000 == 0: |
| 68 | print(f"Inserted {i} records") |
| 69 | |
| 70 | |
| 71 | async def main(): |