class PGError < StandardError; end unless defined?(PGError)
module PG
  class Error < StandardError; end unless defined?(::PG::Error)
end
class Mysql; class Error < StandardError; end; end unless defined?(Mysql)
module Mysql2; class Error < StandardError; end; end unless defined?(Mysql2)
module ActiveRecord
  class JDBCError < StandardError; end unless defined?(::ActiveRecord::JDBCError)
end

# :nocov:
unless defined?(::SQLite3::Exception)
  module SQLite3
    class Exception < StandardError; end
  end
end

module Rpush
  module Daemon
    module Store
      class ActiveRecord
        module Reconnectable
          ADAPTER_ERRORS = [::ActiveRecord::StatementInvalid, PGError, PG::Error,
                            Mysql::Error, Mysql2::Error, ::ActiveRecord::JDBCError,
                            SQLite3::Exception]

          def with_database_reconnect_and_retry
            ::ActiveRecord::Base.connection_pool.with_connection do
              yield
            end
          rescue *ADAPTER_ERRORS => e
            Rpush.logger.error(e)
            sleep_to_avoid_thrashing
            database_connection_lost
            retry
          end

          def database_connection_lost
            Rpush.logger.warn("Lost connection to database, reconnecting...")
            attempts = 0
            loop do
              begin
                Rpush.logger.warn("Attempt #{attempts += 1}")
                reconnect_database
                check_database_is_connected
                break
              rescue *ADAPTER_ERRORS => e
                Rpush.logger.error(e)
                sleep_to_avoid_thrashing
              end
            end
            Rpush.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.
            Rpush::Client::ActiveRecord::Notification.count
          end

          def sleep_to_avoid_thrashing
            sleep 2
          end
        end
      end
    end
  end
end