Fixing 'server has gone away' issue in MySQL2 Gem

Using the MySQL2 gem? Getting the error ‘server has gone away’? Then read on my friend!

After spending WAY too much time trying to find an answer for the correct fix for the error ActiveRecord::StatementInvalid Mysql2::Error: MySQL server has gone away: I eventually came across this answer.

Put the following code into a file config/initializers/mysql_fix.rb:

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

This is only a slight tweak from the MySQL gem version, but properly works for the MySQL2 gem.