rapns 3.2.0-java → 3.3.0-java

data/lib/rapns/daemon/database_reconnectable.rb

@@ -1,57 +0,0 @@
-class PGError < StandardError; end if !defined?(PGError)
-class Mysql; class Error < StandardError; end; end if !defined?(Mysql)
-module Mysql2; class Error < StandardError; end; end if !defined?(Mysql2)
-module ActiveRecord; end
-class ActiveRecord::JDBCError < StandardError; end if !defined?(ActiveRecord::JDBCError)
-
-module Rapns
-  module Daemon
-    module DatabaseReconnectable
-      ADAPTER_ERRORS = [ActiveRecord::StatementInvalid, PGError, Mysql::Error,
-                        Mysql2::Error, ActiveRecord::JDBCError]
-
-      def with_database_reconnect_and_retry
-        begin
-          ActiveRecord::Base.connection_pool.with_connection do
-            yield
-          end
-        rescue *ADAPTER_ERRORS => e
-          Rapns.logger.error(e)
-          database_connection_lost
-          retry
-        end
-      end
-
-      def database_connection_lost
-        Rapns.logger.warn("Lost connection to database, reconnecting...")
-        attempts = 0
-        loop do
-          begin
-            Rapns.logger.warn("Attempt #{attempts += 1}")
-            reconnect_database
-            check_database_is_connected
-            break
-          rescue *ADAPTER_ERRORS => e
-            Rapns.logger.error(e, :airbrake_notify => false)
-            sleep_to_avoid_thrashing
-          end
-        end
-        Rapns.logger.warn("Database reconnected")
-      end
-
-      def reconnect_database
-        ActiveRecord::Base.clear_all_connections!
-        ActiveRecord::Base.establish_connection
-      end
-
-      def check_database_is_connected
-        # Simply asking the adapter for the connection state is not sufficient.
-        Rapns::Notification.count
-      end
-
-      def sleep_to_avoid_thrashing
-        sleep 2
-      end
-    end
-  end
-end