Блог

Я использую объединение таблиц, чтобы выбирать элементы из моей базы данных по списку кортежей параметров. Запрос работает нормально, но cur.fetchall() не возвращает всю таблицу, которую я хочу.

Например:

 data = (
    (1, '2020-11-19'),
    (1, '2020-11-20'),
    (1, '2020-11-21'),
    (2, '2020-11-19'),
    (2, '2020-11-20'),
    (2, '2020-11-21')
)
        
query = """
    with data(song_id, date) as (
        values %s
    )
    select t.*
    from my_table t
    join data d 
    on t.song_id = d.song_id and t.date = d.date::date
"""
execute_values(cursor, query, data)
results = cursor.fetchall()
 

На практике мой список кортежей для проверки состоит из тысяч строк, и я ожидаю, что ответ также будет длиной в тысячи строк.

Но я получаю только 5 строк обратно, если я вызываю cur.fetchall() в конце этого запроса.

Я знаю, что это связано execute_values с тем, что запросы обрабатываются пакетами, но есть какое-то странное поведение.

Если я пройду page_size=10 , я получу 2 элемента обратно. И если я установлю fetch=True , я вообще не получу никаких результатов (даже если rowcount это не соответствует этому).

Я думал, что нужно выполнить эти запросы, но page_size для пакета не соответствует количеству элементов, которые я ожидаю для каждого пакета.

Как мне изменить этот запрос, чтобы я мог получить все ожидаемые результаты?