ractorize 0.0.9 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 72a53fba452511ad30a39dc46714c285570a2904f167c960e97d519e63ee1829
-  data.tar.gz: 77c170a01ff7ec50bfb63c8c70b905be6d999bdaa2b8cd0e3826afd7bc5bb9d1
+  metadata.gz: f7827fd6bd911724e81479f2966f75aafb825c1320fb870531b08db1079b5b9c
+  data.tar.gz: bf7da9b30088e3554637b646ef4bb5ceab085d076f3e0097c68f236d647b3baf
 SHA512:
-  metadata.gz: 3655da0fc231e1d6b5c9bd063cf8e676ab6a9ce7f3b09421e48d085978257574bbbf988d538acb190dd0512d3838b5fe49a9436fa85e65f67a09aebfe137ff8b
-  data.tar.gz: 0dad22a3bf13eedff038063711e18e473b25f5e6e760b3b400de4603996240f71d3635788c8f06e2705da2a8103df236c5c76f94303ff44900122baedd368a9d
+  metadata.gz: e7858c116a76c408a6c341ee8007a91a50e4caae089978564d2704535845248e14272c18909bd8f4e5458e6dbe97262afb2acc088fd79412f5231ff1f56d3de1
+  data.tar.gz: 0e3b0165b66489c83b3dc7ebe37873ad6600e4fb9239eeaa326440d700112bf13255a67532f84fdfd15c3d37afcb93285742a4a50e5eef0993231bab1c5976ff

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## [0.0.10] - 2026-06-30
+
+- Handle thunks being cloned
+
 ## [0.0.9] - 2026-06-29
 
 - Fix bug that results from moving Time, which is neither shareable nor movable!

data/src/ractorize/garbage_collection/tracker.rb

@@ -2,11 +2,13 @@ module Ractorize
   module GarbageCollection
     class Tracker
       attr_accessor :ractorized_object_id_to_ractor,
-                    :thunk_id_to_ractor
+                    :thunk_id_to_ractor,
+                    :ractor_to_thunk_ids
 
       def initialize
         self.ractorized_object_id_to_ractor = ObjectSpace::WeakMap.new
         self.thunk_id_to_ractor = ObjectSpace::WeakMap.new
+        self.ractor_to_thunk_ids = ObjectSpace::WeakKeyMap.new
       end
 
       def track_ractorized_object(ractorized_object)
@@ -24,9 +26,32 @@ module Ractorize
         thunk_id_to_ractor[thunk_id] = ractor
       end
 
+      def thunk_cloned(old_thunk_id, new_thunk_id, thunk_ractor)
+        thunk_ids = ractor_to_thunk_ids[thunk_ractor]
+
+        if thunk_ids
+          thunk_ids << new_thunk_id
+        else
+          ractor_to_thunk_ids[thunk_ractor] = [old_thunk_id, new_thunk_id]
+        end
+
+        thunk_id_to_ractor[new_thunk_id] = thunk_ractor
+      end
+
       def cleanup_after_thunk(thunk_id)
         ractor = thunk_id_to_ractor.delete(thunk_id)
-        ractor&.<<(:__close__)
+
+        return unless ractor
+
+        thunk_ids = ractor_to_thunk_ids[ractor]
+
+        if thunk_ids
+          thunk_ids.delete(thunk_id)
+
+          return unless thunk_ids.empty?
+        end
+
+        ractor << :__close__
       rescue Ractor::ClosedError
         # do nothing
       end

data/src/ractorize/garbage_collection/tracking_ractor.rb

@@ -21,6 +21,8 @@ module Ractorize
                 tracker.cleanup_after_ractorized_object(ractorized_object_id)
               in :track_thunk, thunk_id, thunk_ractor
                 tracker.track_thunk(thunk_id, thunk_ractor)
+              in :thunk_cloned, old_thunk_id, new_thunk_id, thunk_ractor
+                tracker.thunk_cloned(old_thunk_id, new_thunk_id, thunk_ractor)
               in :cleanup_after_thunk, thunk_id
                 tracker.cleanup_after_thunk(thunk_id)
               end

data/src/ractorize/garbage_collection.rb

@@ -14,7 +14,7 @@ module Ractorize
       end
 
       def track_thunk(thunk)
-        # We have to define the finalizer here, not in the tracker, because it's not frozen yet
+        # We have to define the finalizer here, not in the tracker, because it's not shareable
         ObjectSpace.define_finalizer(thunk, &finalize_thunk_proc)
 
         begin
@@ -24,6 +24,24 @@ module Ractorize
         end
       end
 
+      def thunk_cloned(old_thunk, new_thunk)
+        ractor = old_thunk.__thunk_ractor__
+
+        # We have to define the finalizer here, not in the tracker, because it's not shareable
+        # ObjectSpace.define_finalizer(new_thunk, &finalize_thunk_proc)
+
+        begin
+          TRACKING_RACTOR << [
+            :thunk_cloned,
+            old_thunk.__object_id__,
+            new_thunk.__object_id__,
+            ractor
+          ].freeze
+        rescue TrackingRactor::ClosedError
+          # do nothing
+        end
+      end
+
       def cleanup_after_ractorized_object(ractorized_object_id)
         TRACKING_RACTOR << [:cleanup_after_ractorized_object, ractorized_object_id].freeze
       rescue Ractor::ClosedError

data/src/ractorize/thunk.rb

@@ -10,13 +10,20 @@ module Ractorize
       ::Ractorize::GarbageCollection.track_thunk(self)
     end
 
-    def initialize_clone(...)
-      # :nocov:
-      raise "CAREFUL! THUNK CLONED!!"
-      # :nocov:
-      # is this actually necessary?? Seems so?
+    def initialize_clone(original_thunk)
+      if __resolved__? && original_thunk.__resolved__?
+        # Seems this could happen if we had a frozen thunk whose @__value__ is not shareable
+        # Since the thunk's ractor is already gone nothing to worry about
+        return
+      end
+
+      self.__object_id__ = ::Object.instance_method(:object_id).bind_call(self)
+
+      ::Ractorize::GarbageCollection.thunk_cloned(original_thunk, self)
     end
 
+    def __resolved__? = defined?(@__value__)
+
     def method_missing(...)
       __value__.__send__(...)
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ractorize
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.10
 platform: ruby
 authors:
 - Miles Georgi