1 files changed, 24 insertions, 0 deletions
diff --git a/devices/cpu_linux.go b/devices/cpu_linux.go
new file mode 100644
index 0000000..2368218
--- /dev/null
+++ b/devices/cpu_linux.go
@@ -0,0 +1,24 @@
+// +build linux
+
+package devices
+
+// TODO gopsutil is at v3, and we're using v2. See if v3 is released and upgrade if so.
+import "github.com/shirou/gopsutil/cpu"
+
+func CpuCount() (int, error) {
+	cpuCount, err := cpu.Counts(false)
+	if err != nil {
+		return 0, err
+	}
+	if cpuCount == 0 {
+		is, err := cpu.Info()
+		if err != nil {
+			return 0, err
+		}
+		if is[0].Cores > 0 {
+			return len(is) / 2, nil
+		}
+		return len(is), nil
+	}
+	return cpuCount, nil
+}