Включение пользовательского предложения SELECT в запрос ActiveRecord

Question

У меня есть большая таблица оценок, в этой таблице есть несколько целочисленных столбцов с числовыми баллами. Я пытаюсь удалить N+1 итерацию, чтобы ускорить мое приложение.

Я могу сгенерировать этот запрос в сыром SQL, чтобы собрать все оценки

query = 
  'SELECT ' + 
  Assessment.rating_attributes.collect{|attribute|
    1.upto(5).collect do |i|
      %Q{SUM(CASE WHEN #{attribute.to_s} = #{i} then 1 else 0 end) as #{attribute}_score_#{i}}
    end
    }.join(', ') + ' FROM assessments'

Когда я выполняю этот запрос с:

  $> ActiveRecord::Base.connection.execute(query)
 => #<PG::Result:0x007fc9e7e541e8 status=PGRES_TUPLES_OK ntuples=1 nfields=55 cmd_tuples=1>

Я получаю правильный результат:

 ActiveRecord::Base.connection.execute(query).first.values
 => ["2400", "458", "3343", "1021", "93", "2352", "455", "3119", "1174", "150", "2378", "357", "2976", "1357", "181", "2042", "258", "3024", "1646", "280", "1274", "193", "3907", "1704", "172", "799", "93", "4593", "1670", "95", "1759", "361", "2542", "947", "69", "21", "100", "81", "42", "38", "8", "32", "51", "78", "112", "19", "119", "72", "43", "29", "0", "0", "0", "0", "0"]

Я хотел бы выполнить этот же запрос, но не по всей таблице. Я хотел бы иметь возможность связать его в запрос activerecord, например:

Assessment.where(selection_attribute: 1038).select(query_without_select_keyword_or_from_clause)

ActiveRecord::StatementInvalid: PG::GroupingError: ERROR:  column "assessments.id" must appear in the GROUP BY clause or be used in an aggregate function

и дать мне такой же результат, но на подмножестве. Мои наивные усилия (например, вызов.select(query) в моем примере) потерпели неудачу, и я даже не уверен, как подойти к этому.

Я врезался в стену, и я был бы признателен за любые рекомендации, даже о том, как выработать поиск решения. Спасибо.

Answer 1

Ну, это было просто. С помощью .first была проблема. Я предполагаю, что это потому, что нет записей модели, возвращаемых запросом. Если я обращаюсь непосредственно к индексу 0 результата, я получаю все свои оценки.

так что этот код работает:

 query = 
      Assessment.rating_attributes.collect{|attribute|
        1.upto(5).collect do |i|
        %Q{sum(CASE WHEN #{attribute.to_s} = #{i} then 1 else 0 end) as #{attribute}_score_#{i}}
      end
    }.join(', ')

Assessment.where(selection_attribute: 1038).select(query)[0]
=> #<Assessment id: nil>

Assessment.where(selection_attribute: 1038).select(query)[0].first_attribute_score_1
=> 3