Skip to content

Instantly share code, notes, and snippets.

@alispat
Forked from mauricioszabo/connection_fix.rb
Created March 15, 2012 12:34
Show Gist options
  • Save alispat/2043981 to your computer and use it in GitHub Desktop.
Save alispat/2043981 to your computer and use it in GitHub Desktop.
MySQL server has gone away fix
# 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/
#Prevents Rails from loading multiple times. I could use a global variable too, but still...
if !$LOADED_FEATURES.include?(__FILE__)
if ActiveRecord::ConnectionAdapters.const_defined?(:MysqlAdapter)
module ActiveRecord::ConnectionAdapters
class MysqlAdapter
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
warn "Server timed out, retrying"
reconnect!
retry
else
raise e
end
end
end
end
elsif ActiveRecord::ConnectionAdapters.const_defined?(: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
warn "Server timed out, retrying"
reconnect!
retry
else
raise e
end
end
end
end
end
end
#Prevents Rails from loading multiple times. I could use a global variable too, but still...
$LOADED_FEATURES << __FILE__
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment