Show space usage for materialized views (#485)

ankane · Feb 20, 2024 · 20d9ffc · 20d9ffc
1 parent 1b5d6e1
commit 20d9ffc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/lib/pghero/methods/space.rb b/lib/pghero/methods/space.rb
@@ -10,7 +10,7 @@ def relation_sizes
           SELECT
             n.nspname AS schema,
             c.relname AS relation,
-            CASE WHEN c.relkind = 'r' THEN 'table' ELSE 'index' END AS type,
+            CASE c.relkind WHEN 'r' THEN 'table' WHEN 'm' then 'matview' ELSE 'index' END AS type,
             pg_table_size(c.oid) AS size_bytes
           FROM
             pg_class c
@@ -19,7 +19,7 @@ def relation_sizes
           WHERE
             n.nspname NOT IN ('pg_catalog', 'information_schema')
             AND n.nspname !~ '^pg_toast'
-            AND c.relkind IN ('r', 'i')
+            AND c.relkind IN ('r', 'm', 'i')
           ORDER BY
             pg_table_size(c.oid) DESC,
             2 ASC

diff --git a/test/internal/db/schema.rb b/test/internal/db/schema.rb
@@ -31,7 +31,7 @@
     t.string :name
   end
 
-  create_table :users, force: true do |t|
+  create_table :users, force: :cascade do |t|
     t.integer :city_id
     t.integer :login_attempts
     t.string :email

diff --git a/test/test_helper.rb b/test/test_helper.rb
@@ -73,3 +73,4 @@ class User < ActiveRecord::Base
   end
 User.insert_all!(users)
 ActiveRecord::Base.connection.execute("ANALYZE users")
+ActiveRecord::Base.connection.execute("CREATE MATERIALIZED VIEW all_users AS SELECT * FROM users")