rails数据库记录问题

Question

这是我在模型文件中的查询方法：

def self.sum_by_brand_category
  result = Product.joins(:brand, :category)
           .select("brands.id as brand_id, categories.id as category_id, sum(products.quantity) as count")
           .group("brands.id, categories.id")
  return result
end

这是我得到的示例数据库查询结果：

[
  {
    "id":null,
    "brand_id":43,
    "category_id":1,
    "count":2
  },
  {
    "id":null,
    "brand_id":43,
    "category_id":2,
    "count":5
  },
  {
    "id":null,
    "brand_id":43,
    "category_id":3,
    "count":4
  },
  ....
]

我希望在视图中使用的最终 JSON 结果应该是这样的：

[
  {
    "id":null,
    "brand_id":43,
    "quantity": [
      {
        "category_id": 1,
        "count": 2
      },
      {
        "category_id": 2,
        "count": 5
      },
      {
        "category_id": 3,
        "count": 4
      }
    ]
  },
  ....
]

我怎样才能实现它？改变模型方法？在将结果发送到视图之前在控制器中重建结果？如何？

任何建议将不胜感激。谢谢你。

---

更新：

根据@cmrichards 的回答，我想出了这个私有方法，在控制器中调用，然后在视图中使用。我在这里包括我的工作，虽然这些代码不是那么干燥：

private
  def get_sum_by_brand_category
    query_results = Product.sum_by_brand_category

    results = []

    query_results.group_by(&:brand_id).each do |brand_id, query_result|
      result = {}
      result[:id] = nil
      result[:brand_id] = brand_id
      quantity_array = []
      query_result.each do |data|
        quantity_block = {}
        quantity_block[:category_id] = data.category_id
        quantity_block[:count] = data.count
        quantity_array.push(quantity_block)
      end
      result[:quantity] = quantity_array

      results.push(result)
    end

    return results
  end

如果您愿意，请通过编辑我的问题将它们弄干。 ;)

Answer 1

一种选择是使用 jBuilder，这是一个由 Rails 团队维护的流行 gem，它提供了一种简单的 DSL（特定于域的语言），允许您在视图中定义 JSON 结构。

https://github.com/rails/jbuilder

使用您现有的方法，您可以这样做

控制器

def index
    @data = Product.sum_by_brand_category    
end

views/products/index.json.jbuilder

json.array! @data.group_by(&:brand_id) do |grouped|
  brand_id, category_data = grouped
  json.id nil
  json.brand_id brand_id
  json.quantity category_data, :category_id, :count
end