From a30fd8e3f6ecec854652efcb5320290ff99c0167 Mon Sep 17 00:00:00 2001
From: Manoj M J <mmj@gitlab.com>
Date: Wed, 29 Nov 2023 11:27:39 +0100
Subject: [PATCH] Make changes to LFK script

---
 scripts/decomposition/generate-loose-foreign-key | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scripts/decomposition/generate-loose-foreign-key b/scripts/decomposition/generate-loose-foreign-key
index d52fa2b4f3f96..56d19cb3aa1d1 100755
--- a/scripts/decomposition/generate-loose-foreign-key
+++ b/scripts/decomposition/generate-loose-foreign-key
@@ -161,7 +161,8 @@ def generate_migration(definition)
   content = <<-EOF.strip_heredoc
     # frozen_string_literal: true
 
-    class Remove#{definition.to_table.camelcase}#{definition.from_table.camelcase}#{definition.column.camelcase}Fk < Gitlab::Database::Migration[2.1]
+    class Remove#{definition.to_table.camelcase}#{definition.from_table.camelcase}#{definition.column.camelcase}Fk < Gitlab::Database::Migration[#{Gitlab::Database::Migration.current_version}]
+      milestone '#{Gitlab.current_milestone}'
       disable_ddl_transaction!
 
       FOREIGN_KEY_NAME = "#{definition.name}"
-- 
GitLab