From 71adb75383abfcfd4e97054aae270bec4b446e37 Mon Sep 17 00:00:00 2001
From: Lomanic <Lomanic@users.noreply.github.com>
Date: Wed, 14 Aug 2019 23:52:41 +0200
Subject: [PATCH] [cpu][windows] Add support for logical arg in Counts #640
 #628

---
 cpu/cpu_windows.go | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/cpu/cpu_windows.go b/cpu/cpu_windows.go
index 4807964..97c0e34 100644
--- a/cpu/cpu_windows.go
+++ b/cpu/cpu_windows.go
@@ -5,7 +5,6 @@ package cpu
 import (
 	"context"
 	"fmt"
-	"runtime"
 	"unsafe"
 
 	"github.com/StackExchange/wmi"
@@ -24,6 +23,7 @@ type Win32_Processor struct {
 	Manufacturer              string
 	Name                      string
 	NumberOfLogicalProcessors uint32
+	NumberOfCores             uint32
 	ProcessorID               *string
 	Stepping                  *string
 	MaxClockSpeed             uint32
@@ -241,5 +241,15 @@ func CountsWithContext(ctx context.Context, logical bool) (int, error) {
 		return int(systemInfo.dwNumberOfProcessors), nil
 	}
 	// physical cores https://github.com/giampaolo/psutil/blob/d01a9eaa35a8aadf6c519839e987a49d8be2d891/psutil/_psutil_windows.c#L499
-	return runtime.NumCPU(), nil
+	// for the time being, try with unreliable and slow WMI call…
+	var dst []Win32_Processor
+	q := wmi.CreateQuery(&dst, "")
+	if err := common.WMIQueryWithContext(ctx, q, &dst); err != nil {
+		return 0, err
+	}
+	var count uint32
+	for _, d := range dst {
+		count += d.NumberOfCores
+	}
+	return int(count), nil
 }