From 6a60446eca3c2d8940458e81019e53526663f7c2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Hasan=20=C3=96zoval=C4=B1?= <ozovalihasan@gmail.com>
Date: Mon, 14 Jul 2025 16:24:50 +0300
Subject: [PATCH] Fix a result of a raw SQL executed in MySQL2

---
 db/migrate/20250606094103_drop_legacy_models.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/db/migrate/20250606094103_drop_legacy_models.rb b/db/migrate/20250606094103_drop_legacy_models.rb
index 1d0e05e5d98..d260c8b070f 100644
--- a/db/migrate/20250606094103_drop_legacy_models.rb
+++ b/db/migrate/20250606094103_drop_legacy_models.rb
@@ -2,12 +2,12 @@ class DropLegacyModels < ActiveRecord::Migration[7.2]
   def up
     # Check if data migration is complete
     if DataMigrationStatus.find_by(name: "push_model_migration")&.completed? ||
-        ActiveRecord::Base.connection.execute(<<-SQL).first["total_count"]&.to_i == 0
+        ActiveRecord::Base.connection.select_value(<<-SQL) == 0
          SELECT (
            (SELECT COUNT(*) FROM passwords) +
            (SELECT COUNT(*) FROM urls) +
            (SELECT COUNT(*) FROM file_pushes)
-         ) as total_count
+         )
         SQL
       drop_table :passwords, force: :cascade
       drop_table :urls, force: :cascade