kotovalexarian-likes-github/database_cleaner
1
0
Fork 0
mirror of https://github.com/DatabaseCleaner/database_cleaner synced 2023-03-27 23:22:03 -04:00
Code Releases Activity

Merge pull request #119 from rsutphin/dm-pg-single-trunc

Browse source 
Use a single statement for DataMapper truncation on PostgreSQL
This commit is contained in:
Ben Mabey 2013-12-09 08:09:41 -08:00
commit 12ff049b19
1 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lib/database_cleaner/data_mapper/truncation.rb
View file

@ -10,6 +10,12 @@ module DataMapper
raise NotImplementedError
end
def truncate_tables(table_names)
table_names.each do |table_name|
adapter.truncate_table table_name
end
end
end
class MysqlAdapter < DataObjectsAdapter
@ -136,6 +142,12 @@ module DataMapper
execute("TRUNCATE TABLE #{quote_name(table_name)} RESTART IDENTITY CASCADE;")
end
# override to use a single statement
def truncate_tables(table_names)
quoted_names = table_names.collect { |n| quote_name(n) }.join(', ')
execute("TRUNCATE TABLE #{quoted_names} RESTART IDENTITY;")
end
# FIXME
# copied from activerecord
def supports_disable_referential_integrity?
@ -177,9 +189,7 @@ module DatabaseCleaner
def clean(repository = self.db)
adapter = ::DataMapper.repository(repository).adapter
adapter.disable_referential_integrity do
tables_to_truncate(repository).each do |table_name|
adapter.truncate_table table_name
end
adapter.truncate_tables(tables_to_truncate(repository))
end
end