Rails filtrando matriz de objetos por valor de atributo

95

Então, eu executo uma consulta ao banco de dados e tenho uma matriz completa de objetos:

@attachments = Job.find(1).attachments

Agora que tenho uma matriz de objetos, não quero realizar outra consulta de banco de dados, mas gostaria de filtrar a matriz com base nos Attachmentobjetos file_typepara que eu possa ter uma lista de attachmentsonde está o tipo de arquivo 'logo'e, em seguida, outra lista de attachmentsonde o tipo de arquivo é'image'

Algo assim:

@logos  = @attachments.where("file_type = ?", 'logo')
@images = @attachments.where("file_type = ?", 'image')

Mas na memória, em vez de uma consulta db.

ruby-on-rails activerecord

— joepour
fonte

Parece um bom caso de uso partition- por exemplo, aqui .

— SRack

172

Experimentar :

Isto é bom :

@logos = @attachments.select { |attachment| attachment.file_type == 'logo' }
@images = @attachments.select { |attachment| attachment.file_type == 'image' }

mas em termos de desempenho, você não precisa iterar @attachments duas vezes:

@logos , @images = [], []
@attachments.each do |attachment|
  @logos << attachment if attachment.file_type == 'logo'
  @images << attachment if attachment.file_type == 'image'
end

— Vik
fonte

2

Como a solução de @Vik é bastante ideal, acrescentarei que, em casos binários, você poderia usar uma função de 'partição' para tornar as coisas mais fáceis. ruby-doc.org/core-1.9.3/Enumerable.html#method-i-partition

— Vlad

Obrigado @Vlad, isso é legal, mas dá suporte apenas se precisarmos coletar apenas duas coisas do objeto.

— Vik de

1

Sim, é por isso que disse "binário" :). Na pergunta, aparentemente havia uma escolha de logotipo ou imagem, então eu adicionei isso para completar.

— Vlad

8

Se seus anexos são

@attachments = Job.find(1).attachments

Esta será uma matriz de objetos de anexo

Use o método select para filtrar com base em file_type.

@logos = @attachments.select { |attachment| attachment.file_type == 'logo' }
@images = @attachments.select { |attachment| attachment.file_type == 'image' }

Isso não acionará nenhuma consulta de banco de dados.

— Soundar Rathinasamy
fonte

2

você já tentou carregar antecipadamente?

@attachments = Job.includes(:attachments).find(1).attachments

— Siwei Shen 申思维
fonte

Desculpe, não estou sendo claro: como faço para filtrar pelo valor de um atributo de objeto sem percorrer a matriz?

— joepour

Se bem entendi, você quer menos consulta de banco de dados, especialmente, uma vez que uma consulta como @attachments = Job.first.attachmentsexecutada, você deseja fazer um loop, @attachmentsenquanto isso, você não quer mais consultas de banco de dados. é isso que você quer fazer?

— Siwei Shen 申思维

Eu faço uma consulta de banco de dados e recebo uma matriz de objetos. Em seguida, quero criar duas listas separadas desse array filtrando os objetos com base no valor de seus atributos (veja a postagem original)

— saúde

0

Você pode filtrar usando onde

Job.includes(:attachments).where(file_type: ["logo", "image"])

— Darlan Dieterich
fonte

0

Eu faria isso de maneira um pouco diferente. Estruture sua consulta para recuperar apenas o que você precisa e divida a partir daí.

Portanto, faça sua consulta da seguinte maneira:

#                                vv or Job.find(1) vv
attachments = Attachment.where(job_id: @job.id, file_type: ["logo", "image"])
# or 
Job.includes(:attachments).where(id: your_job_id, attachments: { file_type: ["logo", "image"] })

E então particionar os dados:

@logos, @images = attachments.partition { |attachment| attachment.file_type == "logo" }

Isso obterá os dados que você procura de maneira organizada e eficiente.

— SRack
fonte