mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-23 06:09:59 +00:00
30a7dd630c
## Problem `TYPE_CHECKING` is used inconsistently across Python tests. ## Summary of changes - Update `ruff`: 0.7.0 -> 0.11.2 - Enable TC (flake8-type-checking): https://docs.astral.sh/ruff/rules/#flake8-type-checking-tc - (auto)fix all new issues
105 lines
4.1 KiB
Python
105 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
import time
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from fixtures.log_helper import log
|
|
from fixtures.utils import USE_LFC
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import NeonEnv
|
|
|
|
|
|
@pytest.mark.timeout(600)
|
|
@pytest.mark.skipif(not USE_LFC, reason="LFC is disabled, skipping")
|
|
def test_lfc_prefetch(neon_simple_env: NeonEnv):
|
|
"""
|
|
Test resizing the Local File Cache
|
|
"""
|
|
env = neon_simple_env
|
|
endpoint = env.endpoints.create_start(
|
|
"main",
|
|
config_lines=[
|
|
"neon.max_file_cache_size=1GB",
|
|
"neon.file_cache_size_limit=1GB",
|
|
"effective_io_concurrency=100",
|
|
"shared_buffers=1MB",
|
|
"enable_bitmapscan=off",
|
|
"enable_seqscan=off",
|
|
"autovacuum=off",
|
|
],
|
|
)
|
|
conn = endpoint.connect()
|
|
cur = conn.cursor()
|
|
cur.execute("create extension neon")
|
|
cur.execute("create table t(pk integer, sk integer, filler text default repeat('x',200))")
|
|
cur.execute("set statement_timeout=0")
|
|
cur.execute("select setseed(0.5)")
|
|
cur.execute("insert into t values (generate_series(1,1000000),random()*1000000)")
|
|
cur.execute("create index on t(sk)")
|
|
cur.execute("vacuum t")
|
|
|
|
# reset LFC
|
|
cur.execute("alter system set neon.file_cache_size_limit=0")
|
|
cur.execute("select pg_reload_conf()")
|
|
time.sleep(1)
|
|
cur.execute("alter system set neon.file_cache_size_limit='1GB'")
|
|
cur.execute("select pg_reload_conf()")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 100000 and 200000 limit 100) s1"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 200000 and 300000 limit 100) s2"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 300000 and 400000 limit 100) s3"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 100000 and 200000 limit 100) s4"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
# if prefetch requests are not stored in LFC, we continue to sent unused prefetch request tyo PS
|
|
assert prefetch_expired > 0
|
|
|
|
cur.execute("set neon.store_prefetch_result_in_lfc=on")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 500000 and 600000 limit 100) s5"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 600000 and 700000 limit 100) s6"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 700000 and 800000 limit 100) s7"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
cur.execute(
|
|
"explain (analyze,prefetch,format json) select sum(pk) from (select pk from t where sk between 500000 and 600000 limit 100) s8"
|
|
)
|
|
prefetch_expired = cur.fetchall()[0][0][0]["Plan"]["Prefetch Expired Requests"]
|
|
log.info(f"Unused prefetches: {prefetch_expired}")
|
|
|
|
# No redundant prefethc requrests if prefetch results are stored in LFC
|
|
assert prefetch_expired == 0
|