mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-22 21:59:59 +00:00
5bd8e2363a
This will help to keep us from using deprecated Python features going forward. Signed-off-by: Tristan Partin <tristan@neon.tech>
83 lines
3.3 KiB
Python
83 lines
3.3 KiB
Python
from __future__ import annotations
|
|
|
|
import requests
|
|
from fixtures.benchmark_fixture import MetricReport, NeonBenchmarker
|
|
from fixtures.neon_fixtures import NeonEnvBuilder
|
|
|
|
|
|
# Just start and measure duration.
|
|
#
|
|
# This test runs pretty quickly and can be informative when used in combination
|
|
# with emulated network delay. Some useful delay commands:
|
|
#
|
|
# 1. Add 2msec delay to all localhost traffic
|
|
# `sudo tc qdisc add dev lo root handle 1:0 netem delay 2msec`
|
|
#
|
|
# 2. Test that it works (you should see 4ms ping)
|
|
# `ping localhost`
|
|
#
|
|
# 3. Revert back to normal
|
|
# `sudo tc qdisc del dev lo root netem`
|
|
#
|
|
# NOTE this test might not represent the real startup time because the basebackup
|
|
# for a large database might be larger if there's a lof of transaction metadata,
|
|
# or safekeepers might need more syncing, or there might be more operations to
|
|
# apply during config step, like more users, databases, or extensions. By default
|
|
# we load extensions 'neon,pg_stat_statements,timescaledb,pg_cron', but in this
|
|
# test we only load neon.
|
|
def test_startup_simple(neon_env_builder: NeonEnvBuilder, zenbenchmark: NeonBenchmarker):
|
|
neon_env_builder.num_safekeepers = 3
|
|
env = neon_env_builder.init_start()
|
|
|
|
env.create_branch("test_startup")
|
|
|
|
endpoint = None
|
|
|
|
# We do two iterations so we can see if the second startup is faster. It should
|
|
# be because the compute node should already be configured with roles, databases,
|
|
# extensions, etc from the first run.
|
|
for i in range(2):
|
|
# Start
|
|
with zenbenchmark.record_duration(f"{i}_start_and_select"):
|
|
if endpoint:
|
|
endpoint.start()
|
|
else:
|
|
endpoint = env.endpoints.create_start(
|
|
"test_startup",
|
|
# Shared buffers need to be allocated during startup, so they
|
|
# impact startup time. This is the default value we use for
|
|
# 1CPU pods (maybe different for VMs).
|
|
#
|
|
# TODO extensions also contribute to shared memory allocation,
|
|
# and this test doesn't include all default extensions we
|
|
# load.
|
|
config_lines=["shared_buffers=262144"],
|
|
)
|
|
endpoint.safe_psql("select 1;")
|
|
|
|
# Get metrics
|
|
metrics = requests.get(f"http://localhost:{endpoint.http_port}/metrics.json").json()
|
|
durations = {
|
|
"wait_for_spec_ms": f"{i}_wait_for_spec",
|
|
"sync_safekeepers_ms": f"{i}_sync_safekeepers",
|
|
"sync_sk_check_ms": f"{i}_sync_sk_check",
|
|
"basebackup_ms": f"{i}_basebackup",
|
|
"start_postgres_ms": f"{i}_start_postgres",
|
|
"config_ms": f"{i}_config",
|
|
"total_startup_ms": f"{i}_total_startup",
|
|
}
|
|
for key, name in durations.items():
|
|
value = metrics[key]
|
|
zenbenchmark.record(name, value, "ms", report=MetricReport.LOWER_IS_BETTER)
|
|
|
|
# Check basebackup size makes sense
|
|
basebackup_bytes = metrics["basebackup_bytes"]
|
|
if i > 0:
|
|
assert basebackup_bytes < 100 * 1024
|
|
|
|
# Stop so we can restart
|
|
endpoint.stop()
|
|
|
|
# Imitate optimizations that console would do for the second start
|
|
endpoint.respec(skip_pg_catalog_updates=True)
|