From b44f6d1da85dd153467260df1716269bbd2ba8b0 Mon Sep 17 00:00:00 2001
From: Nikolay Samokhvalov <github2@samokhvalov.com>
Date: Sun, 26 Aug 2018 14:35:12 -0700
Subject: [PATCH] Fix: b4 fails if there are multiple schemas, reported by
 @sgrinko #21

---
 sql/b4_btree_pgstattuple.sql | 48 ++++++++++++++++++++----------------
 1 file changed, 27 insertions(+), 21 deletions(-)

diff --git a/sql/b4_btree_pgstattuple.sql b/sql/b4_btree_pgstattuple.sql
index 7c38fdf..e5ae597 100644
--- a/sql/b4_btree_pgstattuple.sql
+++ b/sql/b4_btree_pgstattuple.sql
@@ -3,31 +3,37 @@
 --https://github.com/dataegret/pg-utils/tree/master/sql
 --pgstattuple extension required
 --WARNING: without index name/mask query will read all available indexes which could cause I/O spikes
-with indexes as (
-    select * from pg_stat_user_indexes
-)
-select schemaname,
-table_name,
-pg_size_pretty(table_size) as table_size,
-index_name,
-pg_size_pretty(index_size) as index_size,
-idx_scan as index_scans,
-round((free_space*100/index_size)::numeric, 1) as waste_percent,
-pg_size_pretty(free_space) as waste
-from (
-    select schemaname, p.relname as table_name, indexrelname as index_name,
-    (select (case when avg_leaf_density = 'NaN' then 0 
-        else greatest(ceil(index_size * (1 - avg_leaf_density / (coalesce((SELECT (regexp_matches(reloptions::text, E'.*fillfactor=(\\d+).*'))[1]),'90')::real)))::bigint, 0) end)
-        from pgstatindex(schemaname || '.' || p.indexrelid::regclass::text)
+with data as (
+  select
+    schemaname as schema_name,
+    p.relname as table_name,
+    (select spcname from pg_tablespace where oid = c_table.reltablespace) as table_tblspace,
+    (select spcname from pg_tablespace where oid = c.reltablespace) as index_tblspace,
+    indexrelname as index_name,
+    (
+      select (case when avg_leaf_density = 'NaN' then 0
+        else greatest(ceil(index_size * (1 - avg_leaf_density / (coalesce((SELECT (regexp_matches(c.reloptions::text, E'.*fillfactor=(\\d+).*'))[1]),'90')::real)))::bigint, 0) end)
+      from pgstatindex(p.indexrelid::regclass::text)
     ) as free_space,
     pg_relation_size(p.indexrelid) as index_size,
     pg_relation_size(p.relid) as table_size,
     idx_scan
-    from indexes p
-    join pg_class c on p.indexrelid = c.oid
-    where pg_get_indexdef(p.indexrelid) like '%USING btree%' and
+  from pg_stat_user_indexes p
+  join pg_class c on p.indexrelid = c.oid
+  join pg_class c_table on p.relid = c_table.oid
+  where
+    pg_get_indexdef(p.indexrelid) like '%USING btree%'
     --put your index name/mask here
-    indexrelname ~ ''
-) t
+    and indexrelname ~ ''
+)
+select
+  coalesce(nullif(schema_name, 'public') || '.', '') || table_name || coalesce(' [' || table_tblspace || ']', '') as "Table",
+  coalesce(nullif(schema_name, 'public') || '.', '') || index_name || coalesce(' [' || index_tblspace || ']', '') as "Index",
+  pg_size_pretty(table_size) as "Table size",
+  pg_size_pretty(index_size) as "Index size",
+  idx_scan as "Index Scans",
+  round((free_space*100/index_size)::numeric, 1) as "Wasted, %",
+  pg_size_pretty(free_space) as "Wasted"
+from data
 order by free_space desc;