Add benchmark script

Author: pocke

Gemfile

@@ -13,6 +13,7 @@ gem "rubocop-rubycw"
 gem "json"
 gem "json-schema"
 gem 'stackprof'
+gem 'memory_profiler'
 gem "goodcheck"
 gem "dbm"
 gem 'digest'

Gemfile.lock

@@ -33,6 +33,7 @@ GEM
       addressable (>= 2.8)
     language_server-protocol (3.17.0.3)
     marcel (1.0.2)
+    memory_profiler (1.0.1)
     minitest (5.22.3)
     mutex_m (0.2.0)
     net-protocol (0.2.2)
@@ -110,6 +111,7 @@ DEPENDENCIES
   goodcheck
   json
   json-schema
+  memory_profiler
   minitest
   mutex_m
   net-smtp

benchmark/memory_new_env.rb

@@ -0,0 +1,11 @@
+require_relative './utils'
+
+require 'memory_profiler'
+
+env = nil
+
+r = MemoryProfiler.report do
+  env = new_env
+end
+
+r.pretty_print

benchmark/memory_new_rails_env.rb

@@ -0,0 +1,13 @@
+require_relative './utils'
+
+tmpdir = prepare_collection!
+
+require 'memory_profiler'
+
+env = nil
+
+r = MemoryProfiler.report do
+  env = new_rails_env(tmpdir)
+end
+
+r.pretty_print

benchmark/utils.rb

@@ -0,0 +1,33 @@
+require "rbs"
+require 'tmpdir'
+
+def prepare_collection!
+  tmpdir = Pathname(Dir.mktmpdir)
+  at_exit { tmpdir.rmtree }
+
+  tmpdir.join('Gemfile').write(<<~RUBY)
+    source "https://rubygems.org"
+    gem 'rails'
+  RUBY
+  Bundler.with_original_env do
+    system('bundle', 'lock', chdir: tmpdir.to_s)
+    system('rbs', 'collection', 'init', chdir: tmpdir.to_s)
+    system('rbs', 'collection', 'install', chdir: tmpdir.to_s)
+  end
+
+  tmpdir
+end
+
+def new_env
+  loader = RBS::EnvironmentLoader.new()
+  yield loader if block_given?
+  RBS::Environment.from_loader(loader).resolve_type_names
+end
+
+def new_rails_env(collection_dir)
+  new_env do |loader|
+    lock_path = collection_dir.join('rbs_collection.lock.yaml')
+    lock = RBS::Collection::Config::Lockfile.from_lockfile(lockfile_path: lock_path, data: YAML.load_file(lock_path.to_s))
+    loader.add_collection(lock)
+  end
+end