This repository has been archived by the owner on Oct 12, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 16
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
ff44c52
commit 1d30fd3
Showing
6 changed files
with
112 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| package cpu |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| // +build linux | ||
|
|
||
| package cpu | ||
|
|
||
| import ( | ||
| psutils_cpu "github.com/shirou/gopsutil/cpu" | ||
| ) | ||
|
|
||
| func PerCpuPercent() ([]float64, error) { | ||
| return psutils_cpu.Percent(0, true) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,94 @@ | ||
| // +build windows | ||
|
|
||
| package cpu | ||
|
|
||
| import ( | ||
| "context" | ||
| "github.com/StackExchange/wmi" | ||
| "time" | ||
| ) | ||
|
|
||
| const Timeout = 3 * time.Second | ||
|
|
||
| type CPUStat struct { | ||
| value uint64 | ||
| timestamp uint64 | ||
| } | ||
|
|
||
| var lastCpus map[string]CPUStat = make(map[string]CPUStat) | ||
|
|
||
| type win32_PerfRawData_Counters_ProcessorInformation struct { | ||
| Name string | ||
| PercentDPCTime uint64 | ||
| PercentIdleTime uint64 | ||
| PercentUserTime uint64 | ||
| PercentProcessorTime uint64 | ||
| PercentInterruptTime uint64 | ||
| PercentPriorityTime uint64 | ||
| PercentPrivilegedTime uint64 | ||
| TimeStamp_Sys100NS uint64 | ||
| InterruptsPerSec uint32 | ||
| ProcessorFrequency uint32 | ||
| DPCRate uint32 | ||
| } | ||
|
|
||
| func PerCpuPercent() ([]float64, error) { | ||
| return perCPUPercentWithContext(context.Background()) | ||
| } | ||
|
|
||
| func perCPUPercentWithContext(ctx context.Context) ([]float64, error) { | ||
| var ret []float64 | ||
| stats, err := perfInfoWithContext(ctx) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| for _, v := range stats { | ||
| last := lastCpus[v.Name] | ||
|
|
||
| lastCpus[v.Name] = CPUStat{ | ||
| value: v.PercentProcessorTime, | ||
| timestamp: v.TimeStamp_Sys100NS, | ||
| } | ||
|
|
||
| if last.timestamp != 0 { | ||
| value := (1 - (float64(v.PercentProcessorTime-last.value) / float64(v.TimeStamp_Sys100NS-last.timestamp))) * 100 | ||
| ret = append(ret, value) | ||
| } | ||
| } | ||
| return ret, nil | ||
| } | ||
|
|
||
| // 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. | ||
| func perfInfoWithContext(ctx context.Context) ([]win32_PerfRawData_Counters_ProcessorInformation, error) { | ||
| var ret []win32_PerfRawData_Counters_ProcessorInformation | ||
|
|
||
| q := wmi.CreateQuery(&ret, "WHERE NOT Name LIKE '%_Total'") | ||
| err := WMIQueryWithContext(ctx, q, &ret) | ||
| if err != nil { | ||
| return []win32_PerfRawData_Counters_ProcessorInformation{}, err | ||
| } | ||
|
|
||
| return ret, err | ||
| } | ||
|
|
||
| func WMIQueryWithContext(ctx context.Context, query string, dst interface{}, connectServerArgs ...interface{}) error { | ||
| if _, ok := ctx.Deadline(); !ok { | ||
| ctxTimeout, cancel := context.WithTimeout(ctx, Timeout) | ||
| defer cancel() | ||
| ctx = ctxTimeout | ||
| } | ||
|
|
||
| errChan := make(chan error, 1) | ||
| go func() { | ||
| errChan <- wmi.Query(query, dst, connectServerArgs...) | ||
| }() | ||
|
|
||
| select { | ||
| case <-ctx.Done(): | ||
| return ctx.Err() | ||
| case err := <-errChan: | ||
| return err | ||
| } | ||
| } |