Function do_inserts

postgres/tests/py_tests/test_concurrent_insert_index.py:61–146

()

    Runs in separate process to avoid blocking.
    """
    async def do_inserts():
        user = os.environ.get("USER", "postgres")
        conn = await asyncpg.connect(
            database="postgres",
            user=user,
            host="localhost",
            statement_cache_size=0
        )

        try:
            print(f"Insert worker: Starting {insert_count} inserts in batches of {batch_size}...")

            completed = 0
            batch_num = 0

            while completed < insert_count:
                current_batch_size = min(batch_size, insert_count - completed)
                batch_num += 1

                # Single insert for first iteration
                if completed == 0:
                    await conn.execute("""
                        INSERT INTO chunk_text_image (id, text, image, metadata, chunk_index, file_id, created_at)
                        VALUES (
                            'chunk_12345',
                            'This is the text content of the chunk',
                            '\\xDEADBEEF'::bytea,
                            '{"key": "value", "tags": ["tag1", "tag2"]}'::jsonb,
                            1,
                            '550e8400-e29b-41d4-a716-446655440000'::uuid,
                            1700000000
                        )
                    """)
                    completed += 1
                    print(f"Insert worker: Completed 1 insert")
                    continue

                # Batch inserts
                values_list = []
                for i in range(current_batch_size):
                    idx = completed + i
                    chunk_id = f'chunk_{idx:09d}'
                    text = f'Chunk text content number {idx}'

                    if idx % 5 == 0:
                        image = 'NULL'
                    elif idx % 3 == 0:
                        image = "'\\x89504E47'::bytea"
                    else:
                        image = "'\\xDEADBEEF'::bytea"

                    metadata = f'{{"index": {idx}, "batch": {batch_num}}}'
                    chunk_index = idx
                    file_id = '550e8400-e29b-41d4-a716-446655440000' if idx % 2 == 0 else '660e8400-e29b-41d4-a716-446655440001'
                    created_at = 1700000000 + idx

                    values_list.append(
                        f"('{chunk_id}', '{text}', {image}, '{metadata}'::jsonb, "