Extract `truncate` and `truncate_tables` into database statements

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

1 files changed, 6 insertions, 0 deletions

diff --git a/activerecord/lib/active_record/connection_adapters/postgresql/database_statements.rb b/activerecord/lib/active_record/connection_adapters/postgresql/database_statements.rb
index 41633872e2..208934385f 100644
--- a/activerecord/lib/active_record/connection_adapters/postgresql/database_statements.rb
+++ b/activerecord/lib/active_record/connection_adapters/postgresql/database_statements.rb
@@ -143,6 +143,12 @@ module ActiveRecord
           end
         end
 
+        def truncate_tables(*table_names) # :nodoc:
+          unless table_names.empty?
+            execute "TRUNCATE TABLE #{table_names.map(&method(:quote_table_name)).join(", ")}"
+          end
+        end
+
         # Begins a transaction.
         def begin_db_transaction
           execute "BEGIN"