Add latest changes from gitlab-org/gitlab@13-9-stable-eev13.9.0-rc42

1 files changed, 31 insertions, 0 deletions

diff --git a/lib/gitlab/database/migrations/observers/total_database_size_change.rb b/lib/gitlab/database/migrations/observers/total_database_size_change.rb
new file mode 100644
index 00000000000..0b76b0bef5e
--- /dev/null
+++ b/lib/gitlab/database/migrations/observers/total_database_size_change.rb
@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Database
+    module Migrations
+      module Observers
+        class TotalDatabaseSizeChange < MigrationObserver
+          def before
+            @size_before = get_total_database_size
+          end
+
+          def after
+            @size_after = get_total_database_size
+          end
+
+          def record(observation)
+            return unless @size_after && @size_before
+
+            observation.total_database_size_change = @size_after - @size_before
+          end
+
+          private
+
+          def get_total_database_size
+            connection.execute("select pg_database_size(current_database())").first['pg_database_size']
+          end
+        end
+      end
+    end
+  end
+end