mirror of https://github.com/k3s-io/k3s
103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
// +build windows
|
|
|
|
/*
|
|
Copyright 2017 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package winstats
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
"unsafe"
|
|
|
|
"github.com/JeffAshton/win_pdh"
|
|
)
|
|
|
|
const (
|
|
cpuQuery = "\\Processor(_Total)\\% Processor Time"
|
|
memoryPrivWorkingSetQuery = "\\Process(_Total)\\Working Set - Private"
|
|
memoryCommittedBytesQuery = "\\Memory\\Committed Bytes"
|
|
// Perf counters are updated every second. This is the same as the default cadvisor collection period
|
|
// see https://github.com/google/cadvisor/blob/master/docs/runtime_options.md#housekeeping
|
|
perfCounterUpdatePeriod = 1 * time.Second
|
|
)
|
|
|
|
type perfCounter struct {
|
|
queryHandle win_pdh.PDH_HQUERY
|
|
counterHandle win_pdh.PDH_HCOUNTER
|
|
}
|
|
|
|
func newPerfCounter(counter string) (*perfCounter, error) {
|
|
var queryHandle win_pdh.PDH_HQUERY
|
|
var counterHandle win_pdh.PDH_HCOUNTER
|
|
|
|
ret := win_pdh.PdhOpenQuery(0, 0, &queryHandle)
|
|
if ret != win_pdh.ERROR_SUCCESS {
|
|
return nil, errors.New("unable to open query through DLL call")
|
|
}
|
|
|
|
ret = win_pdh.PdhValidatePath(counter)
|
|
if ret != win_pdh.ERROR_SUCCESS {
|
|
return nil, fmt.Errorf("unable to valid path to counter. Error code is %x", ret)
|
|
}
|
|
|
|
ret = win_pdh.PdhAddEnglishCounter(queryHandle, counter, 0, &counterHandle)
|
|
if ret != win_pdh.ERROR_SUCCESS {
|
|
return nil, fmt.Errorf("unable to add process counter. Error code is %x", ret)
|
|
}
|
|
|
|
ret = win_pdh.PdhCollectQueryData(queryHandle)
|
|
if ret != win_pdh.ERROR_SUCCESS {
|
|
return nil, fmt.Errorf("unable to collect data from counter. Error code is %x", ret)
|
|
}
|
|
|
|
return &perfCounter{
|
|
queryHandle: queryHandle,
|
|
counterHandle: counterHandle,
|
|
}, nil
|
|
}
|
|
|
|
func (p *perfCounter) getData() (uint64, error) {
|
|
ret := win_pdh.PdhCollectQueryData(p.queryHandle)
|
|
if ret != win_pdh.ERROR_SUCCESS {
|
|
return 0, fmt.Errorf("unable to collect data from counter. Error code is %x", ret)
|
|
}
|
|
|
|
var bufSize, bufCount uint32
|
|
var size = uint32(unsafe.Sizeof(win_pdh.PDH_FMT_COUNTERVALUE_ITEM_DOUBLE{}))
|
|
var emptyBuf [1]win_pdh.PDH_FMT_COUNTERVALUE_ITEM_DOUBLE // need at least 1 addressable null ptr.
|
|
var data uint64
|
|
|
|
ret = win_pdh.PdhGetFormattedCounterArrayDouble(p.counterHandle, &bufSize, &bufCount, &emptyBuf[0])
|
|
if ret != win_pdh.PDH_MORE_DATA {
|
|
return 0, fmt.Errorf("unable to collect data from counter. Error code is %x", ret)
|
|
}
|
|
|
|
filledBuf := make([]win_pdh.PDH_FMT_COUNTERVALUE_ITEM_DOUBLE, bufCount*size)
|
|
ret = win_pdh.PdhGetFormattedCounterArrayDouble(p.counterHandle, &bufSize, &bufCount, &filledBuf[0])
|
|
if ret != win_pdh.ERROR_SUCCESS {
|
|
return 0, fmt.Errorf("unable to collect data from counter. Error code is %x", ret)
|
|
}
|
|
|
|
for i := 0; i < int(bufCount); i++ {
|
|
c := filledBuf[i]
|
|
data = uint64(c.FmtValue.DoubleValue)
|
|
}
|
|
|
|
return data, nil
|
|
}
|