ようへいの日々精進XP

よかろうもん

Active Support が便利だった

ruby 雑なメモ メモ 俺のメモ ショートメモ 超メモ

経緯

ほんと、タイトルの通り。

以下のような、ネストされたハッシュのキーをシンボルに変換したくてインターネットを彷徨っていたら、Active Support の deep_transform_keys が便利に使えるよとのことだったので、見様見真似で使ってみたら意図した結果を得られたのでメモしておく。

 {"aa"=>{"bb"=>[{"cc"=>"11", "dd"=>"22"}, {"ee"=>"33", "ff"=>"44"}, {"gg"=>"55", "hh"=>"66"}], "ii"=>"77"}}

本記事で扱う Ruby のバージョンは以下の通り。

ruby 3.0.4p208 (2022-04-12 revision 3fa771dded) [x86_64-darwin21]

Active Support について

railsguides.jp

Active SupportはRuby on Railsのコンポーネントであり、Ruby言語の拡張やユーティリティを提供します。

ドキュメントの冒頭に上記のように記載されている。Rails だけのものだと思っていたけど、そうじゃないんだなー。

デモ

Ruby 本体にも transform_keys というメソッドがあった

docs.ruby-lang.org

以下の通り、トップレベルのキーだけはシンボルに変換された。

irb(main):002:0> h = {"aa"=>{"bb"=>[{"cc"=>"11", "dd"=>"22"}, {"ee"=>"33", "ff"=>"44"}, {"gg"=>"55", "hh"=>"66"}], "ii"=>"77"}}
irb(main):005:0> h.transform_keys(&:to_sym)
=> {:aa=>{"bb"=>[{"cc"=>"11", "dd"=>"22"}, {"ee"=>"33", "ff"=>"44"}, {"gg"=>"55", "hh"=>"66"}], "ii"=>"77"}}

これでは要件を満たせない。

Active Support の deep_transform_keys メソッドでやってみる

Gemfile に追加。

# frozen_string_literal: true

source "https://rubygems.org"

git_source(:github) { |repo_name| "https://github.com/#{repo_name}" }

gem 'active_support'

フットプリントを最小限にするため、Active Support はデフォルトでは最小限の依存関係を読み込むとのことだけど、今回、以下のようにハッシュの拡張機能のみを読み込んでいる (つもり)。実際のところ、どのくらいメモリ使用量が変わるんだろうか。気になるところ。

irb(main):001:0> require 'active_support/core_ext/hash/keys'
=> true
irb(main):002:0> h = {"aa"=>{"bb"=>[{"cc"=>"11", "dd"=>"22"}, {"ee"=>"33", "ff"=>"44"}, {"gg"=>"55", "hh"=>"66"}], "ii"=>"77"}}
=> {"aa"=>{"bb"=>[{"cc"=>"11", "dd"=>"22"}, {"ee"=>"33", "ff"=>"44"}, {"gg"=>"55", "hh"=>"66"}], "ii"=>"77"}}
irb(main):003:0> h.deep_transform_keys(&:to_sym)
=> {:aa=>{:bb=>[{:cc=>"11", :dd=>"22"}, {:ee=>"33", :ff=>"44"}, {:gg=>"55", :hh=>"66"}], :ii=>"77"}}

おお、簡単！素晴らしい。

Active Support を使わないパターンを考えた

stackoverflow.com

上記の記事を参考にして、以下のようなコードを書いてみた。んー、良いんだか悪いだな。

require 'minitest/autorun'

def make_all_keys_symbol(hash)
  symbold_hash = {}
  hash.each do |k, v|
    if v.is_a?(Hash)
      symbold_hash[k.to_sym] = make_all_keys_symbol(v)
    elsif v.is_a?(Array)
      nested_array = []
      v.each do |e|
        nested_array << e.transform_keys(&:to_sym)
      end
      symbold_hash[k.to_sym] = nested_array
    else
      symbold_hash[k.to_sym] = v
    end
  end
  symbold_hash
end

class TestMakeAllKeysSymbol < Minitest::Test
  def setup
    @h = {"aa"=>{"bb"=>[{"cc"=>"11", "dd"=>"22"}, {"ee"=>"33", "ff"=>"44"}, {"gg"=>"55", "hh"=>"66"}], "ii"=>"77"}}
  end

  def test_make_all_keys_symbol
    assert_equal({:aa=>{:bb=>[{:cc=>"11", :dd=>"22"}, {:ee=>"33", :ff=>"44"}, {:gg=>"55", :hh=>"66"}], :ii=>"77"}}, make_all_keys_symbol(@h))
  end
end

いちおう、問題なく動く。

$ ruby test.rb
Run options: --seed 60294

# Running:

.

Finished in 0.000705s, 1418.4397 runs/s, 1418.4397 assertions/s.

1 runs, 1 assertions, 0 failures, 0 errors, 0 skips

ということで

Active Support が便利だったメモでした。

おまけ

irb で require 'active_support' した時と require 'active_support/core_ext/hash/keys' した時のメモリ使用量を ps コマンドで見てみた。

USER               PID  %CPU %MEM      VSZ    RSS   TT  STAT STARTED      TIME COMMAND
kappa            72224   0.0  0.1 34924040  26564 s012  S+    3:01PM   0:00.36 irb
kappa            72224   0.0  0.1 34925192  31072 s012  S+    3:01PM   0:00.46 irb # require 'active_support'
kappa            75776   0.0  0.1 35071496  27224 s012  S+    3:02PM   0:00.37 irb # require 'active_support/core_ext/hash/keys'

確かに、若干ではあるが、ハッシュの拡張機能をのみを読み込んだ時の方がメモリ使用量 (RSS) が少ないように見える。