在 Ruby 中按多个条件排序

Question

我有一个 Post 对象的集合，我希望能够根据这些条件对它们进行排序：

• 首先，按类别（新闻、事件、实验室、作品集等）
• 然后按日期（如果是日期）或按位置（如果为其设置了特定索引）

有些帖子会有日期（新闻和事件），其他帖子会有明确的职位（实验室和作品集）。

我希望能够调用posts.sort!，因此我已覆盖<=>，但我正在寻找根据这些条件进行排序的最有效方法。下面是一个伪方法：

def <=>(other)
  # first, everything is sorted into 
  # smaller chunks by category
  self.category <=> other.category

  # then, per category, by date or position
  if self.date and other.date
    self.date <=> other.date
  else
    self.position <=> other.position
  end
end

看来我实际上必须分别进行两次排序，而不是将所有内容都塞进那个方法中。比如sort_by_category，然后是sort!。最红宝石的方法是什么？

Answer 1

您应该始终按照相同的标准进行排序，以确保排序有意义。如果比较两个nil 日期，则position 将判断顺序很好，但如果将一个nil 日期与设置日期进行比较，您必须决定哪个先行，而不管位置如何（例如通过将nil 映射到过去的一天）。

否则想象如下：

a.date = nil                   ; a.position = 1
b.date = Time.now - 1.day      ; b.position = 2
c.date = Time.now              ; c.position = 0

按照您的原始标准，您将有：a

您还想立即进行排序。对于您的<=> 实现，请使用#nonzero?：

def <=>(other)
  return nil unless other.is_a?(Post)
  (self.category <=> other.category).nonzero? ||
  ((self.date || AGES_AGO) <=> (other.date || AGES_AGO)).nonzero? ||
  (self.position <=> other.position).nonzero? ||
  0
end

如果您只使用一次比较标准，或者如果该标准不是通用的，因此不想定义 <=>，您可以将 sort 与块一起使用：

post_ary.sort{|a, b| (a.category <=> ...).non_zero? || ... }

更好的是，您可以使用sort_by 和sort_by! 构建一个数组，以便比较哪些优先级：

post_ary.sort_by{|a| [a.category, a.date || AGES_AGO, a.position] }

除了更短之外，使用sort_by 的优点是您只能获得有序的条件。

注意事项：

• sort_by! 是在 Ruby 1.9.2 中引入的。您可以require 'backports/1.9.2/array/sort_by' 将其与旧版红宝石一起使用。
• 我假设Post 不是ActiveRecord::Base 的子类（在这种情况下，您希望由数据库服务器完成排序）。

Answer 2

或者，您可以一举在一个数组中进行排序，唯一的问题是处理其中一个属性为 nil 的情况，尽管如果您通过选择适当的 nil 保护知道数据集，仍然可以处理这种情况.此外，从您的伪代码中也不清楚日期和位置比较是按优先顺序还是按优先顺序列出的（即，如果两者都存在，则使用日期）。第一个解决方案假设使用，类别，后跟日期，然后是位置

def <=>(other)
    [self.category, self.date, self.position] <=> [other.category, other.date, other.position]
end

第二个假设是日期或位置

def <=>(other)
    if self.date && other.date
        [self.category, self.date] <=> [other.category, other.date]
    else
        [self.category, self.position] <=> [other.category, other.position]
    end
end