Merge pull request #34315 from backstage/freben/search-extended-statistics
Add extended statistics on search(key, value) for query planner accuracy
This commit is contained in:
@@ -0,0 +1,5 @@
|
||||
---
|
||||
'@backstage/plugin-catalog-backend': patch
|
||||
---
|
||||
|
||||
Added extended multi-column statistics on `(key, value)` in the `search` table (PostgreSQL only). This tells the query planner about the correlation between the `key` and `value` columns, fixing severe row count estimation errors on compound filter queries. Without this, the planner could choose to materialize and sort thousands of rows instead of using the LIMIT short-circuit index scan — causing 10-40x slower catalog list views when multiple filters are active.
|
||||
@@ -0,0 +1,87 @@
|
||||
/*
|
||||
* Copyright 2026 The Backstage Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
// @ts-check
|
||||
|
||||
/**
|
||||
* Creates extended multi-column statistics on (key, value) in the search
|
||||
* table. These statistics capture the correlation between `key` and `value`
|
||||
* columns, which the planner cannot infer from standard single-column
|
||||
* statistics alone. Without them, compound filter queries like
|
||||
* `WHERE key='kind' AND value='component'` get wildly underestimated
|
||||
* (e.g. 13 estimated vs 17,000 actual rows), causing the planner to
|
||||
* choose materialize-then-sort plans instead of LIMIT-short-circuit
|
||||
* index scans.
|
||||
*
|
||||
* ## What this creates
|
||||
*
|
||||
* On PostgreSQL 10+:
|
||||
* CREATE STATISTICS search_key_value_stats (dependencies, ndistinct, mcv)
|
||||
* ON key, value FROM search;
|
||||
*
|
||||
* - `dependencies`: tells the planner that `value` depends on `key`
|
||||
* - `ndistinct`: number of distinct (key, value) combinations
|
||||
* - `mcv`: most common (key, value) pairs with their actual frequencies
|
||||
*
|
||||
* ## Cost
|
||||
*
|
||||
* - **Creation**: `CREATE STATISTICS` is metadata-only (instant).
|
||||
* - **ANALYZE**: reads a sample of the table (~30k rows by default) to
|
||||
* compute the statistics. Takes 2-4 seconds on a 14M-row table. This
|
||||
* happens once on migration and then automatically during regular
|
||||
* autovacuum analyze cycles.
|
||||
* - **Storage**: a few KB in `pg_statistic_ext_data` — negligible.
|
||||
* - **Maintenance**: autovacuum refreshes the statistics during its
|
||||
* regular `ANALYZE` passes, just like single-column statistics.
|
||||
* No manual maintenance needed.
|
||||
*
|
||||
* MySQL and SQLite do not support extended statistics; this migration
|
||||
* is a no-op on those engines.
|
||||
*/
|
||||
|
||||
/**
|
||||
* @param {import('knex').Knex} knex
|
||||
*/
|
||||
exports.up = async function up(knex) {
|
||||
if (!knex.client.config.client.includes('pg')) {
|
||||
return;
|
||||
}
|
||||
|
||||
const exists = await knex.raw(
|
||||
`SELECT 1 FROM pg_statistic_ext WHERE stxname = 'search_key_value_stats'`,
|
||||
);
|
||||
|
||||
if (exists.rows.length > 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
await knex.raw(
|
||||
`CREATE STATISTICS search_key_value_stats (dependencies, ndistinct, mcv) ON key, value FROM search`,
|
||||
);
|
||||
|
||||
await knex.raw(`ANALYZE search`);
|
||||
};
|
||||
|
||||
/**
|
||||
* @param {import('knex').Knex} knex
|
||||
*/
|
||||
exports.down = async function down(knex) {
|
||||
if (!knex.client.config.client.includes('pg')) {
|
||||
return;
|
||||
}
|
||||
|
||||
await knex.raw(`DROP STATISTICS IF EXISTS search_key_value_stats`);
|
||||
};
|
||||
@@ -1353,4 +1353,63 @@ describe.each(databases.eachSupportedId())('migrations, %p', databaseId => {
|
||||
|
||||
await knex.destroy();
|
||||
});
|
||||
|
||||
it('20260519000000_search_extended_statistics.js', async () => {
|
||||
const knex = await databases.init(databaseId);
|
||||
const client = knex.client.config.client;
|
||||
const isPg = typeof client === 'string' && client.includes('pg');
|
||||
|
||||
await migrateUntilBefore(
|
||||
knex,
|
||||
'20260519000000_search_extended_statistics.js',
|
||||
);
|
||||
|
||||
await knex('refresh_state').insert({
|
||||
entity_id: 'e1',
|
||||
entity_ref: 'k:ns/n1',
|
||||
unprocessed_entity: '{}',
|
||||
errors: '[]',
|
||||
next_update_at: knex.fn.now(),
|
||||
last_discovery_at: knex.fn.now(),
|
||||
});
|
||||
await knex('final_entities').insert({
|
||||
entity_id: 'e1',
|
||||
entity_ref: 'k:ns/n1',
|
||||
hash: 'h1',
|
||||
final_entity: '{}',
|
||||
});
|
||||
await knex('search').insert([
|
||||
{
|
||||
entity_id: 'e1',
|
||||
key: 'kind',
|
||||
value: 'component',
|
||||
original_value: 'Component',
|
||||
},
|
||||
{
|
||||
entity_id: 'e1',
|
||||
key: 'metadata.name',
|
||||
value: 'my-svc',
|
||||
original_value: 'my-svc',
|
||||
},
|
||||
]);
|
||||
|
||||
// statsExist returns false on non-PG engines (no extended stats support)
|
||||
async function statsExist(): Promise<boolean> {
|
||||
if (!isPg) return false;
|
||||
const r = await knex.raw(
|
||||
`SELECT 1 FROM pg_statistic_ext WHERE stxname = 'search_key_value_stats'`,
|
||||
);
|
||||
return r.rows.length > 0;
|
||||
}
|
||||
|
||||
expect(await statsExist()).toBe(false);
|
||||
|
||||
await migrateUpOnce(knex);
|
||||
expect(await statsExist()).toBe(isPg);
|
||||
|
||||
await migrateDownOnce(knex);
|
||||
expect(await statsExist()).toBe(false);
|
||||
|
||||
await knex.destroy();
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user