ruby/rbsに取り込まれた私のパッチ

いつもどおりstackprofでプロファイルを取っているとStringScanner#charposがめっちゃ遅くて、実装を見てみるとたしかにめっちゃ遅そうな実装だった。 https://github.com/ruby/strscan/blob/d0c82c20c64323c45ee275f09e5a951a291f1ee2/ext/strscan/strscan.c#L442-L453 これはscannerが読み込んだ位置を文字単位で返すAPIなのだけど、scannerは読み込んだ位置をバイト単位でしか保持していない。なので読み込んだ位置のバイトを使ってString#bytesliceを呼んで、その結果の部分文字列に対してString#lengthを呼んでいる。遅そう。

というわけでStringScanner#charposが遅いので、これの呼び出し回数を減らした。またRBSがASCIIだけで書かれている場合にはバイト単位の位置で充分なためcharposの代わりにposを使うようにした。つまり、RBSはASCIIだけで書くと速い。

https://github.com/ruby/rbs/pull/331github.com

RBS::Environmentをうっかりpすると画面が破滅するのを直した。

https://github.com/ruby/rbs/pull/334github.com

rbs prototype rbで重複するメソッドは1つにまとめるようにした。

class C
  if RUBY_VERSION >= '2.7'
    def foo
      do_something_27
    end
  else
    def foo
      do_something
    end
  end
end

PRのdescriptionにもあるとおり、こういうコードがあるとfooに対する定義が2回出力されてエラーになってしまうため。こういうコードはRailsの内部とかで実際にある。

https://github.com/ruby/rbs/pull/337github.com

重複する定数がある際にいい感じのエラーが出ずにNoMethodErrorになってしまっていたのを直した。これは実際に重複する定数を生成してしまって意味不明なエラーが出ていたとかだったと思う

https://github.com/ruby/rbs/pull/338github.com

rbs validateを速くした。当時のRBS RailsのRBSに対して3.77倍速くなったので結構でかい。

コードを読んでなんとなくの理解しかできていないのだけど、rbsでは型引数をメソッドに適用するために全てのメソッドに対してsubメソッドでメソッド型を表すオブジェクトを作り直していた。一方型引数を使っているメソッドというのは多くなくて、かつこのオブジェクトの作り直しは型引数を使っていないコードに対しては無意味なので、そこで無意味に遅くなってしまっていた。

なのでこのPRでは型引数を使っていないメソッド定義ではオブジェクトを生成しないようにして速くした。みたいな感じだったと思う。

https://github.com/ruby/rbs/pull/457github.com

rbs prototype rbでaliasをサポートした。

これでRailsのコードに対してRBSを生成し直したらalias周りで色々バグが見つかって、いくつかIssueに報告した。むずそう。 https://github.com/ruby/rbs/pull/516

https://github.com/ruby/rbs/pull/462github.com

Enumerable#to_hはブロック引数を受け付けるのだけど、そうなっていなかったので直した。

これはRBS Railsを触っているときにto_hの定義を見に行きたくなって見に行ったら「あれ？ブロック受け取らなかったっけ？」となって足したような気がする。

https://github.com/ruby/rbs/pull/463github.com

RBSが意図しないEOFでSyntax ErrorになったときにNoMethodErrorになってしまっていたのを直した。

raccのためのnext_tokenメソッドが返す値が間違っていたのが原因だった。raccの使い方に少し詳しくなった。

https://github.com/ruby/rbs/pull/465github.com

rbs prototype runtimeがhash spread構文でエラーになってしまっていたのを直した。

https://github.com/ruby/rbs/pull/481github.com

rbs prototype rbでpublicとprivateとmodule_functionに対応した。

正直publicとprivateはおまけで、module_functionの対応が欲しかった。そしてRBSにはprotectedがない。

https://github.com/ruby/rbs/pull/482github.com

Singleton moduleの型を書いた。

この型があって便利かと言うと、まあinclude Singletonとinstanceメソッドがあっても型エラーにならなくなるぐらいかなあ。

class SingletonSingletonTest、ひどい名前だ

https://github.com/ruby/rbs/pull/483github.com

RBSにmonitor libraryの型を追加しようとMonitorMixinに対してrbs prototype runtimeしたらエラーになって、それの対応。 rbs prototype runtimeをarguments forwarding syntaxで定義されたメソッドに対して使うと、Syntax Errorなコードを生成していた。

https://github.com/ruby/rbs/pull/485github.com

monitor libraryの型を書いた。これもRailsが依存していたから追加しておきたかった型。

https://github.com/ruby/rbs/pull/486github.com

rbs prototype rbのmodule_function対応のバグ修正。別の場所で定義されたメソッドをmodule_functionしようとしているとエラーになっていた。これはRailsの型を生成しようとして気がついたやつな気がする。

https://github.com/ruby/rbs/pull/487github.com

rbs prototype rbでrefinementsを無視するようにした。

RBSはrefinementsをサポートしていないし、まあ無視していいかなと言う気持ち。

https://github.com/ruby/rbs/pull/491github.com

rbs prototype rbでpublicやprivateのサポートにもバグがあってそれを直した。なんかこの辺のコード死ぬほど難しくてむずい。書いたの自分だけど……

https://github.com/ruby/rbs/pull/492github.com

rbs prototype rbでextend selfをサポートした。私はextend selfを好きでよく書くからこれが出てくれるのは嬉しい。

https://github.com/ruby/rbs/pull/505github.com