Handle Timeout::Error exceptions which are caught by ActiveRecord.

[rails.git] / config / initializers / abstract_adapter.rb

diff --git a/config/initializers/abstract_adapter.rb b/config/initializers/abstract_adapter.rb
index 7b22a3cc6ccbd95cf5deda3203de5bbb87985d05..fbae428cbbc5cc9b2dbdbfef312db0694ea90f27 100644 (file)
--- a/config/initializers/abstract_adapter.rb
+++ b/config/initializers/abstract_adapter.rb
@@ -13,8 +13,10 @@ module ActiveRecord
           old_log(sql, name)
         end
       rescue ActiveRecord::StatementInvalid => ex
-        if ex =~ /^OSM::APITimeoutError: /
-          raise OSM::APITimeoutError
+        if ex.message =~ /^OSM::APITimeoutError: /
+          raise OSM::APITimeoutError.new
+        elsif ex.message =~ /^Timeout::Error: /
+          raise Timeout::Error.new
         else
           raise
         end