mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-16 18:02:56 +00:00
123 lines
4.7 KiB
Python
123 lines
4.7 KiB
Python
import json
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
from fixtures.benchmark_fixture import NeonBenchmarker
|
|
from fixtures.log_helper import log
|
|
from fixtures.neon_fixtures import NeonEnvBuilder, PgBin, last_flush_lsn_upload
|
|
from fixtures.pageserver.utils import wait_until_tenant_active
|
|
from fixtures.remote_storage import LocalFsStorage, RemoteStorageKind
|
|
from fixtures.types import TenantId
|
|
|
|
|
|
def test_getpage_throughput(
|
|
neon_env_builder: NeonEnvBuilder, zenbenchmark: NeonBenchmarker, pg_bin: PgBin
|
|
):
|
|
neon_env_builder.enable_generations = True
|
|
neon_env_builder.enable_pageserver_remote_storage(RemoteStorageKind.LOCAL_FS)
|
|
env = neon_env_builder.init_start()
|
|
|
|
remote_storage = env.pageserver_remote_storage
|
|
assert isinstance(remote_storage, LocalFsStorage)
|
|
|
|
ps_http = env.pageserver.http_client()
|
|
|
|
# clean up the useless default tenant
|
|
ps_http.tenant_delete(env.initial_tenant)
|
|
|
|
# create our template tenant
|
|
tenant_config_mgmt_api = {
|
|
"gc_period": "0s",
|
|
"checkpoint_timeout": "3650 day",
|
|
"compaction_period": "20 s",
|
|
"compaction_threshold": 10,
|
|
"compaction_target_size": 134217728,
|
|
"checkpoint_distance": 268435456,
|
|
"image_creation_threshold": 3,
|
|
}
|
|
tenant_config_cli = {k: str(v) for k, v in tenant_config_mgmt_api.items()}
|
|
|
|
template_tenant, template_timeline = env.neon_cli.create_tenant(conf=tenant_config_cli)
|
|
template_tenant_gen = int(ps_http.tenant_status(template_tenant)["generation"])
|
|
with env.endpoints.create_start("main", tenant_id=template_tenant) as ep:
|
|
pg_bin.run_capture(["pgbench", "-i", "-s50", ep.connstr()])
|
|
last_flush_lsn_upload(env, ep, template_tenant, template_timeline)
|
|
ps_http.tenant_detach(template_tenant)
|
|
|
|
# stop PS just for good measure
|
|
env.pageserver.stop()
|
|
|
|
# duplicate the tenant in remote storage
|
|
src_timelines_dir: Path = remote_storage.tenant_path(template_tenant) / "timelines"
|
|
assert src_timelines_dir.is_dir(), f"{src_timelines_dir} is not a directory"
|
|
tenants = [template_tenant]
|
|
for i in range(0, 200):
|
|
new_tenant = TenantId.generate()
|
|
tenants.append(new_tenant)
|
|
log.info("Duplicating tenant #%s: %s", i, new_tenant)
|
|
|
|
dst_timelines_dir: Path = remote_storage.tenant_path(new_tenant) / "timelines"
|
|
dst_timelines_dir.parent.mkdir(parents=False, exist_ok=False)
|
|
dst_timelines_dir.mkdir(parents=False, exist_ok=False)
|
|
|
|
for tl in src_timelines_dir.iterdir():
|
|
src_tl_dir = src_timelines_dir / tl.name
|
|
assert src_tl_dir.is_dir(), f"{src_tl_dir} is not a directory"
|
|
dst_tl_dir = dst_timelines_dir / tl.name
|
|
dst_tl_dir.mkdir(parents=False, exist_ok=False)
|
|
for file in tl.iterdir():
|
|
shutil.copy2(file, dst_tl_dir)
|
|
if "__" in file.name:
|
|
cmd: List[str] = [
|
|
str(
|
|
env.neon_binpath / "pagectl"
|
|
), # TODO: abstract this like the other binaries
|
|
"layer",
|
|
"rewrite-summary",
|
|
str(dst_tl_dir / file.name),
|
|
"--new-tenant-id",
|
|
str(new_tenant),
|
|
]
|
|
subprocess.run(cmd, check=True)
|
|
else:
|
|
# index_part etc need no patching
|
|
pass
|
|
|
|
env.pageserver.start()
|
|
assert ps_http.tenant_list() == []
|
|
for tenant in tenants:
|
|
ps_http.tenant_attach(
|
|
tenant, config=tenant_config_mgmt_api, generation=template_tenant_gen + 1
|
|
)
|
|
for tenant in tenants:
|
|
wait_until_tenant_active(ps_http, tenant)
|
|
|
|
# ensure all layers are resident for predictiable performance
|
|
# TODO: ensure all kinds of eviction are disabled (per-tenant, disk-usage-based)
|
|
for tenant in tenants:
|
|
ps_http.download_all_layers(tenant, template_timeline)
|
|
|
|
# run the benchmark with one client per timeline, each doing 10k requests to random keys.
|
|
cmd = [
|
|
str(env.neon_binpath / "pagebench"),
|
|
"get-page-latest-lsn",
|
|
"--mgmt-api-endpoint",
|
|
ps_http.base_url,
|
|
"--page-service-connstring",
|
|
env.pageserver.connstr(password=None),
|
|
"--runtime",
|
|
"10s",
|
|
*[f"{tenant}/{template_timeline}" for tenant in tenants],
|
|
]
|
|
log.info(f"command: {' '.join(cmd)}")
|
|
basepath = pg_bin.run_capture(cmd)
|
|
results_path = Path(basepath + ".stdout")
|
|
log.info(f"Benchmark results at: {results_path}")
|
|
|
|
with open(results_path, "r") as f:
|
|
results = json.load(f)
|
|
|
|
log.info(f"Results:\n{json.dumps(results, sort_keys=True, indent=2)}")
|