如何获取关联对象的current_index？及元素在关联中的索引查询

嘿，针对你提到的两个关于关联对象索引的需求，我整理了具体的实现方案，一起来看看吧：

需求1：获取关联对象的current_index #

如果是在遍历关联集合的场景下，直接用Rails提供的each_with_index方法就能轻松拿到每个元素的当前索引：

# 遍历post的所有评论，同时获取索引
post.comments.each_with_index do |comment, current_index|
  puts "评论 #{comment.id} 的当前索引是: #{current_index}"
end

这个方法会在遍历过程中自动把当前元素的索引传给你定义的变量，非常直观。

需求2：查询元素在关联对象中的索引（实现which_index?方法） #

要实现像post.comments.find(2).which_index?这样调用并返回对应索引的功能，你可以在关联模型（比如Comment）中添加一个自定义方法：

class Comment < ApplicationRecord
  belongs_to :post

  def which_index?
    # 先获取当前post下所有评论的id数组，再找到当前评论id的位置
    post.comments.pluck(:id).index(self.id)
  end
end

验证示例 #

按照你给出的场景：

post.comments.pluck(:id) => [1, 2, 3]
# 调用方法
post.comments.find(2).which_index? # 返回1，和预期一致

额外优化建议 #

• 如果你的关联集合有自定义排序，确保pluck(:id)的顺序和实际关联集合的顺序一致（默认情况下是一致的，除非你在关联定义里加了order）。
• 可以处理元素不在集合中的情况，避免返回nil：

def which_index?
  idx = post.comments.map(&:id).index(self.id)
  idx || -1 # 找不到时返回-1，你可以根据需求调整默认值
end

内容的提问来源于stack exchange，提问作者lili pupu