[前][次][番号順一覧][スレッド一覧]

ruby-changes:66905

From: kvokka <ko1@a...>
Date: Tue, 27 Jul 2021 17:01:52 +0900 (JST)
Subject: [ruby-changes:66905] a288c21a5d (master): [ruby/uri] Fix to support Ruby 3.0 Ractor

https://git.ruby-lang.org/ruby.git/commit/?id=a288c21a5d

From a288c21a5d46418e75c0f03eb12ff0782e51568d Mon Sep 17 00:00:00 2001
From: kvokka <kvokka@y...>
Date: Sat, 29 May 2021 11:29:39 +0100
Subject: [ruby/uri] Fix to support Ruby 3.0 Ractor

https://github.com/ruby/uri/commit/1faa4fdc16
---
 lib/uri.rb              |  2 +-
 lib/uri/common.rb       | 32 +++++++++++++++++++++++++-------
 lib/uri/file.rb         |  2 --
 lib/uri/ftp.rb          |  1 -
 lib/uri/http.rb         |  3 ---
 lib/uri/https.rb        |  1 -
 lib/uri/ldap.rb         |  2 --
 lib/uri/ldaps.rb        |  1 -
 lib/uri/mailto.rb       |  2 --
 lib/uri/ws.rb           |  3 ---
 lib/uri/wss.rb          |  1 -
 test/uri/test_common.rb | 19 +++++++++++++++++++
 12 files changed, 45 insertions(+), 24 deletions(-)

diff --git a/lib/uri.rb b/lib/uri.rb
index 5e820f4..41b6854 100644
--- a/lib/uri.rb
+++ b/lib/uri.rb
@@ -29,8 +29,8 @@ https://github.com/ruby/ruby/blob/trunk/lib/uri.rb#L29
 #   module URI
 #     class RSYNC < Generic
 #       DEFAULT_PORT = 873
+#       URI.refresh_scheme_list
 #     end
-#     @@schemes['RSYNC'] = RSYNC
 #   end
 #   #=> URI::RSYNC
 #
diff --git a/lib/uri/common.rb b/lib/uri/common.rb
index 915c0e9..ab3234c 100644
--- a/lib/uri/common.rb
+++ b/lib/uri/common.rb
@@ -16,6 +16,7 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/common.rb#L16
   REGEXP = RFC2396_REGEXP
   Parser = RFC2396_Parser
   RFC3986_PARSER = RFC3986_Parser.new
+  Ractor.make_shareable(RFC3986_PARSER) if defined?(Ractor)
 
   # URI::Parser.new
   DEFAULT_PARSER = Parser.new
@@ -27,6 +28,7 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/common.rb#L28
   DEFAULT_PARSER.regexp.each_pair do |sym, str|
     const_set(sym, str)
   end
+  Ractor.make_shareable(DEFAULT_PARSER) if defined?(Ractor)
 
   module Util # :nodoc:
     def make_components_hash(klass, array_hash)
@@ -62,10 +64,30 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/common.rb#L64
 
   include REGEXP
 
-  @@schemes = {}
+  SCHEME_LIST_MUTEX = Mutex.new
+  private_constant :SCHEME_LIST_MUTEX
+
   # Returns a Hash of the defined schemes.
+  # The list is lazily calculated.
   def self.scheme_list
-    @@schemes
+    return const_get(:SCHEMES) if defined?(SCHEMES)
+
+    SCHEME_LIST_MUTEX.synchronize do
+      const_set(:SCHEMES, ObjectSpace.
+        each_object(Class).
+        select { |klass| klass < URI::Generic }.
+        each_with_object({}) { |klass, acc| acc[klass.name.split('::').last.upcase] = klass }.
+        freeze)
+    end
+  end
+
+  # Re-calculate scheme list
+  def self.refresh_scheme_list
+    SCHEME_LIST_MUTEX.synchronize do
+      remove_const(:SCHEMES) if defined?(SCHEMES)
+    end
+
+    scheme_list
   end
 
   #
@@ -73,11 +95,7 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/common.rb#L95
   # from +URI.scheme_list+.
   #
   def self.for(scheme, *arguments, default: Generic)
-    if scheme
-      uri_class = @@schemes[scheme.upcase] || default
-    else
-      uri_class = default
-    end
+    uri_class = scheme_list[scheme.to_s.upcase] || default
 
     return uri_class.new(scheme, *arguments)
   end
diff --git a/lib/uri/file.rb b/lib/uri/file.rb
index 561ec70..45a7a8b 100644
--- a/lib/uri/file.rb
+++ b/lib/uri/file.rb
@@ -89,6 +89,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/file.rb#L89
     def set_password(v)
     end
   end
