rake-opt-parallel 0.1.0 → 0.2.0

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    ZjI4NmI1OWE0Y2JhY2FiY2NmMDYwMGZiYTQxYzJlOTU2MTdjMDllNQ==
+    YzUxNTFkYmVjY2MwNzQ3YzQ0ODliYzc2NzYwY2QxOGExMjQwNjlkNQ==
   data.tar.gz: !binary |-
-    YjQ2NGIwZjllMjI5MDRhNGQ2MWUxYWQ5OGVkMjFkNjNkYmRjNDczNA==
+    YTk2ODQ1MjViMGM2MTg3MzlmOGNhMGNlOGZlNDVkODEwMGVkYjExYQ==
 SHA512:
   metadata.gz: !binary |-
-    MmU4ZTJiMjVlMzNhN2Q0MjFkMGI4NmNlOWMxYTU5MzJkMWU1NDdlZGUwYzE4
-    ZmZjZjBkMmNjODgwMWJkNWIwODk1NGZjYWI2ZTllOTVlMGI2M2RhZDNkMzk1
-    MWRjNTRjODZiNDY2NzIyYzNjOWE0OTNkNDkzNTliMzc4YTJiMjA=
+    MzAzMDJjYTUyMDAyMjkzZmRmOTJjNTQxZDgwM2IyOTg5NThlNTM0OWVlNzI5
+    NTJkNDZmOWQyZWY2NjMwY2Y2NjE0YTI4ZGI4YmNmMmUyNWFiNjEyYjQxOTAw
+    ODk1MDJmOWFkMzk0MzlhYWZhYjQwODZjYWIzMWQ5N2Y1NzAwZTY=
   data.tar.gz: !binary |-
-    NGZmM2RmNTMwMDJjZDU5MzhjMjQ5NzNkYTkzMjcyNWM1ZjUzN2Q5YTFlMjJl
-    MDM3YzczNTc3YzFjODI3YWFhNGViZGZkOWFiZjQzM2QwMDJlZWM0NzA4NGY0
-    NTI4OWQ5ZWY3ZTgzMWM0MTE1YzFhZGEyNzczNTUxM2RmNTAwOGU=
+    NWU4YWIxMGMyMzE5MGQ4OTFmOTY5NWNjMDUwMmU5ZWQwMjQwZGQxYTc0MzQ4
+    NjgxNjkzMWVlMjM0ODJhNmQzMDllNmI4OTdmODNhN2Q5NzRiM2UwMmNhMDY1
+    MjRiMzMyMjQ0YzM3MGU5MmMwZTEyNjcxMjhhMzNiNjNkN2RhYmE=

data/lib/rake/opt/parallel/patches/enumerable.rb

@@ -8,7 +8,6 @@ module Enumerable
     return self.each {}
   end
 
-
   def parallel_map(options={}, &block)
     # Convert options hash to a mash
     options = ::Hashie::Mash.new(options)
@@ -25,6 +24,7 @@ module Enumerable
       thread_count = options.unlimited ? self.size : options.force
       # Create a seperate thread pool
       thread_pool = Rake::ThreadPool.new(thread_count-1)
+      thread_pool.add_reference
     else
       # Otherwise, use the existing thread pool
       thread_pool = Rake.application.thread_pool
@@ -37,6 +37,13 @@ module Enumerable
 
     # Fulfill all promises
     promises.reverse_each { |f| f.value }
+
+    # Remove ThreadPool instance references if it's not the main one
+    if thread_pool != Rake.application.thread_pool
+      thread_pool.remove_reference
+      thread_pool = nil
+    end
+
     return promises.map { |f| f.value }
   end
 

data/lib/rake/opt/parallel/patches/rake/thread_pool.rb

@@ -0,0 +1,26 @@
+require 'rake/thread_pool'
+require 'set'
+
+class Rake::ThreadPool
+  @@instance_collector = Set.new
+
+  class << self
+    def all_instances
+      return @@instance_collector
+    end
+
+    def join_all
+      all_instances.each(&:join)
+    end
+  end
+
+  def add_reference
+    @@instance_collector << self
+  end
+
+  def remove_reference
+    @@instance_collector.delete(self)
+  end
+end
+
+Rake.application.thread_pool.add_reference

data/lib/rake/opt/parallel/patches.rb

@@ -1,2 +1,3 @@
 require_relative 'patches/enumerable.rb'
+require_relative 'patches/rake/thread_pool.rb'
 require_relative 'patches/thread.rb'

metadata

@@ -2,14 +2,14 @@
 name: rake-opt-parallel
 version: !ruby/object:Gem::Version
   version: !binary |-
-    MC4xLjA=
+    MC4yLjA=
 platform: ruby
 authors:
 - Jon San Miguel
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-03-14 00:00:00.000000000 Z
+date: 2016-05-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -69,6 +69,7 @@ files:
 - lib/rake/opt/parallel.rb
 - lib/rake/opt/parallel/patches.rb
 - lib/rake/opt/parallel/patches/enumerable.rb
+- lib/rake/opt/parallel/patches/rake/thread_pool.rb
 - lib/rake/opt/parallel/patches/thread.rb
 homepage: http://rubygems.org/gems/rake-opt-parallel
 licenses: