From fdac932707fb16b7d074b1d0bc7c255157d72675 Mon Sep 17 00:00:00 2001 From: Ryuta Kamizono Date: Sun, 10 Mar 2019 01:53:48 +0900 Subject: Extract `truncate` and `truncate_tables` into database statements This is to easier make `truncate_tables` to bulk statements. --- .../connection_adapters/postgresql/database_statements.rb | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'activerecord/lib/active_record/connection_adapters/postgresql') 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" -- cgit v1.2.3