jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

refactored with_timeout helper

Browse Source
This commit is contained in:
albert
2013-01-30 15:04:13 -08:00
parent c6d4500c91
commit c6b9de8486
3 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
config/initializers/active_record_extensions.rb
View File

@@ -8,14 +8,16 @@ module Danbooru
connection.execute("SET STATEMENT_TIMEOUT = 0") unless Rails.env == "test"
yield
ensure
connection.execute("SET STATEMENT_TIMEOUT = 5000") unless Rails.env == "test"
connection.execute("SET STATEMENT_TIMEOUT = 3000") unless Rails.env == "test"
end
def with_timeout(n)
def with_timeout(n, default_value)
connection.execute("SET STATEMENT_TIMEOUT = #{n}") unless Rails.env == "test"
yield
rescue ActiveRecord::StatementInvalid
return default_value
ensure
connection.execute("SET STATEMENT_TIMEOUT = 5000") unless Rails.env == "test"
connection.execute("SET STATEMENT_TIMEOUT = 3000") unless Rails.env == "test"
end
end