将传递给方法的块传递给Ruby中的另一个方法

Question

我正在尝试编写ruby的副本keep_if和delete_if数组方法。这是我的代码。

module Strain 
    def keep 
    self.inject([]) do |extracts, element| 
     yield(element) ? extracts << element : extracts 
    end 
    end 

    def discard 
    self.inject([]) do |extracts, element| 
     !yield(element) ? extracts << element : extracts 
    end 
    end 
end 

class Array 
    include Strain 
end 

This works。但我想要做的事，如：

def discard 
    self - self.keep &block 
end 

期望的行为：

[1, 2, 3].discard { |number| number < 2 } 
# => [2, 3] 

所以我需要通过传递给discard方法块，到keep方法传递。我如何实现这一目标？

Answer 1

您可以参考块明确

def discard(&block) 
    self - self.keep(&block) 
end 

或隐含

def discard 
    self - self.keep(&Proc.new {}) 
end 

在你的情况，我建议第一种方法。