Module: CollectiveIdea::Acts::NestedSet::Model::ClassMethods
- Defined in:
- lib/awesome_nested_set/awesome_nested_set.rb
Instance Method Summary collapse
-
#all_roots_valid? ⇒ Boolean
Wrapper for each_root_valid? that can deal with scope.
- #each_root_valid?(roots_to_validate) ⇒ Boolean
-
#each_with_level(objects) ⇒ Object
Iterates over tree elements and determines the current level in the tree.
- #left_and_rights_valid? ⇒ Boolean
- #no_duplicates_for_columns? ⇒ Boolean
-
#rebuild!(validate_nodes = true) ⇒ Object
Rebuilds the left & rights if unset or invalid.
-
#root ⇒ Object
Returns the first root.
- #valid? ⇒ Boolean
Instance Method Details
#all_roots_valid? ⇒ Boolean
Wrapper for each_root_valid? that can deal with scope.
135 136 137 138 139 140 141 142 143 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 135 def all_roots_valid? if [:scope] roots.group(scope_column_names).group_by{|record| scope_column_names.collect{|col| record.send(col.to_sym)}}.all? do |scope, grouped_roots| each_root_valid?(grouped_roots) end else each_root_valid?(roots) end end |
#each_root_valid?(roots_to_validate) ⇒ Boolean
145 146 147 148 149 150 151 152 153 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 145 def each_root_valid?(roots_to_validate) left = right = 0 roots_to_validate.all? do |root| (root.left > left && root.right > right).tap do left = root.left right = root.right end end end |
#each_with_level(objects) ⇒ Object
Iterates over tree elements and determines the current level in the tree. Only accepts default ordering, odering by an other column than lft does not work. This method is much more efficent than calling level because it doesn’t require any additional database queries.
Example:
Category.each_with_level(Category.root.self_and_descendants) do |o, level|
197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 197 def each_with_level(objects) path = [nil] objects.each do |o| if o.parent_id != path.last # we are on a new level, did we decent or ascent? if path.include?(o.parent_id) # remove wrong wrong tailing paths elements path.pop while path.last != o.parent_id else path << o.parent_id end end yield(o, path.length - 1) end end |
#left_and_rights_valid? ⇒ Boolean
107 108 109 110 111 112 113 114 115 116 117 118 119 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 107 def left_and_rights_valid? joins("LEFT OUTER JOIN #{quoted_table_name} AS parent ON " + "#{quoted_table_name}.#{quoted_parent_column_name} = parent.#{primary_key}"). where( "#{quoted_table_name}.#{quoted_left_column_name} IS NULL OR " + "#{quoted_table_name}.#{quoted_right_column_name} IS NULL OR " + "#{quoted_table_name}.#{quoted_left_column_name} >= " + "#{quoted_table_name}.#{quoted_right_column_name} OR " + "(#{quoted_table_name}.#{quoted_parent_column_name} IS NOT NULL AND " + "(#{quoted_table_name}.#{quoted_left_column_name} <= parent.#{quoted_left_column_name} OR " + "#{quoted_table_name}.#{quoted_right_column_name} >= parent.#{quoted_right_column_name}))" ).count == 0 end |
#no_duplicates_for_columns? ⇒ Boolean
121 122 123 124 125 126 127 128 129 130 131 132 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 121 def no_duplicates_for_columns? scope_string = Array([:scope]).map do |c| connection.quote_column_name(c) end.push(nil).join(", ") [quoted_left_column_name, quoted_right_column_name].all? do |column| # No duplicates select("#{scope_string}#{column}, COUNT(#{column})"). group("#{scope_string}#{column}"). having("COUNT(#{column}) > 1"). first.nil? end end |
#rebuild!(validate_nodes = true) ⇒ Object
Rebuilds the left & rights if unset or invalid. Also very useful for converting from acts_as_tree.
157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 157 def rebuild!(validate_nodes = true) # Don't rebuild a valid tree. return true if valid? scope = lambda{|node|} if [:scope] scope = lambda{|node| scope_column_names.inject(""){|str, column_name| str << "AND #{connection.quote_column_name(column_name)} = #{connection.quote(node.send(column_name.to_sym))} " } } end indices = {} set_left_and_rights = lambda do |node| # set left node[left_column_name] = indices[scope.call(node)] += 1 # find where(["#{quoted_parent_column_name} = ? #{scope.call(node)}", node]).order("#{quoted_left_column_name}, #{quoted_right_column_name}, id").each{|n| set_left_and_rights.call(n) } # set right node[right_column_name] = indices[scope.call(node)] += 1 node.save!(:validate => validate_nodes) end # Find root node(s) root_nodes = where("#{quoted_parent_column_name} IS NULL").order("#{quoted_left_column_name}, #{quoted_right_column_name}, 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 |
#root ⇒ Object
Returns the first root
99 100 101 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 99 def root roots.first end |
#valid? ⇒ Boolean
103 104 105 |
# File 'lib/awesome_nested_set/awesome_nested_set.rb', line 103 def valid? left_and_rights_valid? && no_duplicates_for_columns? && all_roots_valid? end |