Added cpu_count().

postmodern · Feb 4, 2025 · 8e31438 · 8e31438
1 parent 99749a2
commit 8e31438
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/share/ruby-install/system.sh b/share/ruby-install/system.sh
@@ -93,6 +93,25 @@ function detect_downloader()
 	esac
 }
 
+#
+# Gets the number of CPU cores.
+#
+function cpu_count()
+{
+	local count
+
+	case "$os_platform" in
+		Darwin|*BSD)
+			count="$(sysctl -n hw.ncpu)"
+			;;
+		*)
+			count="$(getconf _NPROCESSORS_ONLN || grep -c ^processor /proc/cpuinfo)"
+			;;
+	esac
+
+	echo -n "$count"
+}
+
 detect_os
 detect_sudo
 detect_package_manager

diff --git a/test/system-tests/cpu_count_test.sh b/test/system-tests/cpu_count_test.sh
@@ -0,0 +1,9 @@
+. ./test/helper.sh
+. ./share/ruby-install/system.sh
+
+function test_cpu_count()
+{
+	assertTrue "did not return a number > 0" '(( $(cpu_count) > 0 ))'
+}
+
+SHUNIT_PARENT=$0 . $SHUNIT2