Show space usage for partitioned tables

CHANGELOG.md

@@ -1,5 +1,6 @@
 ## 3.4.2 (unreleased)
 
+- Added partitioned tables to space page
 - Added materialized views to space page
 - Fixed error with `slow_queries` method
 

lib/pghero/methods/space.rb

@@ -10,7 +10,12 @@ def relation_sizes
           SELECT
             n.nspname AS schema,
             c.relname AS relation,
-            CASE c.relkind WHEN 'r' THEN 'table' WHEN 'm' then 'matview' ELSE 'index' END AS type,
+            CASE c.relkind
+              WHEN 'r' THEN 'table'
+              WHEN 'p' THEN 'partitioned table'
+              WHEN 'm' then 'matview'
+              ELSE 'index'
+            END AS type,
             pg_table_size(c.oid) AS size_bytes
           FROM
             pg_class c
@@ -19,7 +24,7 @@ def relation_sizes
           WHERE
             n.nspname NOT IN ('pg_catalog', 'information_schema')
             AND n.nspname !~ '^pg_toast'
-            AND c.relkind IN ('r', 'm', 'i')
+            AND c.relkind IN ('r', 'p', 'm', 'i')
           ORDER BY
             pg_table_size(c.oid) DESC,
             2 ASC

test/space_test.rb

@@ -10,6 +10,9 @@ def test_relation_sizes
     assert relation_sizes.find { |r| r[:relation] == "users" && r[:type] == "table" }
     assert relation_sizes.find { |r| r[:relation] == "users_pkey" && r[:type] == "index" }
     assert relation_sizes.find { |r| r[:relation] == "all_users" && r[:type] == "matview" }
+    assert relation_sizes.find { |r| r[:relation] == "partitioned_users" && r[:type] == "partitioned table" }
+    assert relation_sizes.find { |r| r[:relation] == "partitioned_users_1" && r[:type] == "table" }
+    assert relation_sizes.find { |r| r[:relation] == "partitioned_users_2" && r[:type] == "table" }
   end
 
   def test_table_sizes

test/test_helper.rb

@@ -74,3 +74,14 @@ class User < ActiveRecord::Base
 User.insert_all!(users)
 ActiveRecord::Base.connection.execute("ANALYZE users")
 ActiveRecord::Base.connection.execute("CREATE MATERIALIZED VIEW all_users AS SELECT * FROM users")
+ActiveRecord::Base.connection.execute(<<~SQL)
+  CREATE TABLE partitioned_users (
+    city_id integer NOT NULL
+  ) PARTITION BY RANGE (city_id);
+
+  CREATE TABLE partitioned_users_1 PARTITION OF partitioned_users
+    FOR VALUES FROM (1) TO (1000);
+
+  CREATE TABLE partitioned_users_2 PARTITION OF partitioned_users
+    FOR VALUES FROM (1001) TO (2000);
+SQL