Merge pull request #62 from rage-rb/fiber-id

Preload fiber ID

lib/rage/fiber.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 class Fiber
+  # @private
   AWAIT_ERROR_MESSAGE = "err"
 
   # @private
@@ -23,14 +24,14 @@ def __get_err
     @__err
   end
 
-   # @private
-  def __get_id
-    @__rage_id ||= object_id.to_s
+  # @private
+  def __set_id
+    @__rage_id = object_id.to_s
   end
 
-  # @private
-  def __yielded?
-    !@__rage_id.nil?
+   # @private
+  def __get_id
+    @__rage_id
   end
 
   # @private

lib/rage/fiber_scheduler.rb

@@ -109,6 +109,7 @@ def fiber(&block)
     fiber = if parent == @root_fiber
       # the fiber to wrap a request in
       Fiber.new(blocking: false) do
+        Fiber.current.__set_id
         Fiber.current.__set_result(block.call)
       end
     else

lib/rage/middleware/fiber_wrapper.rb

@@ -17,7 +17,7 @@ def call(env)
       @app.call(env)
     ensure
       # notify Iodine the request can now be resumed
-      Iodine.publish(Fiber.current.__get_id, "", Iodine::PubSub::PROCESS) if Fiber.current.__yielded?
+      Iodine.publish(Fiber.current.__get_id, "", Iodine::PubSub::PROCESS)
     end
 
     # the fiber encountered blocking IO and yielded; instruct Iodine to pause the request

lib/rage/rspec.rb

@@ -18,6 +18,7 @@
 class Fiber
   def self.schedule(&block)
     fiber = Fiber.new(blocking: true) do
+      Fiber.current.__set_id
       Fiber.current.__set_result(block.call)
     end
     fiber.resume