KDOC 659: 1秒より短いクエリも Performance Insights の負荷には実行時間どおりに出る

# 負荷を流す側と標本を取る側の2つのセッションが同時に要るので、pgpg のプロキシを通さず PostgreSQL に直接(15433)つなぐ
sample() {
  psql -h localhost -p 15433 -U postgres -d postgres -qAt <<'SQL'
CREATE TEMP TABLE samples (query_id bigint);
-- Performance Insights と同じく、1秒ごとにアクティブなセッションを300回数える
DO $$
BEGIN
  FOR i IN 1..300 LOOP
    PERFORM pg_stat_clear_snapshot();
    INSERT INTO samples
    SELECT query_id FROM pg_stat_activity
    WHERE application_name = 'pgbench' AND state = 'active';
    PERFORM pg_sleep(1);
  END LOOP;
END $$;
-- 実行時間から求めた負荷(AAE)と、標本から求めた負荷(AAS)を並べる
SELECT format('| %s ms | %s | %s | %s | %s |',
              round(s.mean_exec_time), s.calls, round((s.total_exec_time / 1000 / 300)::numeric, 3),
              n, round(n / 300.0, 3))
FROM pg_stat_statements s,
     LATERAL (SELECT count(*) AS n FROM samples WHERE query_id = s.queryid) c
WHERE s.query LIKE 'SELECT%pg_sleep%'
ORDER BY s.mean_exec_time DESC;
SQL
}

docker exec pgpg-postgres-1 psql -U postgres -qAt -c "SELECT pg_stat_statements_reset()" > /dev/null
# pg_stat_statements は定数をまとめるので、文の形を変えて別の文として数えさせる
printf 'SELECT pg_sleep(1);\n' | docker exec -i pgpg-postgres-1 sh -c 'cat > /tmp/s1000.sql'
printf 'SELECT 1 FROM pg_sleep(0.1);\n' | docker exec -i pgpg-postgres-1 sh -c 'cat > /tmp/s100.sql'
printf 'SELECT 1, 2 FROM pg_sleep(0.01);\n' | docker exec -i pgpg-postgres-1 sh -c 'cat > /tmp/s10.sql'
# どれも実行時間の合計が経過時間の1割(AAE 0.1)になるように、1回の時間に反比例した頻度で流す
docker exec pgpg-postgres-1 pgbench -U postgres -n -R 0.1 -T 301 -f /tmp/s1000.sql postgres > /dev/null 2>&1 &
docker exec pgpg-postgres-1 pgbench -U postgres -n -R 1 -T 301 -f /tmp/s100.sql postgres > /dev/null 2>&1 &
docker exec pgpg-postgres-1 pgbench -U postgres -n -R 10 -T 301 -f /tmp/s10.sql postgres > /dev/null 2>&1 &
sleep 0.5
echo '| 文の時間 | 呼び出し | AAE | 標本(300回中) | AAS |'
echo '|-'
sample
wait