diff --git a/pidusage.go b/pidusage.go new file mode 100644 index 0000000..4e4b378 --- /dev/null +++ b/pidusage.go @@ -0,0 +1,152 @@ +package main + +import ( + "fmt" + "io/ioutil" + "math" + "os/exec" + "path" + "runtime" + "strconv" + "strings" +) + +// SysInfo will record cpu and memory data +type SysInfo struct { + CPU float64 + Memory float64 +} + +// Stat will store CUP time struct +type Stat struct { + utime float64 + stime float64 + cutime float64 + cstime float64 + start float64 + rss float64 + uptime float64 +} + +// type fn func(int) *SysInfo + +// var fnMap map[string]fn +var platform string +var history map[int]Stat + +// func wrapper(statType string) func(pid int) *SysInfo { +// return func(pid int) *SysInfo { +// return stat(pid, statType) +// } +// } +func init() { + platform = runtime.GOOS + if strings.Contains(platform, "win") { + platform = "win" + } + history = make(map[int]Stat) + // fnMap = make(map[string]fn) + // fnMap["darwin"] = wrapper("ps") + // fnMap["sunos"] = wrapper("ps") + // fnMap["freebsd"] = wrapper("ps") + // fnMap["aix"] = wrapper("ps") + // fnMap["linux"] = wrapper("proc") + // fnMap["netbsd"] = wrapper("proc") + // fnMap["win"] = wrapper("win") +} +func formatStdOut(stdout []byte, userfulIndex int) []string { + infoArr := strings.Split(string(stdout), "\n")[userfulIndex] + ret := strings.Fields(infoArr) + return ret +} + +func parseFloat(val string) float64 { + floatVal, _ := strconv.ParseFloat(val, 64) + return floatVal +} + +func stat(pid int, statType string) *SysInfo { + sysInfo := &SysInfo{} + _history := history[pid] + if statType == "ps" { + args := "-o pcpu,rss -p" + if platform == "aix" { + args = "-o pcpu,rssize -p" + } + stdout, _ := exec.Command("ps", args, strconv.Itoa(pid)).Output() + ret := formatStdOut(stdout, 1) + sysInfo.CPU = parseFloat(ret[0]) + sysInfo.Memory = parseFloat(ret[1]) + } else if statType == "proc" { + // default clkTck and pageSize + var clkTck float64 = 100 + var pageSize float64 = 4096 + + uptimeFileBytes, err := ioutil.ReadFile(path.Join("/proc", "uptime")) + uptime := parseFloat(strings.Split(string(uptimeFileBytes), " ")[0]) + fmt.Println("uptime", uptime) + + clkTckStdout, err := exec.Command("getconf", "CLK_TCK").Output() + if err == nil { + clkTck = parseFloat(formatStdOut(clkTckStdout, 0)[0]) + } + + pageSizeStdout, err := exec.Command("getconf", "PAGESIZE").Output() + if err == nil { + pageSize = parseFloat(formatStdOut(pageSizeStdout, 0)[0]) + } + + fmt.Println(clkTck, pageSize) + fmt.Println(path.Join("/proc", strconv.Itoa(pid), "stat")) + procStatFileBytes, err := ioutil.ReadFile(path.Join("/proc", strconv.Itoa(pid), "stat")) + infos := strings.Split(strings.SplitAfter(string(procStatFileBytes), ")")[1], " ") + stat := &Stat{ + utime: parseFloat(infos[12]), + stime: parseFloat(infos[13]), + cutime: parseFloat(infos[14]), + cstime: parseFloat(infos[15]), + start: parseFloat(infos[20]) / clkTck, + rss: parseFloat(infos[22]), + uptime: uptime, + } + + _stime := 0.0 + _utime := 0.0 + if _history.stime != 0.0 { + _stime = _history.stime + } + + if _history.utime != 0.0 { + _utime = _history.utime + } + total := stat.stime - _stime + stat.utime - _utime + total = total / clkTck + + seconds := stat.start - uptime + if _history.uptime != 0.0 { + seconds = uptime - _history.uptime + } + + seconds = math.Abs(seconds) + if seconds == 0 { + seconds = 1 + } + + history[pid] = *stat + fmt.Println(total) + sysInfo.CPU = (total / seconds) * 100 + sysInfo.Memory = stat.rss * pageSize + } + return sysInfo + +} + +// Stat will return current system CPU and memory data +// func Stat(pid int) *SysInfo { +// sysInfo := fnMap[platform](pid) +// return sysInfo +// } + +func main() { + stat(662, "proc") +}