Class: Array

Inherits:

Object

• Object
• Array

Defined in:

lib/hashlib.rb

Instance Method Summary

• #count_distinct(group_by = []) ⇒ Object

Instance Method Details

#count_distinct(group_by = []) ⇒ Object

# File 'lib/hashlib.rb', line 183

def count_distinct(group_by=[])
  rv = {}

  self.each do |i|
    if i.is_a?(Hash)
      components = group_by
      components = components.collect{|j| i.get(j) }
      components = [*components.first].product([*components.last])

      components.each do |component|
        path = []
        stack = []

        component.compact.each do |c|
          if c.is_a?(Array)
            path += [stack].product(c.flatten).collect{|i| i.flatten }
            stack = []
          else
            stack << c
          end
        end

        path = [component] if path.empty?

        path.each do |parts|
          parts[-1] = :null if parts[-1].nil?
          rv.set(parts, rv.get(parts, 0) + 1)
        end
      end
    end
  end


  sum_children = proc do |sum, key, value|
    if value.is_a?(Hash)
      sum += value.inject(0){|s,(k,v)| sum_children.call(s,k,v) }
    else
      sum += value
    end

    sum
  end

  populate = proc do |rv, key, value|
    i = ({
      :id    => ((key.empty? or key.nil? or key == 'null') ? nil : key),
      :count => (value.is_a?(Hash) ? value.inject(0){|s,(k,v)| sum_children.call(s,k,v) } : value)
    })

    i[:children] = value.inject([]){|s,(k,v)| populate.call(s,k,v) } if value.is_a?(Hash)
    rv << i
    rv
  end

  rv.inject([]){|s,(k,v)| populate.call(s,k,v) }
end