-
-  @@schemes['FILE'] = File
 end
diff --git a/lib/uri/ftp.rb b/lib/uri/ftp.rb
index fb38481..ec8b6e6 100644
--- a/lib/uri/ftp.rb
+++ b/lib/uri/ftp.rb
@@ -262,5 +262,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/ftp.rb#L262
       return str
     end
   end
-  @@schemes['FTP'] = FTP
 end
diff --git a/lib/uri/http.rb b/lib/uri/http.rb
index 70cfb2a..ccd09dd 100644
--- a/lib/uri/http.rb
+++ b/lib/uri/http.rb
@@ -81,7 +81,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/http.rb#L81
       url.start_with?(?/.freeze) ? url : ?/ + url
     end
   end
-
-  @@schemes['HTTP'] = HTTP
-
 end
diff --git a/lib/uri/https.rb b/lib/uri/https.rb
index c481b1f..98a0840 100644
--- a/lib/uri/https.rb
+++ b/lib/uri/https.rb
@@ -18,5 +18,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/https.rb#L18
     # A Default port of 443 for URI::HTTPS
     DEFAULT_PORT = 443
   end
-  @@schemes['HTTPS'] = HTTPS
 end
diff --git a/lib/uri/ldap.rb b/lib/uri/ldap.rb
index 14e6163..3a0df23 100644
--- a/lib/uri/ldap.rb
+++ b/lib/uri/ldap.rb
@@ -256,6 +256,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/ldap.rb#L256
       false
     end
   end
-
-  @@schemes['LDAP'] = LDAP
 end
diff --git a/lib/uri/ldaps.rb b/lib/uri/ldaps.rb
index 227e7fa..fd3f7e8 100644
--- a/lib/uri/ldaps.rb
+++ b/lib/uri/ldaps.rb
@@ -17,5 +17,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/ldaps.rb#L17
     # A Default port of 636 for URI::LDAPS
     DEFAULT_PORT = 636
   end
-  @@schemes['LDAPS'] = LDAPS
 end
diff --git a/lib/uri/mailto.rb b/lib/uri/mailto.rb
index d08c2ae..214dea8 100644
--- a/lib/uri/mailto.rb
+++ b/lib/uri/mailto.rb
@@ -288,6 +288,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/mailto.rb#L288
     end
     alias to_rfc822text to_mailtext
   end
-
-  @@schemes['MAILTO'] = MailTo
 end
diff --git a/lib/uri/ws.rb b/lib/uri/ws.rb
index 2bfee59..b846ba2 100644
--- a/lib/uri/ws.rb
+++ b/lib/uri/ws.rb
@@ -78,7 +78,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/ws.rb#L78
       url.start_with?(?/.freeze) ? url : ?/ + url
     end
   end
-
-  @@schemes['WS'] = WS
-
 end
diff --git a/lib/uri/wss.rb b/lib/uri/wss.rb
index 1cfa133..12a53c7 100644
--- a/lib/uri/wss.rb
+++ b/lib/uri/wss.rb
@@ -18,5 +18,4 @@ module URI https://github.com/ruby/ruby/blob/trunk/lib/uri/wss.rb#L18
     # A Default port of 443 for URI::WSS
     DEFAULT_PORT = 443
   end
-  @@schemes['WSS'] = WSS
 end
diff --git a/test/uri/test_common.rb b/test/uri/test_common.rb
index 1afa35f..311adc0 100644
--- a/test/uri/test_common.rb
+++ b/test/uri/test_common.rb
@@ -33,6 +33,25 @@ class TestCommon < Test::Unit::TestCase https://github.com/ruby/ruby/blob/trunk/test/uri/test_common.rb#L33
     end
   end
 
+  def test_ractor
+    return unless defined?(Ractor)
+    r = Ractor.new { URI.parse("https://ruby-lang.org/").inspect }
+    assert_equal(URI.parse("https://ruby-lang.org/").inspect, r.take)
+  end
+
+  def test_register_scheme
+    assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS"].sort, URI.scheme_list.keys.sort)
+
+    begin
+      URI::Generic.const_set :FOOBAR, Class.new(URI::Generic)
+      URI.refresh_scheme_list
+      assert_equal(["FILE", "FTP", "HTTP", "HTTPS", "LDAP", "LDAPS", "MAILTO", "WS", "FOOBAR"].sort, URI.scheme_list.keys.sort)
+    ensure
+      URI::Generic.send(:remove_const, :FOOBAR)
+      URI.refresh_scheme_list
+    end
+  end
+
   def test_regexp
     EnvUtil.suppress_warning do
       assert_instance_of Regexp, URI.regexp
-- 
cgit v1.1


--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/

[前][次][番号順一覧][スレッド一覧]