From 0e20b76bc9d3a3efe3c0671686919ffe4fdf90e5 Mon Sep 17 00:00:00 2001 From: silenceshell Date: Sun, 12 Sep 2021 20:00:13 +0800 Subject: [PATCH] v0.0.1 --- .gitignore | 2 + README.md | 4 +- cmd/topic/topic.go | 109 +++++++++++++++++++++++++++++++++ go.mod | 9 +++ go.sum | 78 ++++++++++++++++++++++++ pkg/cpu.go | 42 +++++++++++++ pkg/load.go | 5 ++ pkg/memory.go | 41 +++++++++++++ pkg/tasks.go | 146 +++++++++++++++++++++++++++++++++++++++++++++ pkg/user.go | 29 +++++++++ pkg/util.go | 20 +++++++ 11 files changed, 483 insertions(+), 2 deletions(-) create mode 100644 .gitignore create mode 100644 cmd/topic/topic.go create mode 100644 go.mod create mode 100644 go.sum create mode 100644 pkg/cpu.go create mode 100644 pkg/load.go create mode 100644 pkg/memory.go create mode 100644 pkg/tasks.go create mode 100644 pkg/user.go create mode 100644 pkg/util.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..d395c9a --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +.idea +cmd/topic/topic diff --git a/README.md b/README.md index 5cdbaf3..2904333 100644 --- a/README.md +++ b/README.md @@ -1,3 +1,3 @@ -# ctop +# topic -top command running in container. \ No newline at end of file +top in a container. diff --git a/cmd/topic/topic.go b/cmd/topic/topic.go new file mode 100644 index 0000000..7cc067f --- /dev/null +++ b/cmd/topic/topic.go @@ -0,0 +1,109 @@ +package main + +import ( + "fmt" + linuxproc "github.com/c9s/goprocinfo/linux" + ui "github.com/gizak/termui/v3" + "github.com/gizak/termui/v3/widgets" + "log" + "strings" + "time" + "topic/pkg" +) + +const ( + menuPrint = " PID USER PR NI VIRT RES SHR S %CPU %MEM TIME+ COMMAND" +) +var ( + termWidth = 0 + termHeight = 0 +) + +func genSummary(stat *linuxproc.Stat) []string { + currentTime := time.Now().Local().Format("15:04:05") + upTime := pkg.GetContainerUpTime(stat) + totalTask, running, sleeping, stopped, zombie := pkg.GetTaskCount() + userCpu, systemCpu, idleCpu := pkg.GetCpuUsage() + totalMem, freeMem, usedMem, cacheMem := pkg.GetTotalMemInMiB() + avail := freeMem + cacheMem + return []string{ + fmt.Sprintf("top - %v up %s, %d users, load average: %s", currentTime, upTime, pkg.GetUsers(), pkg.GetLoad()), + fmt.Sprintf("Tasks: [%3d](mod:bold) total, [%3d](mod:bold) running, [%3d](mod:bold) sleeping, [%3d](mod:bold) stopped, [%3d](mod:bold) zombie", + totalTask, running, sleeping, stopped, zombie), + fmt.Sprintf("%%Cpu(s): [%2.1f](mod:bold) us, [%2.1f](mod:bold) sy, [0.0](mod:bold) ni, [%2.1f](mod:bold) id, [0.0](mod:bold) wa, [0.0 hi,](mod:bold) [0.0](mod:bold) si, [0.0](mod:bold) st", + userCpu, systemCpu, idleCpu), + fmt.Sprintf("MiB Mem : [%7.1f](mod:bold) total, [%7.1f](mod:bold) free, [%7.1f](mod:bold) used, [%7.1f](mod:bold) buff/cache", + totalMem, freeMem, usedMem, cacheMem), + fmt.Sprintf("MiB Swap: [0](mod:bold) total, [0](mod:bold) free, [0](mod:bold) used. [%7.1f](mod:bold) avail Mem", avail), + } +} + +func genMenu() string { + spaceFmt := fmt.Sprintf("%%%vs", termWidth-len(menuPrint)-1) + paddingSpace := fmt.Sprintf(spaceFmt, " ") + return fmt.Sprintf("[%s%s](fg:black,bg:white)", menuPrint, paddingSpace) +} + +func genProcesses(taskMonitor *pkg.TaskMonitor) string { + result := []string{ + genMenu(), + } + result = append(result, taskMonitor.GetTaskInfos()...) + return strings.Join(result, "\n") +} + +func main() { + if err := ui.Init(); err != nil { + log.Fatalf("failed to initialize termui: %v", err) + } + defer ui.Close() + + stat, err := linuxproc.ReadStat("/proc/stat") + if err != nil { + log.Fatal("stat read fail") + } + + termWidth, termHeight = ui.TerminalDimensions() + + taskMonitor := pkg.NewTaskMonitor(stat) + + summary := widgets.NewList() + summary.Rows = genSummary(stat) + summary.TextStyle = ui.NewStyle(ui.ColorWhite) + summary.WrapText = false + summary.SetRect(-1, -1, termWidth, termHeight) + summary.Border = false + + processes := widgets.NewParagraph() + processes.Text = genProcesses(taskMonitor) + processes.WrapText = false + processes.SetRect(-1, 5, termWidth, termHeight) + processes.Border = false + + draw := func() { + stat, err := linuxproc.ReadStat("/proc/stat") + if err != nil { + log.Fatal("stat read fail") + } + + termWidth, termHeight = ui.TerminalDimensions() + summary.Rows = genSummary(stat) + processes.Text = genProcesses(taskMonitor) + ui.Render(summary, processes) + } + draw() + + uiEvents := ui.PollEvents() + ticker := time.NewTicker(time.Second).C + for { + select { + case e := <-uiEvents: + switch e.ID { + case "q", "": + return + } + case <-ticker: + draw() + } + } +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..e28cdc6 --- /dev/null +++ b/go.mod @@ -0,0 +1,9 @@ +module topic + +go 1.16 + +require ( + github.com/c9s/goprocinfo v0.0.0-20210130143923-c95fcf8c64a8 + github.com/containerd/cgroups v1.0.1 + github.com/gizak/termui/v3 v3.1.0 +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..f81e622 --- /dev/null +++ b/go.sum @@ -0,0 +1,78 @@ +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/c9s/goprocinfo v0.0.0-20210130143923-c95fcf8c64a8 h1:SjZ2GvvOononHOpK84APFuMvxqsk3tEIaKH/z4Rpu3g= +github.com/c9s/goprocinfo v0.0.0-20210130143923-c95fcf8c64a8/go.mod h1:uEyr4WpAH4hio6LFriaPkL938XnrvLpNPmQHBdrmbIE= +github.com/cilium/ebpf v0.4.0/go.mod h1:4tRaxcgiL706VnOzHOdBlY8IEAIdxINsQBcU4xJJXRs= +github.com/containerd/cgroups v1.0.1 h1:iJnMvco9XGvKUvNQkv88bE4uJXxRQH18efbKo9w5vHQ= +github.com/containerd/cgroups v1.0.1/go.mod h1:0SJrPIenamHDcZhEcJMNBB85rHcUsw4f25ZfBiPYRkU= +github.com/coreos/go-systemd/v22 v22.1.0 h1:kq/SbG2BCKLkDKkjQf5OWwKWUKj1lgs3lFI4PxnR5lg= +github.com/coreos/go-systemd/v22 v22.1.0/go.mod h1:xO0FLkIi5MaZafQlIrOotqXZ90ih+1atmu1JpKERPPk= +github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU= +github.com/cpuguy83/go-md2man/v2 v2.0.0/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/docker/go-units v0.4.0 h1:3uh0PgVws3nIA0Q+MwDC8yjEPf9zjRfZZWXZYDct3Tw= +github.com/docker/go-units v0.4.0/go.mod h1:fgPhTUdO+D/Jk86RDLlptpiXQzgHJF7gydDDbaIK4Dk= +github.com/frankban/quicktest v1.11.3/go.mod h1:wRf/ReqHper53s+kmmSZizM8NamnL3IM0I9ntUbOk+k= +github.com/gizak/termui/v3 v3.1.0 h1:ZZmVDgwHl7gR7elfKf1xc4IudXZ5qqfDh4wExk4Iajc= +github.com/gizak/termui/v3 v3.1.0/go.mod h1:bXQEBkJpzxUAKf0+xq9MSWAvWZlE7c+aidmyFlkYTrY= +github.com/godbus/dbus/v5 v5.0.3 h1:ZqHaoEF7TBzh4jzPmqVhE/5A1z9of6orkAe5uHoAeME= +github.com/godbus/dbus/v5 v5.0.3/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA= +github.com/gogo/protobuf v1.3.2 h1:Ov1cvc58UF3b5XjBnZv7+opcTcQFZebYjWzi34vdm4Q= +github.com/gogo/protobuf v1.3.2/go.mod h1:P1XiOD3dCwIKUDQYPy72D8LYyHL2YPYrpS2s69NZV8Q= +github.com/google/go-cmp v0.5.4/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/kisielk/errcheck v1.5.0/go.mod h1:pFxgyoBC7bSaBwPgfKdkLd5X25qrDl4LWUI2bnpBCr8= +github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= +github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/mattn/go-runewidth v0.0.2 h1:UnlwIPBGaTZfPQ6T1IGzPI0EkYAQmT9fAEJ/poFC63o= +github.com/mattn/go-runewidth v0.0.2/go.mod h1:LwmH8dsx7+W8Uxz3IHJYH5QSwggIsqBzpuz5H//U1FU= +github.com/mitchellh/go-wordwrap v0.0.0-20150314170334-ad45545899c7 h1:DpOJ2HYzCv8LZP15IdmG+YdwD2luVPHITV96TkirNBM= +github.com/mitchellh/go-wordwrap v0.0.0-20150314170334-ad45545899c7/go.mod h1:ZXFpozHsX6DPmq2I0TCekCxypsnAUbP2oI0UX1GXzOo= +github.com/nsf/termbox-go v0.0.0-20190121233118-02980233997d h1:x3S6kxmy49zXVVyhcnrFqxvNVCBPb2KZ9hV2RBdS840= +github.com/nsf/termbox-go v0.0.0-20190121233118-02980233997d/go.mod h1:IuKpRQcYE1Tfu+oAQqaLisqDeXgjyyltCfsaoYN18NQ= +github.com/opencontainers/runtime-spec v1.0.2 h1:UfAcuLBJB9Coz72x1hgl8O5RVzTdNiaglX6v2DM6FI0= +github.com/opencontainers/runtime-spec v1.0.2/go.mod h1:jwyrGlmzljRJv/Fgzds9SsS/C5hL+LL3ko9hs6T5lQ0= +github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4= +github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= +github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc= +github.com/sirupsen/logrus v1.7.0/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= +github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/urfave/cli v1.22.2/go.mod h1:Gos4lmkARVdJ6EkW0WaNv/tZAAMe9V7XWyB60NtXRu0= +github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c h1:VwygUrnw9jn88c4u8GD3rZQbqrP/tgas88tPUbBxQrk= +golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20200619180055-7c47624df98f/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE= +golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4fOWvOPXz4eW1wZW4PmDk9uLelYpA= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/pkg/cpu.go b/pkg/cpu.go new file mode 100644 index 0000000..2fc7545 --- /dev/null +++ b/pkg/cpu.go @@ -0,0 +1,42 @@ +package pkg + +import linuxproc "github.com/c9s/goprocinfo/linux" + +var ( + prevUser int64 = 0 + prevSystem int64 = 0 +) + +func GetCpuCount(stat *linuxproc.Stat) (count float64) { + cfsQuota := getCgoupValueByPath("/sys/fs/cgroup/cpu/cpu.cfs_quota_us") + cfsPeriod := getCgoupValueByPath("/sys/fs/cgroup/cpu/cpu.cfs_period_us") + + if cfsPeriod == -1 { + return float64(len(stat.CPUStats)) + } + + return float64(cfsPeriod / cfsQuota) +} + +// should be called every 1 seconds. not quite precise. +func GetCpuUsage() (user, system, idle float64) { + var user_, system_ int64 + user_ = getCgoupValueByPath("/sys/fs/cgroup/cpuacct/cpuacct.usage_user") + system_ = getCgoupValueByPath("/sys/fs/cgroup/cpuacct/cpuacct.usage_sys") + + if prevUser == 0 && prevSystem == 0 { + prevUser = user_ + prevSystem = system_ + return + } + + //todo: should divided by count of cpus. + user = float64(user_-prevUser) / 10000000 // / 1000,000,000 * 100 = /10,000,000 + system = float64(system_-prevSystem) / 10000000 // / 1000,000,000 * 100 = /10,000,000 + idle = 100 - user - system + + prevUser = user_ + prevSystem = system_ + + return +} diff --git a/pkg/load.go b/pkg/load.go new file mode 100644 index 0000000..0ff142a --- /dev/null +++ b/pkg/load.go @@ -0,0 +1,5 @@ +package pkg + +func GetLoad() string { + return "--, --, --" +} diff --git a/pkg/memory.go b/pkg/memory.go new file mode 100644 index 0000000..c840c66 --- /dev/null +++ b/pkg/memory.go @@ -0,0 +1,41 @@ +package pkg + +import ( + linuxproc "github.com/c9s/goprocinfo/linux" + "github.com/containerd/cgroups" + v1 "github.com/containerd/cgroups/stats/v1" +) + +func GetMemInfoInByte() (total, free, used, cache float64) { + stats := v1.Metrics{} + mem := cgroups.NewMemory("/sys/fs/cgroup/", cgroups.IgnoreModules("memsw")) + err := mem.Stat("", &stats) + if err != nil { + return + } + + hostMemInfo, err := linuxproc.ReadMemInfo("/proc/meminfo") + if err != nil { + return + } + + // convert Byte to MiB + if stats.Memory.Usage.Limit/1024 > hostMemInfo.MemTotal { + total = float64(hostMemInfo.MemTotal) * 1024 + } else { + total = float64(stats.Memory.Usage.Limit) + } + used = float64(stats.Memory.Usage.Usage) + cache = float64(stats.Memory.Cache) + free = total - used + return +} + +func GetTotalMemInMiB() (total, free, used, cache float64) { + total, free, used, cache = GetMemInfoInByte() + return total/1048576, free/1048576, used/1048576, cache/1048576 +} +func GetTotalMemInKiB() (total, free, used, cache float64) { + total, free, used, cache = GetMemInfoInByte() + return total/1024, free/1024, used/1024, cache/1024 +} \ No newline at end of file diff --git a/pkg/tasks.go b/pkg/tasks.go new file mode 100644 index 0000000..b9324a2 --- /dev/null +++ b/pkg/tasks.go @@ -0,0 +1,146 @@ +package pkg + +import ( + "fmt" + linuxproc "github.com/c9s/goprocinfo/linux" + "os" + "strconv" + "time" +) + +func GetContainerUpTime(stat *linuxproc.Stat) string { + btime := stat.BootTime + + process1, _ := linuxproc.ReadProcessStat("/proc/1/stat") + startTime := process1.Starttime + + uptime := btime.Add((time.Duration(startTime) * 10) * time.Millisecond) + + timeSince := time.Since(uptime) + if timeSince > time.Hour*24 { + days := timeSince / time.Hour / 24 + hours := timeSince / time.Hour % 24 + minutes := timeSince / time.Minute % 60 + return fmt.Sprintf("%d days, %d:%d", days, hours, minutes) + } + return uptime.Local().Format("15:04") +} + +func GetTaskCount() (total, running, sleeping, stopped, zombie int) { + f, err := os.Open("/proc") + if err != nil { + fmt.Println(err) + return + } + files, err := f.Readdir(0) + if err != nil { + fmt.Println(err) + return + } + + for _, v := range files { + fileName := v.Name() + if fileName[0] < '0' || fileName[0] > '9' { + continue + } + pid, err := strconv.Atoi(v.Name()) + if err == nil { + p, err := linuxproc.ReadProcessStat(fmt.Sprintf("/proc/%d/stat", pid)) + if err != nil { + return + } + switch p.State { + case "R": + running++ + case "t", "T": + stopped++ + case "Z": + zombie++ + default: + sleeping++ + } + total++ + } + } + return +} + +const ( + taskInfoFmt = "%5d %-8s%4d%4d%8d %6d %6d %s %4.1f %4.1f %8s %-s" +) + +func convertDuration(t time.Duration) string { + minute := t / time.Minute + t = t % time.Minute + second := t / time.Second + t = t % time.Second + milliseconds := t / time.Millisecond / 10 + return fmt.Sprintf("%d:%02d.%02d", minute, second, milliseconds) +} + +func (t *TaskMonitor) GetTaskInfos() (infos []string) { + f, err := os.Open("/proc") + if err != nil { + fmt.Println(err) + return + } + files, err := f.Readdir(0) + if err != nil { + fmt.Println(err) + return + } + + total, _, _, _ := GetTotalMemInKiB() + + taskInfos := make([]string, 0) + for _, v := range files { + fileName := v.Name() + if fileName[0] < '0' || fileName[0] > '9' { + continue + } + pid, err := strconv.Atoi(v.Name()) + if err == nil { + procInfo, err := linuxproc.ReadProcess(uint64(pid), "/proc") + if err != nil { + fmt.Println(err) + return + } + + // in KiB + virt := procInfo.Stat.Vsize / 1024 + res := procInfo.Statm.Resident * 4 + shr := procInfo.Statm.Share * 4 + + var cpuUsage uint64 + if t.taskPrevUser[pid] != 0 || t.taskPrevSystem[pid] != 0 { + cpuUsage = procInfo.Stat.Utime+procInfo.Stat.Stime - t.taskPrevUser[pid] - t.taskPrevSystem[pid] + } + t.taskPrevUser[pid] = procInfo.Stat.Utime + t.taskPrevSystem[pid] = procInfo.Stat.Stime + + memUsage := float64(res) / total * 100 + uptime := (time.Duration(procInfo.Stat.Utime+procInfo.Stat.Stime) * 10) * time.Millisecond + + taskInfo := fmt.Sprintf(taskInfoFmt, pid, "root", procInfo.Stat.Priority, procInfo.Stat.Nice, + virt, res, shr, procInfo.Stat.State, float64(cpuUsage), memUsage, + convertDuration(uptime), procInfo.Cmdline) + taskInfos = append(taskInfos, taskInfo) + } + } + return taskInfos +} + +type TaskMonitor struct { + stat *linuxproc.Stat + taskPrevUser map[int]uint64 + taskPrevSystem map[int]uint64 +} + +func NewTaskMonitor(stat *linuxproc.Stat) *TaskMonitor { + taskMonitor := TaskMonitor{ + stat: stat, + taskPrevUser: make(map[int]uint64), + taskPrevSystem: make(map[int]uint64), + } + return &taskMonitor +} \ No newline at end of file diff --git a/pkg/user.go b/pkg/user.go new file mode 100644 index 0000000..7c15693 --- /dev/null +++ b/pkg/user.go @@ -0,0 +1,29 @@ +package pkg + +import ( + "fmt" + "os" + "strconv" +) + +func GetUsers() int { + f, err := os.Open("/dev/pts") + if err != nil { + fmt.Println(err) + return 0 + } + files, err := f.Readdir(0) + if err != nil { + fmt.Println(err) + return 0 + } + + c := 0 + for _, v := range files { + _, err := strconv.Atoi(v.Name()) + if err == nil { + c++ + } + } + return c +} diff --git a/pkg/util.go b/pkg/util.go new file mode 100644 index 0000000..3cbaa42 --- /dev/null +++ b/pkg/util.go @@ -0,0 +1,20 @@ +package pkg + +import ( + "fmt" + "os" +) + +func getCgoupValueByPath(path string) int64 { + data, err := os.ReadFile(path) + if err != nil { + return 0 + } + + var value int64 + n, err := fmt.Sscanf(string(data), "%d", &value) + if err != nil || n != 1 { + return 0 + } + return value +} \ No newline at end of file