Class: Immunio::ArelNodeVisitor

Inherits:
Arel::Visitors::Visitor
  • Object
show all
Defined in:
lib/immunio/plugins/active_record.rb

Overview

Arel AST visitor to collect params and modifiers. Based on Arel::Visitors::DepthFirst.

Whenever a query is built in Rails, a tree of Ruby objects (AST) is built to represent that query. Arel (the engine building this tree) uses a Visitor to build the SQL statement represented by tree. We use the same base class as Arel, but instead of building some SQL, we track the params and modifiers.

See en.wikipedia.org/wiki/Visitor_pattern

Constant Summary collapse

VISITABLES =

Only accepts statements to avoid duplicates in params if branches are visited multiple times.

[Arel::Nodes::SelectStatement, Arel::Nodes::InsertStatement,
Arel::Nodes::UpdateStatement, Arel::Nodes::DeleteStatement].freeze
IGNORED_EXPRESSIONS = 
["", "*", "1 AS one"].freeze
DISPATCH =

Copied from Arel::Visitors::Visitor to ensure the cached dispatch table isn’t shared with other Arel visitors under Rails 3.2.

Hash.new do |hash, klass|
  hash[klass] = "visit_#{(klass.name || '').gsub('::', '_')}"
end

Instance Method Summary collapse

Constructor Details

#initialize(connection_id) ⇒ ArelNodeVisitor

Returns a new instance of ArelNodeVisitor.



114
115
116
117
 
# File 'lib/immunio/plugins/active_record.rb', line 114

def initialize(connection_id)
  @connection_id = connection_id
  super()
end

Instance Method Details

#accept(object) ⇒ Object

Entry point into the visitor.



120
121
122
123
124
 
# File 'lib/immunio/plugins/active_record.rb', line 120

def accept(object)
  if VISITABLES.include?(object.class)
    visit object, {}
  end
end

#dispatch ⇒ Object 



126
127
128
 
# File 'lib/immunio/plugins/active_record.rb', line 126

def dispatch
  DISPATCH
end