From 605b6f0c44e4ee455c601ca01aaadc72d15bb9cd Mon Sep 17 00:00:00 2001
From: Julien Portalier <julien@portalier.com>
Date: Tue, 3 Dec 2024 19:59:22 +0100
Subject: [PATCH] Add Thread::LinkedList#each to safely iterate lists

Also adds `Thread.each(&)` and `Fiber.each(&)`.
---
 src/crystal/system/thread.cr             | 4 ++++
 src/crystal/system/thread_linked_list.cr | 7 +++++++
 src/fiber.cr                             | 5 +++++
 3 files changed, 16 insertions(+)

diff --git a/src/crystal/system/thread.cr b/src/crystal/system/thread.cr
index 0d6f5077633a..6976fd33516a 100644
--- a/src/crystal/system/thread.cr
+++ b/src/crystal/system/thread.cr
@@ -74,6 +74,10 @@ class Thread
     @@threads.try(&.unsafe_each { |thread| yield thread })
   end
 
+  def self.each(&)
+    threads.each { |thread| yield thread }
+  end
+
   def self.lock : Nil
     threads.@mutex.lock
   end
diff --git a/src/crystal/system/thread_linked_list.cr b/src/crystal/system/thread_linked_list.cr
index b6f3ccf65d4e..2af38fd1368e 100644
--- a/src/crystal/system/thread_linked_list.cr
+++ b/src/crystal/system/thread_linked_list.cr
@@ -24,6 +24,13 @@ class Thread
       end
     end
 
+    # Safely iterates the list.
+    def each(&) : Nil
+      @mutex.synchronize do
+        unsafe_each { |node| yield node }
+      end
+    end
+
     # Appends a node to the tail of the list. The operation is thread-safe.
     #
     # There are no guarantees that a node being pushed will be iterated by
diff --git a/src/fiber.cr b/src/fiber.cr
index 2b596a16017c..55745666c66d 100644
--- a/src/fiber.cr
+++ b/src/fiber.cr
@@ -78,6 +78,11 @@ class Fiber
     @@fibers.try(&.unsafe_each { |fiber| yield fiber })
   end
 
+  # :nodoc:
+  def self.each(&)
+    fibers.each { |fiber| yield fiber }
+  end
+
   # Creates a new `Fiber` instance.
   #
   # When the fiber is executed, it runs *proc* in its context.