Class: Rake::CpuCounter
- Inherits:
-
Object
- Object
- Rake::CpuCounter
- Defined in:
- lib/rake/cpu_counter.rb
Overview
Based on a script at:
http://stackoverflow.com/questions/891537/ruby-detect-number-of-cpus-installed
Class Method Summary collapse
-
.count ⇒ Object
:nodoc: all.
Instance Method Summary collapse
- #count ⇒ Object
- #count_via_cpuinfo ⇒ Object
- #count_via_hwprefs_cpu_count ⇒ Object
- #count_via_hwprefs_thread_count ⇒ Object
- #count_via_java_runtime ⇒ Object
- #count_via_sysctl ⇒ Object
- #count_via_win32 ⇒ Object
- #count_with_default(default = 4) ⇒ Object
- #in_path_command(command) ⇒ Object
- #look_for_command(dir, command) ⇒ Object
- #resolve_command(command) ⇒ Object
- #run(command, *args) ⇒ Object
Class Method Details
.count ⇒ Object
:nodoc: all
8 9 10 |
# File 'lib/rake/cpu_counter.rb', line 8 def self.count new.count_with_default end |
Instance Method Details
#count ⇒ Object
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
# File 'lib/rake/cpu_counter.rb', line 18 def count if defined?(Java::Java) count_via_java_runtime else case RbConfig::CONFIG['host_os'] when /darwin9/ count_via_hwprefs_cpu_count when /darwin/ count_via_hwprefs_thread_count || count_via_sysctl when /linux/ count_via_cpuinfo when /bsd/ count_via_sysctl when /mswin|mingw/ count_via_win32 else # Try everything count_via_win32 || count_via_sysctl || count_via_hwprefs_thread_count || count_via_hwprefs_cpu_count end end end |
#count_via_cpuinfo ⇒ Object
58 59 60 61 62 |
# File 'lib/rake/cpu_counter.rb', line 58 def count_via_cpuinfo open('/proc/cpuinfo') { |f| f.readlines }.grep(/processor/).size rescue StandardError nil end |
#count_via_hwprefs_cpu_count ⇒ Object
68 69 70 |
# File 'lib/rake/cpu_counter.rb', line 68 def count_via_hwprefs_cpu_count run 'hwprefs', 'cpu_count' end |
#count_via_hwprefs_thread_count ⇒ Object
64 65 66 |
# File 'lib/rake/cpu_counter.rb', line 64 def count_via_hwprefs_thread_count run 'hwprefs', 'thread_count' end |
#count_via_java_runtime ⇒ Object
43 44 45 46 47 |
# File 'lib/rake/cpu_counter.rb', line 43 def count_via_java_runtime Java::Java.lang.Runtime.getRuntime.availableProcessors rescue StandardError nil end |
#count_via_sysctl ⇒ Object
72 73 74 |
# File 'lib/rake/cpu_counter.rb', line 72 def count_via_sysctl run 'sysctl', '-n', 'hw.ncpu' end |
#count_via_win32 ⇒ Object
49 50 51 52 53 54 55 56 |
# File 'lib/rake/cpu_counter.rb', line 49 def count_via_win32 require 'win32ole' wmi = WIN32OLE.connect("winmgmts://") cpu = wmi.ExecQuery("select NumberOfCores from Win32_Processor") # TODO count hyper-threaded in this cpu.to_enum.first.NumberOfCores rescue StandardError, LoadError nil end |
#count_with_default(default = 4) ⇒ Object
12 13 14 15 16 |
# File 'lib/rake/cpu_counter.rb', line 12 def count_with_default(default=4) count || default rescue StandardError default end |
#in_path_command(command) ⇒ Object
98 99 100 101 102 |
# File 'lib/rake/cpu_counter.rb', line 98 def in_path_command(command) IO.popen ['which', command] do |io| io.eof? ? nil : command end end |
#look_for_command(dir, command) ⇒ Object
93 94 95 96 |
# File 'lib/rake/cpu_counter.rb', line 93 def look_for_command(dir, command) path = File.join(dir, command) File.exist?(path) ? path : nil end |
#resolve_command(command) ⇒ Object
87 88 89 90 91 |
# File 'lib/rake/cpu_counter.rb', line 87 def resolve_command(command) look_for_command("/usr/sbin", command) || look_for_command("/sbin", command) || in_path_command(command) end |
#run(command, *args) ⇒ Object
76 77 78 79 80 81 82 83 84 85 |
# File 'lib/rake/cpu_counter.rb', line 76 def run(command, *args) cmd = resolve_command(command) if cmd IO.popen [cmd, *args] do |io| io.read.to_i end else nil end end |