Function main

benchmarks/src/main/python/insert_bench.py:71–100

()

async def main():
    db_config = {
        'host': 'localhost',
        'port': 5432,
        'user': 'admin',
        'password': 'quest',
        'database': 'qdb'
    }

    pool = await asyncpg.create_pool(**db_config, min_size=20, max_size=20)

    try:
        await create_table(pool)

        print("Generating test data...")
        records = generate_test_data(1_000_000)

        start_time = datetime.now()

        print("Starting batch insert...")
        await batch_insert(pool, records)

        end_time = datetime.now()
        duration = (end_time - start_time).total_seconds()

        print(f"Successfully inserted 1,000,000 records in {duration:.2f} seconds")
        print(f"Average insertion rate: {1_000_000 / duration:.2f} records/second")

    finally:
        await pool.close()


if __name__ == '__main__':