module Mongoid::Acts::NestedSet::Rebuild
Public Instance Methods
rebuild!(options = {})
click to toggle source
Rebuilds the left & rights if unset or invalid. Also very useful for converting from acts_as_tree. Warning: Very expensive!
# File lib/mongoid_nested_set/rebuild.rb, line 7 def rebuild!(options = {}) # Don't rebuild a valid tree. return true if valid? scope = lambda{ |node| {} } if acts_as_nested_set_options[:scope] scope = lambda { |node| node.nested_set_scope.options.merge(node.nested_set_scope.selector) } end indices = {} set_left_and_rights = lambda do |node| # set left left = (indices[scope.call(node)] += 1) # find node.nested_set_scope.where(parent_field_name => node.id).asc(left_field_name).asc(right_field_name).each { |n| set_left_and_rights.call(n) } # set right right = (indices[scope.call(node)] += 1) q = node.class.collection.find(:_id => node.id) set = {"$set" => {left_field_name => left, right_field_name => right}} if q.respond_to?(:update_one) q.update_one(set) else q.update(set) end end # scope rebuild for specified node tree_scope = self.where(parent_field_name => nil) if acts_as_nested_set_options[:scope] && options[:scope] tree_scope = tree_scope.where( acts_as_nested_set_options[:scope] => options[:scope] ) end # Find root node(s) root_nodes = tree_scope.asc(left_field_name).asc(right_field_name).asc(:_id).each do |root_node| # setup index for this scope indices[scope.call(root_node)] ||= 0 set_left_and_rights.call(root_node) end end