Merge pull request #601 from Lomanic/issue551

Fix #551 Filter out '_Total' results from PerfInfoWithContext and unexport this function.

Also propagate context in perCPUTimes (renamed to perCPUTimesWithContext) and unexport Win32_PerfFormattedData_Counters_ProcessorInformation type.
pull/603/head
Lomanic 7 years ago committed by GitHub
commit 4254ce2081
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -23,8 +23,8 @@ type Win32_Processor struct {
MaxClockSpeed uint32 MaxClockSpeed uint32
} }
// Win32_PerfFormattedData_Counters_ProcessorInformation stores instance value of the perf counters // win32_PerfFormattedData_Counters_ProcessorInformation stores instance value of the perf counters
type Win32_PerfFormattedData_Counters_ProcessorInformation struct { type win32_PerfFormattedData_Counters_ProcessorInformation struct {
Name string Name string
PercentDPCTime uint64 PercentDPCTime uint64
PercentIdleTime uint64 PercentIdleTime uint64
@ -51,7 +51,7 @@ func Times(percpu bool) ([]TimesStat, error) {
func TimesWithContext(ctx context.Context, percpu bool) ([]TimesStat, error) { func TimesWithContext(ctx context.Context, percpu bool) ([]TimesStat, error) {
if percpu { if percpu {
return perCPUTimes() return perCPUTimesWithContext(ctx)
} }
var ret []TimesStat var ret []TimesStat
@ -119,17 +119,13 @@ func InfoWithContext(ctx context.Context) ([]InfoStat, error) {
// PerfInfo returns the performance counter's instance value for ProcessorInformation. // PerfInfo returns the performance counter's instance value for ProcessorInformation.
// Name property is the key by which overall, per cpu and per core metric is known. // Name property is the key by which overall, per cpu and per core metric is known.
func PerfInfo() ([]Win32_PerfFormattedData_Counters_ProcessorInformation, error) { func perfInfoWithContext(ctx context.Context) ([]win32_PerfFormattedData_Counters_ProcessorInformation, error) {
return PerfInfoWithContext(context.Background()) var ret []win32_PerfFormattedData_Counters_ProcessorInformation
}
func PerfInfoWithContext(ctx context.Context) ([]Win32_PerfFormattedData_Counters_ProcessorInformation, error) {
var ret []Win32_PerfFormattedData_Counters_ProcessorInformation
q := wmi.CreateQuery(&ret, "") q := wmi.CreateQuery(&ret, "WHERE NOT Name LIKE '%_Total'")
err := common.WMIQueryWithContext(ctx, q, &ret) err := common.WMIQueryWithContext(ctx, q, &ret)
if err != nil { if err != nil {
return []Win32_PerfFormattedData_Counters_ProcessorInformation{}, err return []win32_PerfFormattedData_Counters_ProcessorInformation{}, err
} }
return ret, err return ret, err
@ -152,9 +148,9 @@ func ProcInfoWithContext(ctx context.Context) ([]Win32_PerfFormattedData_PerfOS_
} }
// perCPUTimes returns times stat per cpu, per core and overall for all CPUs // perCPUTimes returns times stat per cpu, per core and overall for all CPUs
func perCPUTimes() ([]TimesStat, error) { func perCPUTimesWithContext(ctx context.Context) ([]TimesStat, error) {
var ret []TimesStat var ret []TimesStat
stats, err := PerfInfo() stats, err := perfInfoWithContext(ctx)
if err != nil { if err != nil {
return nil, err return nil, err
} }

Loading…
Cancel
Save