使方法调用可链接

Question

我有一个 Ruby 类，它上面的每个方法都根据特定条件保存哈希数组的索引。

例如（代码自原始发布以来已被编辑）

module Dronestream
  class Strike

    class << self

    ...
    def strike
      @strike ||= all
    end

    def all
      response['strike'] # returns an array of hashes, each individual strike
    end

    def in_country(country)
      strike.keep_if { |strike| strike['country'] == country }
      self
    end

    def in_town(town)
      strike.keep_if { |strike| strike['town'] == town }
      self
    end
    ...
  end
end  

这样，你可以做Dronestream::Strike.in_country('Yemen')，或者Dronestream::Strike.in_town('Taizz')，每个都返回一个数组。但我希望能够做到Dronestream::Strike.in_country('Yemen').in_town('Taizz')，并让它只返回也门那个城镇的罢工。

但到目前为止，每个单独的方法都返回一个数组。我知道如果我让他们返回self，他们就会有我需要的方法。但是它们不会返回一个数组，我也不能调用它们，例如，first 或each，就像我可以调用一个数组一样，我需要这样做。我试图使Strike < Array，但是，first 是Array 上的实例方法，而不是类方法。

我该怎么办？

编辑

这是我的测试套件的一部分。按照下面的答案，测试单独通过，但随后失败。

describe Dronestream::Strike do

  let(:strike) { Dronestream::Strike }

  before :each do
    VCR.insert_cassette 'strike', :record => :new_episodes
    @strike = nil
  end

  after do
    VCR.eject_cassette
  end
  ...
  # passes when run by itself and when the whole file runs together
  describe '#country' do
    let(:country_name) { 'Yemen' }
    it 'takes a country and returns strikes from that country' do
      expect(strike.in_country(country_name).first['country']).to eq(country_name)
    end
  end

  # passes when run by itself, but fails when the whole file runs together
  describe '#in_town' do
    let(:town_name) { 'Wadi Abida' }
    it 'returns an array of strikes for a given town' do
      expect(strike.in_town(town_name).first['town'].include?(town_name)).to be_true 
    end
  end
  ...
end

Answer 1

您可以覆盖method_missing 来处理这个问题。
在您的 in_country 或 in_town 方法中返回 self。然后当调用first给它时，将它传递给all数组来处理。
代码可能是这样的：

module Dronestream
  class Strike

  class << self

  ...
  def all
    ...
  end

  def in_country(country)
    all.keep_if { |strike| strike['country'] == country }
    self
  end

  def in_town(town)
    all.keep_if { |strike| strike['town'] == town }
    self
  end
  ...

  def method_missing(name,*args,&block)
    return all.send(name.to_sym, *args, &block) if all.respond_to? name.to_sym
    super
  end
end