BigW Consortium Gitlab

connection_fix.rb 932 Bytes
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
# from http://gist.github.com/238999
#
# If your workers are inactive for a long period of time, they'll lose
# their MySQL connection.
#
# This hack ensures we re-connect whenever a connection is
# lost. Because, really. why not?
#
# Stick this in RAILS_ROOT/config/initializers/connection_fix.rb (or somewhere similar)
#
# From:
#   http://coderrr.wordpress.com/2009/01/08/activerecord-threading-issues-and-resolutions/

if defined?(ActiveRecord::ConnectionAdapters::Mysql2Adapter)
  module ActiveRecord::ConnectionAdapters
    class Mysql2Adapter
      alias_method :execute_without_retry, :execute

      def execute(*args)
        execute_without_retry(*args)
      rescue ActiveRecord::StatementInvalid => e
        if e.message =~ /server has gone away/i
23
          warn "Lost connection to MySQL server during query"
24 25 26 27 28 29 30 31 32
          reconnect!
          retry
        else
          raise e
        end
      end
    end
  end
end