deep_rejectが欲しくてrefinementsで実装しようとした話

再帰的に条件判定を行うrejectが欲しくなりました。
こういう動きです。

1
2
3
4
5
6
7
8

[1,2].deep_reject{|val| val == 1 }  # => [2]

[1,2, [1,2,3]].deep_reject{|val| val == 1 } # => [2, [2, 3]]

[1, 2, {a: "a", b: "b" }].deep_reject{|key, val| key == :a } # => [1, 2, {:b=>"b"}]

[1, 2, {a: "a", b: "b", c: {a: "c", c: "d"} }].deep_reject{|key, val| key == :a }
# => [1, 2, {:b=>"b", :c=>{:c=>"d"}}]

かなり便利そうです。

refinmetsで実装してみます。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

  module HashEx
    refine Hash do
      def deep_reject(&block)
        self.each_with_object({}) do |(k, v), result|
          unless block.call(k, v)
            if v.is_a?(Hash)
              result[k] = v.deep_reject(&block)
            else
              result[k] = v
            end
          end
        end
      end
    end
  end

  module ArrayEx
    refine Array do
      def deep_reject(&block)
        self.each_with_object([]) do |v, result|
          unless block.call(v)
            if v.is_a?(Array)
              result << v.deep_reject(&block)
            elsif v.is_a?(Hash)
              result << v.deep_reject(&block) # 呼べない！
            else
              result << v
            end
          end
        end
      end
    end
  end

ですがこれを

1
2

using ArrayEx
using HashEx

と読み込んで使おうとしたところ残念ながら配列の中にハッシュが紛れ込んでいるようなケースだとArrayExのdeep_rejectからHashExのdeep_rejectが参照できず思った通りの結果になりませんでした。
refinementsはかなり厳密なスコープで動くようです。

まぁあったら有意義なメソッドだしいいかということで結局オープンクラスで実装しました。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

class Array
  def deep_reject(&block)
    self.each_with_object([]) do |v, result|
      unless block.call(v)
        if v.is_a?(Array)
          result << v.deep_reject(&block)
        elsif v.is_a?(Hash)
          result << v.deep_reject(&block)
        else
          result << v
        end
      end
    end
  end
end

class Hash
  def deep_reject(&block)
    self.each_with_object({}) do |(k, v), result|
      unless block.call(k, v)
        if v.is_a?(Hash)
          result[k] = v.deep_reject(&block)
        elsif v.is_a?(Array)
          result[k] = v.deep_reject(&block)
        else
          result[k] = v
        end
      end
    end
  end
end

Comments