Extract `truncate` and `truncate_tables` into database statements

This is to easier make `truncate_tables` to bulk statements.

4 files changed, 5 insertions, 34 deletions

diff --git a/activerecord/lib/active_record/tasks/database_tasks.rb b/activerecord/lib/active_record/tasks/database_tasks.rb
index a8433fa0db..5a46003732 100644
--- a/activerecord/lib/active_record/tasks/database_tasks.rb
+++ b/activerecord/lib/active_record/tasks/database_tasks.rb
@@ -185,14 +185,17 @@ module ActiveRecord
       def truncate_tables(configuration)
         ActiveRecord::Base.connected_to(database: { truncation: configuration }) do
           table_names = ActiveRecord::Base.connection.tables
-          internal_table_names = [
+          table_names -= [
             ActiveRecord::Base.schema_migrations_table_name,
             ActiveRecord::Base.internal_metadata_table_name
           ]
 
-          class_for_adapter(configuration["adapter"]).new(configuration).truncate_tables(*table_names.without(*internal_table_names))
+          ActiveRecord::Base.connection.disable_referential_integrity do
+            ActiveRecord::Base.connection.truncate_tables(*table_names)
+          end unless table_names.empty?
         end
       end
+      private :truncate_tables
 
       def truncate_all(environment = env)
         ActiveRecord::Base.configurations.configs_for(env_name: environment).each do |db_config|
diff --git a/activerecord/lib/active_record/tasks/mysql_database_tasks.rb b/activerecord/lib/active_record/tasks/mysql_database_tasks.rb
index f2b4ead98d..1c1b29b5e1 100644
--- a/activerecord/lib/active_record/tasks/mysql_database_tasks.rb
+++ b/activerecord/lib/active_record/tasks/mysql_database_tasks.rb
@@ -31,16 +31,6 @@ module ActiveRecord
         connection.recreate_database configuration["database"], creation_options
       end
 
-      def truncate_tables(*table_names)
-        return if table_names.empty?
-
-        ActiveRecord::Base.connection.disable_referential_integrity do
-          table_names.each do |table_name|
-            ActiveRecord::Base.connection.truncate(table_name)
-          end
-        end
-      end
-
       def charset
         connection.charset
       end
diff --git a/activerecord/lib/active_record/tasks/postgresql_database_tasks.rb b/activerecord/lib/active_record/tasks/postgresql_database_tasks.rb
index dc368eb97d..8acb11f75f 100644
--- a/activerecord/lib/active_record/tasks/postgresql_database_tasks.rb
+++ b/activerecord/lib/active_record/tasks/postgresql_database_tasks.rb
@@ -48,18 +48,6 @@ module ActiveRecord
         create true
       end
 
-      def truncate_tables(*table_names)
-        return if table_names.empty?
-
-        ActiveRecord::Base.connection.disable_referential_integrity do
-          quoted_table_names = table_names.map do |table_name|
-            ActiveRecord::Base.connection.quote_table_name(table_name)
-          end
-
-          ActiveRecord::Base.connection.execute "TRUNCATE TABLE #{quoted_table_names.join(", ")}"
-        end
-      end
-
       def structure_dump(filename, extra_flags)
         set_psql_env
 
diff --git a/activerecord/lib/active_record/tasks/sqlite_database_tasks.rb b/activerecord/lib/active_record/tasks/sqlite_database_tasks.rb
index cedbae6b7f..a82cea80ca 100644
--- a/activerecord/lib/active_record/tasks/sqlite_database_tasks.rb
+++ b/activerecord/lib/active_record/tasks/sqlite_database_tasks.rb
@@ -33,16 +33,6 @@ module ActiveRecord
         create
       end
 
-      def truncate_tables(*table_names)
-        return if table_names.empty?
-
-        ActiveRecord::Base.connection.disable_referential_integrity do
-          table_names.each do |table_name|
-            ActiveRecord::Base.connection.truncate(table_name)
-          end
-        end
-      end
-
       def charset
         connection.encoding
       end