From c7dc4003730765bbb50fa16d5faf7cc633ce036b Mon Sep 17 00:00:00 2001
From: Dmitry Gruzd <dgruzd@gitlab.com>
Date: Mon, 19 Feb 2024 12:20:17 +0100
Subject: [PATCH] Fix regenerate-schema for partitioned tables

---
 scripts/regenerate-schema | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/scripts/regenerate-schema b/scripts/regenerate-schema
index 75ee0c33beab..7ea89488f09e 100755
--- a/scripts/regenerate-schema
+++ b/scripts/regenerate-schema
@@ -58,6 +58,7 @@ class SchemaRegenerator
       setup_db
       unhide_migrations
       migrate
+      dump_schema
       rollback if @rollback_testing
     ensure
       unhide_migrations
@@ -195,6 +196,12 @@ class SchemaRegenerator
     run %q(bin/rails db:migrate RAILS_ENV=test)
   end
 
+  ##
+  # Run rake task to dump schema.
+  def dump_schema
+    run %q(bin/rails db:schema:dump RAILS_ENV=test)
+  end
+
   ##
   # Run rake task to rollback migrations.
   def rollback
-- 
GitLab