Vendor psutils and remove neko

2015-04-03 17:22:31 -07:00
parent db74acb86d
commit 9c42aea28c
84 changed files with 9370 additions and 74 deletions
--- a/plugins/system/ps/cpu/cpu_windows.go
+++ b/plugins/system/ps/cpu/cpu_windows.go
@@ -0,0 +1,101 @@
+// +build windows
+
+package cpu
+
+import (
+	"strconv"
+	"syscall"
+	"time"
+	"unsafe"
+
+	common "github.com/shirou/gopsutil/common"
+)
+
+// TODO: Get percpu
+func CPUTimes(percpu bool) ([]CPUTimesStat, error) {
+	var ret []CPUTimesStat
+
+	var lpIdleTime common.FILETIME
+	var lpKernelTime common.FILETIME
+	var lpUserTime common.FILETIME
+	r, _, _ := common.ProcGetSystemTimes.Call(
+		uintptr(unsafe.Pointer(&lpIdleTime)),
+		uintptr(unsafe.Pointer(&lpKernelTime)),
+		uintptr(unsafe.Pointer(&lpUserTime)))
+	if r == 0 {
+		return ret, syscall.GetLastError()
+	}
+
+	LOT := float64(0.0000001)
+	HIT := (LOT * 4294967296.0)
+	idle := ((HIT * float64(lpIdleTime.DwHighDateTime)) + (LOT * float64(lpIdleTime.DwLowDateTime)))
+	user := ((HIT * float64(lpUserTime.DwHighDateTime)) + (LOT * float64(lpUserTime.DwLowDateTime)))
+	kernel := ((HIT * float64(lpKernelTime.DwHighDateTime)) + (LOT * float64(lpKernelTime.DwLowDateTime)))
+	system := (kernel - idle)
+
+	ret = append(ret, CPUTimesStat{
+		Idle:   float64(idle),
+		User:   float64(user),
+		System: float64(system),
+	})
+	return ret, nil
+}
+
+func CPUInfo() ([]CPUInfoStat, error) {
+	var ret []CPUInfoStat
+	lines, err := common.GetWmic("cpu", "get", "Family,L2CacheSize,Manufacturer,Name,NumberOfLogicalProcessors,ProcessorId,Stepping")
+	if err != nil {
+		return ret, err
+	}
+	for i, t := range lines {
+		cache, err := strconv.Atoi(t[2])
+		if err != nil {
+			cache = 0
+		}
+		cores, err := strconv.Atoi(t[5])
+		if err != nil {
+			cores = 0
+		}
+		stepping := 0
+		if len(t) > 7 {
+			stepping, err = strconv.Atoi(t[6])
+			if err != nil {
+				stepping = 0
+			}
+		}
+		cpu := CPUInfoStat{
+			CPU:        int32(i),
+			Family:     t[1],
+			CacheSize:  int32(cache),
+			VendorID:   t[3],
+			ModelName:  t[4],
+			Cores:      int32(cores),
+			PhysicalID: t[6],
+			Stepping:   int32(stepping),
+			Flags:      []string{},
+		}
+		ret = append(ret, cpu)
+	}
+	return ret, nil
+}
+
+func CPUPercent(interval time.Duration, percpu bool) ([]float64, error) {
+	ret := []float64{}
+
+	lines, err := common.GetWmic("cpu", "get", "loadpercentage")
+	if err != nil {
+		return ret, err
+	}
+	for _, l := range lines {
+		if len(l) < 2 {
+			continue
+		}
+		p, err := strconv.Atoi(l[1])
+		if err != nil {
+			p = 0
+		}
+		// but windows can only get one percent.
+		ret = append(ret, float64(p)/100.0)
+	}
+	return ret, nil
+}