Add abort_on_exception to threaded_map

sferik · Oct 20, 2012 · 15c9a7c · 15c9a7c
1 parent 6228922
commit 15c9a7c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/twitter/core_ext/enumerable.rb b/lib/twitter/core_ext/enumerable.rb
@@ -1,11 +1,15 @@
 module Enumerable
 
   def threaded_map
+    initial_abort_on_exception = Thread.abort_on_exception
+    Thread.abort_on_exception ||= true
     threads = []
     each do |object|
-      threads << Thread.new{yield object}
+      threads << Thread.new { yield object }
     end
-    threads.map(&:value)
+    values = threads.map(&:value)
+    Thread.abort_on_exception = initial_abort_on_exception
+    values
   end
 
 end