remove context cancellation

pull/1839/head
Stefano Balzarotti 3 weeks ago
parent d46c4a822d
commit ab48e6b803
No known key found for this signature in database
GPG Key ID: 7E7FC24DA4B9F8C9

@ -226,7 +226,7 @@ type systemLogicalProcessorInformationEx struct {
Processor processorRelationship Processor processorRelationship
} }
func getPhysicalCoreCount(ctx context.Context) (int, error) { func getPhysicalCoreCount() (int, error) {
var length uint32 var length uint32
const relationAll = 0xffff const relationAll = 0xffff
const relationProcessorCore = 0x0 const relationProcessorCore = 0x0
@ -250,17 +250,12 @@ func getPhysicalCoreCount(ctx context.Context) (int, error) {
offset := uintptr(0) offset := uintptr(0)
ncpus := 0 ncpus := 0
for offset < uintptr(length) { for offset < uintptr(length) {
select {
case <-ctx.Done():
return 0, ctx.Err()
default:
info := (*systemLogicalProcessorInformationEx)(unsafe.Pointer(uintptr(unsafe.Pointer(&buffer[0])) + offset)) info := (*systemLogicalProcessorInformationEx)(unsafe.Pointer(uintptr(unsafe.Pointer(&buffer[0])) + offset))
if info.Relationship == relationProcessorCore { if info.Relationship == relationProcessorCore {
ncpus++ ncpus++
} }
offset += uintptr(info.Size) offset += uintptr(info.Size)
} }
}
return ncpus, nil return ncpus, nil
} }
@ -282,5 +277,5 @@ func CountsWithContext(ctx context.Context, logical bool) (int, error) {
} }
// Get physical core count https://github.com/giampaolo/psutil/blob/d01a9eaa35a8aadf6c519839e987a49d8be2d891/psutil/_psutil_windows.c#L499 // Get physical core count https://github.com/giampaolo/psutil/blob/d01a9eaa35a8aadf6c519839e987a49d8be2d891/psutil/_psutil_windows.c#L499
return getPhysicalCoreCount(ctx) return getPhysicalCoreCount()
} }

Loading…
Cancel
Save