fast_rake 0.1.0 → 0.2.0

data/lib/fast_rake/fast_runner.rb

@@ -24,7 +24,7 @@ class FastRake::FastRunner
     clean_previous_results
     @start = Time.now
 
-    put_w_time %{Started at #{Time.now.strftime("%H:%M:%S")}}
+    put_w_time %{Started at #{Time.now.strftime("%H:%M:%S")}, (will run #{@process_count} parallel processes)}
     at_exit { kill_remaining_children }
 
     start_some_children

data/lib/fast_rake/version.rb

@@ -1,6 +1,6 @@
 module FastRake
   VERSION_MAJOR = 0
-  VERSION_MINOR = 1
+  VERSION_MINOR = 2
   VERSION_PATCH = 0
   VERSION = [VERSION_MAJOR, VERSION_MINOR, VERSION_PATCH].join('.')
 end

data/lib/fast_rake.rb

@@ -4,22 +4,45 @@ require 'fast_rake/fast_runner'
 module FastRake
   extend Rake::DSL
 
-  def self.fast_runner(setup_tasks, run_tasks, fail_fast=true)
-    fast_runner_task(:two, 2, setup_tasks, run_tasks, fail_fast)
-    fast_runner_task(:four, 4, setup_tasks, run_tasks, fail_fast)
-    fast_runner_task(:eight, 8, setup_tasks, run_tasks, fail_fast)
-  end
-
-  def self.fast_runner_task(name, processes, setup_tasks, run_tasks, fail_fast=true)
-    desc "Fast test runner for #{processes} cpus"
-    task name, [:list] => setup_tasks do |t, args|
+  def self.fast_runner_task(name, setup_tasks, run_tasks, fail_fast=true, processes=nil)
+    desc "Fast test runner for #{name.to_s}"
+    task name, [:count, :list] => setup_tasks do |t, args|
       tasks_to_run = if !args[:list].nil?
         args[:list].split(' ')
       else
         run_tasks
       end
+      if !args[:count].nil? and args[:count].to_i != 0
+        processes = args[:count].to_i
+      elsif processes.nil?
+        processes = _processor_count 
+        puts "#{processes} processors detected" 
+      end
       FastRunner.new(tasks_to_run, processes, fail_fast).run
     end
   end
 
+  #stolen from https://github.com/grosser/parallel
+  def self._processor_count
+    case RbConfig::CONFIG['host_os']
+    when /darwin9/
+      `hwprefs cpu_count`.to_i
+    when /darwin/
+      (hwprefs_available? ? `hwprefs thread_count` : `sysctl -n hw.ncpu`).to_i
+    when /linux|cygwin/
+      `grep -c processor /proc/cpuinfo`.to_i
+    when /(open|free)bsd/
+      `sysctl -n hw.ncpu`.to_i
+    when /mswin|mingw/
+      require 'win32ole'
+      wmi = WIN32OLE.connect("winmgmts://")
+      cpu = wmi.ExecQuery("select NumberOfLogicalProcessors from Win32_Processor")
+      cpu.to_enum.first.NumberOfLogicalProcessors
+    when /solaris2/
+      `psrinfo -p`.to_i # this is physical cpus afaik
+    else
+      $stderr.puts "Unknown architecture ( #{RbConfig::CONFIG["host_os"]} ) assuming one processor."
+      1
+    end
+  end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: fast_rake
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 23
   prerelease: 
   segments: 
   - 0
-  - 1
+  - 2
   - 0
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Jonathan Ricketson
@@ -52,7 +52,7 @@ licenses: []
 post_install_message: 
 rdoc_options: 
 - --title
-- fast_rake-0.1.0 Documentation
+- fast_rake-0.2.0 Documentation
 require_paths: 
 - lib
 required_ruby_version: !ruby/object:Gem::Requirement