execute amcheck before doing backups

Author: evgeni

definitions/checks/candlepin/db_index.rb

@@ -0,0 +1,21 @@
+module Checks
+  module Candlepin
+    class DBIndex < ForemanMaintain::Check
+      metadata do
+        description 'Make sure Candlepin DB indexes are OK'
+        label :candlepin_db_index
+        tags :db_index
+        for_feature :candlepin_database
+        confine do
+          feature(:candlepin_database)&.local?
+        end
+      end
+
+      def run
+        status, output = feature(:candlepin_database).amcheck
+
+        assert(status == 0, "Candlepin DB indexes have issues:\n#{output}")
+      end
+    end
+  end
+end

definitions/checks/foreman/db_index.rb

@@ -0,0 +1,21 @@
+module Checks
+  module Foreman
+    class DBIndex < ForemanMaintain::Check
+      metadata do
+        description 'Make sure Foreman DB indexes are OK'
+        label :foreman_db_index
+        tags :db_index
+        for_feature :foreman_database
+        confine do
+          feature(:foreman_database)&.local?
+        end
+      end
+
+      def run
+        status, output = feature(:foreman_database).amcheck
+
+        assert(status == 0, "Foreman DB indexes have issues:\n#{output}")
+      end
+    end
+  end
+end

definitions/checks/pulpcore/db_index.rb

@@ -0,0 +1,21 @@
+module Checks
+  module Pulpcore
+    class DBIndex < ForemanMaintain::Check
+      metadata do
+        description 'Make sure Pulpcore DB indexes are OK'
+        label :pulpcore_db_index
+        tags :db_index
+        for_feature :pulpcore_database
+        confine do
+          feature(:pulpcore_database)&.local?
+        end
+      end
+
+      def run
+        status, output = feature(:pulpcore_database).amcheck
+
+        assert(status == 0, "Pulpcore DB indexes have issues:\n#{output}")
+      end
+    end
+  end
+end

lib/foreman_maintain/concerns/base_database.rb

@@ -140,6 +140,37 @@ def raise_psql_missing_error
                 ' Make sure system has psql utility installed.'
       end
 
+      def amcheck
+        return unless local?
+
+        # creating the extension and executing the check requires superuser privileges
+
+        psql_cmd = "runuser - postgres -c 'psql --set=ON_ERROR_STOP=on #{configuration['database']}'"
+
+        amcheck_create = 'CREATE EXTENSION IF NOT EXISTS amcheck;'
+
+        execute!(psql_cmd, :stdin => amcheck_create)
+
+        amcheck_query = <<~SQL
+          SELECT bt_index_check(index => c.oid, heapallindexed => i.indisunique),
+               c.relname,
+               c.relpages
+          FROM pg_index i
+          JOIN pg_opclass op ON i.indclass[0] = op.oid
+          JOIN pg_am am ON op.opcmethod = am.oid
+          JOIN pg_class c ON i.indexrelid = c.oid
+          JOIN pg_namespace n ON c.relnamespace = n.oid
+          WHERE am.amname = 'btree' AND n.nspname = 'public'
+          -- Don't check temp tables, which may be from another session:
+          AND c.relpersistence != 't'
+          -- Function may throw an error when this is omitted:
+          AND c.relkind = 'i' AND i.indisready AND i.indisvalid
+          ORDER BY c.relpages DESC;
+        SQL
+
+        execute_with_status(psql_cmd, :stdin => amcheck_query)
+      end
+
       private
 
       def base_env