From 954746bac087ddbd05c6f6c5ccee2c0b0a796a43 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carlos=20P=C3=A9rez-Aradros=20Herce?= Date: Fri, 3 Nov 2023 17:16:04 +0100 Subject: [PATCH] Fix primary keys retrieval query (#198) The previous query had some issues when used under clusters with many schema objects, this one improves query performance --- pkg/state/state.go | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/pkg/state/state.go b/pkg/state/state.go index 531c5a2..5736f91 100644 --- a/pkg/state/state.go +++ b/pkg/state/state.go @@ -143,20 +143,20 @@ BEGIN ) c ), 'primaryKey', ( - SELECT json_agg(kcu.column_name) AS primary_key_columns - FROM information_schema.table_constraints AS tc - JOIN information_schema.key_column_usage AS kcu - ON tc.constraint_name = kcu.constraint_name - WHERE tc.table_name = t.relname - AND tc.table_schema = schemaname - AND tc.constraint_type = 'PRIMARY KEY' + SELECT json_agg(pg_attribute.attname) AS primary_key_columns + FROM pg_index, pg_attribute + WHERE + indrelid = t.oid AND + nspname = schemaname AND + pg_attribute.attrelid = t.oid AND + pg_attribute.attnum = any(pg_index.indkey) + AND indisprimary ), 'indexes', ( SELECT json_object_agg(pi.indexrelid::regclass, json_build_object( 'name', pi.indexrelid::regclass )) FROM pg_index pi - INNER JOIN pg_class pgc ON pi.indexrelid = pgc.oid WHERE pi.indrelid = t.oid::regclass ) )) FROM pg_class AS t