ruby-on-rails - 更多红宝石方法来计算符合条件的对象

Question

这是我现在正在做的事情：

def get_counts
  products = Product.all
  a_count, b_count, c_count = 0, 0, 0
  products.collect{ |p| a_count+=1 if p.some_attribute == 'a' }
  products.collect{ |p| b_count+=1 if p.some_attribute == 'b' }
  products.collect{ |p| c_count+=1 if p.some_attribute == 'c' }
  return a_count, b_count, c_count
end

这对我来说感觉非常糟糕。我尝试使用注入，但无法让它按我想要的方式工作。有没有人有更好的方法来做到这一点？

Answer 1

改进@xdazz 的答案

def get_counts
  Product.where(some_attribute: ['a','b','c']).
          count(group: "some_attribute")
end

这将返回以下形式的哈希：

{'a' => 3, 'b' => 4, 'c' => 5}

Answer 2

def get_counts
  return Product.where(:some_attribute => 'a').count, Product.where(:some_attribute => 'b').count, Product.where(:some_attribute => 'c').count
end

如果您只需要一个查询，请使用 group by。