Added pghero:clean_space_stats rake task [skip ci]

CHANGELOG.md

@@ -1,5 +1,6 @@
 ## 3.2.0 (unreleased)
 
+- Added `pghero:clean_space_stats` rake task
 - Added support for specifying retention period with `clean_query_stats`
 - Removed reset button when historical query stats are enabled
 

lib/pghero.rb

@@ -228,9 +228,9 @@ def clean_query_stats(before: nil)
       end
     end
 
-    def clean_space_stats
+    def clean_space_stats(before: nil)
       each_database do |database|
-        database.clean_space_stats
+        database.clean_space_stats(before: before)
       end
     end
 

lib/pghero/methods/space.rb

@@ -129,8 +129,9 @@ def capture_space_stats
         insert_stats("pghero_space_stats", columns, values) if values.any?
       end
 
-      def clean_space_stats
-        PgHero::SpaceStats.where(database: id).where("captured_at < ?", 90.days.ago).delete_all
+      def clean_space_stats(before: nil)
+        before ||= 90.days.ago
+        PgHero::SpaceStats.where(database: id).where("captured_at < ?", before).delete_all
       end
 
       def space_stats_enabled?

lib/tasks/pghero.rake

@@ -26,4 +26,12 @@ namespace :pghero do
     options[:before] = Integer(ENV["KEEP_DAYS"]).days.ago if ENV["KEEP_DAYS"].present?
     PgHero.clean_query_stats(**options)
   end
+
+  desc "Remove old space stats"
+  task clean_space_stats: :environment do
+    puts "Deleting old space stats..."
+    options = {}
+    options[:before] = Integer(ENV["KEEP_DAYS"]).days.ago if ENV["KEEP_DAYS"].present?
+    PgHero.clean_space_stats(**options)
+  end
 end