Ruby on Rails：干燥初始化一些变量的重复代码块

Question

我有一个重复的代码块，它在一堆不同的控制器方法中初始化一些变量。有没有办法让我使用模型方法进行 DRY，而不是在每个控制器方法中重复相同的代码块？

基本上，它用于社交网站，它会提取用户的朋友列表，然后根据用户拥有的存储在友谊模型中的权限构建朋友桶。这种桶的重复初始化正是我试图做到的 DRY。

通常我会使用模型方法，但在这种情况下，3 个单独的变量将基于一次数据库拉取进行初始化，而且调用频率足够高，我不想通过访问数据库 3 次来使其不必要的低效。在 C 中，我只使用作为变量传入的指针。

事情是这样的：

def example_method
  friendships = @user.friendships
  view_permission_friends = []
  write_permission_friends = []
  message_permission_friends = []
  for friendship in friendships
    if friendship.view_permission then view_permission_friends << friendship.friend_id end
    if friendship.write_permission then write_permission_friends << friendship.friend_id end
    if friendship.message_permission then message_permission_friends << friendship.friend_id end
  end
  #Do something with the 3 initialized arrays here
end

Answer 1

我想了一下，我认为这段代码应该进入你的用户模型。 （或者上面示例中的 @user 类。）两个原因：

1. 它对于用户及其关系非常具体，最重要的是，它们如何存储在数据库中以及如何从数据库中检索。
2. 您只需编写一次代码：在用户模型中。

依赖于 Rails 内部查询缓存的快速而简单的方法如下所示。添加到用户模型中：

def view_permission_friends
  return friendships.select{|f| f.view_permission}
end

(etc.)

然后您的控制器只需执行以下操作：（

@viewers = @user.view_permission_friends
(etc.)

注意 - 通过延迟缓存和参数化权限，这里有更多的优化空间和更好的灵活性。）

Answer 2

怎么样...

rv = {}
%w(view write message).each do |type|
  rv["#{type}_permission_friends"] = @user.friendships.select{|f|f.send(:"#{type}_permission")}.collect(&:friend_id)
end

这给你一个带有键的散列而不是单独的数组，但应该足够了。

Answer 3

使用Enumerable#inject，Enumerable#find_all 和 Object#instance_variable_set：

def example_method
  %w{view write message}.inject({}) do |memo, s|
    friendships = @user.friendships.find_all{ |f| f.send("#{s}_permission") }
    memo["#{s}_permission_friends"] = friendships
  end.each do |key, value|
    instance_variable_set "@#{key}", value
  end

  # now you have 3 arrays:
  # @view_permission_friends, @write_permission_friends and @message_permission_friends
end