module ActionDispatch::Session::StaleSessionCheck

def stale_session_check!
  yield
rescue ArgumentError => argument_error
  if argument_error.message =~ %r{undefined class/module ([\w:]*\w)}
    begin
      # Note that the regexp does not allow $1 to end with a ':'
      $1.constantize
    rescue LoadError, NameError => e
      raise ActionDispatch::Session::SessionRestoreError, e, e.backtrace
    end
    retry
  else
    raise
  end
end