Merge pull request #916 from devlights/add-gopsutil-cpu

Author: devlights

examples/psutil/README.md

@@ -0,0 +1,5 @@
+# これは何？
+
+[gopsutil](https://github.com/shirou/gopsutil) のサンプルを配置しています。
+
+pythonのpsutilをGoにポートされているライブラリで、非常に便利です。

examples/psutil/cpu/percent/.gitignore

@@ -0,0 +1 @@
+app

examples/psutil/cpu/percent/README.md

@@ -0,0 +1,26 @@
+# これは何？
+
+[gopsutil](https://github.com/shirou/gopsutil)を用いてCPU使用率を取得するサンプルです。
+
+```sh
+$ task
+task: [default] go build -o app .
+task: [default] ./app
+06:01:48 [4]
+06:01:49 [6]
+06:01:50 [8]
+06:01:51 [6]
+06:01:52 [10]
+task: [default] ./app -percpu
+06:01:54 [4 5]
+06:01:55 [5 6]
+06:01:56 [6 7]
+06:01:57 [8 9]
+06:01:58 [2 3]
+task: [default] ./app -percpu -spincpu
+06:02:00 [83 85]
+06:02:01 [84 82]
+06:02:02 [85 82]
+06:02:03 [84 82]
+06:02:04 [85 81]
+```

examples/psutil/cpu/percent/Taskfile.yml

@@ -0,0 +1,11 @@
+# https://taskfile.dev
+
+version: '3'
+
+tasks:
+  default:
+    cmds:
+      - go build -o app .
+      - ./app
+      - ./app -percpu
+      - ./app -percpu -spincpu

examples/psutil/cpu/percent/main.go

@@ -0,0 +1,111 @@
+package main
+
+import (
+	"context"
+	"errors"
+	"flag"
+	"log"
+	"math"
+	"runtime"
+	"time"
+
+	"github.com/shirou/gopsutil/v4/cpu"
+)
+
+type (
+	Args struct {
+		perCpu  bool
+		spinCpu bool
+	}
+)
+
+var (
+	args Args
+)
+
+func init() {
+	flag.BoolVar(&args.perCpu, "percpu", false, "per-cpu")
+	flag.BoolVar(&args.spinCpu, "spincpu", false, "spin-cpu")
+}
+
+func main() {
+	log.SetFlags(log.Ltime)
+	flag.Parse()
+
+	if err := run(); err != nil {
+		log.Fatal(err)
+	}
+}
+
+func run() error {
+	type (
+		cpuval struct {
+			val []int
+			err error
+		}
+	)
+	var (
+		interval = 1 * time.Second
+		percpu   = args.perCpu
+		cpuCh    = make(chan *cpuval)
+		ctx, cxl = context.WithTimeout(context.Background(), 6*time.Second)
+	)
+	defer cxl()
+	defer close(cpuCh)
+
+	// 必要であれば無駄にCPUを回す
+	if args.spinCpu {
+		go spinCpu(ctx)
+	}
+
+	// CPU使用率を取得
+	go func(ctx context.Context, ch chan<- *cpuval) {
+		for {
+			select {
+			case <-ctx.Done():
+				return
+			default:
+				v, err := cpu.PercentWithContext(ctx, interval, percpu)
+
+				vals := make([]int, len(v))
+				for i, f := range v {
+					vals[i] = int(f)
+				}
+
+				cpuCh <- &cpuval{vals, err}
+			}
+		}
+	}(ctx, cpuCh)
+
+	// 出力
+	for v := range cpuCh {
+		if v.err != nil {
+			if errors.Is(v.err, context.DeadlineExceeded) {
+				return nil
+			}
+			return v.err
+		}
+
+		log.Printf("%v\n", v.val)
+	}
+
+	return nil
+}
+
+func spinCpu(ctx context.Context) {
+	for range runtime.NumCPU() {
+		go func(ctx context.Context) {
+			for {
+				select {
+				case <-ctx.Done():
+					return
+				default:
+					for i := range 100000 {
+						_ = math.Sqrt(float64(i * i))
+					}
+					time.Sleep(1 * time.Microsecond)
+				}
+			}
+		}(ctx)
+	}
+}

go.mod

@@ -20,4 +20,17 @@ require (
 
 require github.com/muesli/cancelreader v0.2.2
 
-require github.com/devlights/fdpassing v1.0.1
+require (
+	github.com/devlights/fdpassing v1.0.1
+	github.com/shirou/gopsutil/v4 v4.25.3
+)
+
+require (
+	github.com/ebitengine/purego v0.8.2 // indirect
+	github.com/go-ole/go-ole v1.2.6 // indirect
+	github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 // indirect
+	github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c // indirect
+	github.com/tklauser/go-sysconf v0.3.12 // indirect
+	github.com/tklauser/numcpus v0.6.1 // indirect
+	github.com/yusufpapurcu/wmi v1.2.4 // indirect
+)

go.sum

@@ -4,27 +4,47 @@ github.com/devlights/fdpassing v1.0.1 h1:O9JBm3wpay8NIbjteVV7elZwtqi6ie+X1PbH1WR
 github.com/devlights/fdpassing v1.0.1/go.mod h1:eNoNi77gbfz/CsUtmnH3kZ64HVcK3ttFsKYxJopSPZ4=
 github.com/devlights/gomy v0.6.0 h1:7BT8bSxr+ZeNkgEYNufuM2rSc6kIoN6g2FSZvrcT9zw=
 github.com/devlights/gomy v0.6.0/go.mod h1:d28qyQ+/s7JravMlss2kIFxWxhY2KLlpe7rIBZ5YKeA=
+github.com/ebitengine/purego v0.8.2 h1:jPPGWs2sZ1UgOSgD2bClL0MJIqu58nOmIcBuXr62z1I=
+github.com/ebitengine/purego v0.8.2/go.mod h1:iIjxzd6CiRiOG0UyXP+V1+jWqUXVjPKLAI0mRfJZTmQ=
+github.com/go-ole/go-ole v1.2.6 h1:/Fpf6oFPoeFik9ty7siob0G6Ke8QvQEuVcuChpwXzpY=
+github.com/go-ole/go-ole v1.2.6/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=
 github.com/google/go-cmp v0.5.6/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
-github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=
-github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
+github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
+github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
 github.com/integrii/flaggy v1.5.2 h1:bWV20MQEngo4hWhno3i5Z9ISPxLPKj9NOGNwTWb/8IQ=
 github.com/integrii/flaggy v1.5.2/go.mod h1:dO13u7SYuhk910nayCJ+s1DeAAGC1THCMj1uSFmwtQ8=
+github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 h1:6E+4a0GO5zZEnZ81pIr0yLvtUWk2if982qA3F3QD6H4=
+github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0/go.mod h1:zJYVVT2jmtg6P3p1VtQj7WsuWi/y4VnjVBn7F8KPB3I=
 github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA=
 github.com/muesli/cancelreader v0.2.2/go.mod h1:3XuTXfFS2VjM+HTLZY9Ak0l6eUKfijIfMUZ4EgX0QYo=
 github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M=
 github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c h1:ncq/mPwQF4JjgDlrVEn3C11VoGHZN7m8qihwgMEtzYw=
+github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c/go.mod h1:OmDBASR4679mdNQnz2pUhc2G8CO2JrUAVFDRBDP/hJE=
+github.com/shirou/gopsutil/v4 v4.25.3 h1:SeA68lsu8gLggyMbmCn8cmp97V1TI9ld9sVzAUcKcKE=
+github.com/shirou/gopsutil/v4 v4.25.3/go.mod h1:xbuxyoZj+UsgnZrENu3lQivsngRR5BdjbJwf2fv4szA=
 github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k=
 github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME=
-github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
-github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
+github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/tklauser/go-sysconf v0.3.12 h1:0QaGUFOdQaIVdPgfITYzaTegZvdCjmYO52cSFAEVmqU=
+github.com/tklauser/go-sysconf v0.3.12/go.mod h1:Ho14jnntGE1fpdOqQEEaiKRpvIavV0hSfmBq8nJbHYI=
+github.com/tklauser/numcpus v0.6.1 h1:ng9scYS7az0Bk4OZLvrNXNSAO2Pxr1XXRAPyjhIx+Fk=
+github.com/tklauser/numcpus v0.6.1/go.mod h1:1XfjsgE2zo8GVw7POkMbHENHzVg3GzmoZ9fESEdAacY=
+github.com/yusufpapurcu/wmi v1.2.4 h1:zFUKzehAFReQwLys1b/iSMl+JQGSCSjtVqQn9bBrPo0=
+github.com/yusufpapurcu/wmi v1.2.4/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
 golang.org/x/crypto v0.36.0 h1:AnAEvhDddvBdpY+uR+MyHmuZzzNqXSe/GvuDeob5L34=
 golang.org/x/crypto v0.36.0/go.mod h1:Y4J0ReaxCR1IMaabaSMugxJES1EpwhBHhv2bDHklZvc=
 golang.org/x/exp v0.0.0-20240904232852-e7e105dedf7e h1:I88y4caeGeuDQxgdoFPUq097j7kNfw6uvuiNxUBfcBk=
 golang.org/x/exp v0.0.0-20240904232852-e7e105dedf7e/go.mod h1:akd2r19cwCdwSwWeIdzYQGa/EZZyqcOdwWiwj5L5eKQ=
 golang.org/x/sync v0.12.0 h1:MHc5BpPuC30uJk597Ri8TV3CNZcTLu6B6z4lJy+g6Jw=
 golang.org/x/sync v0.12.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
+golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20201204225414-ed752295db88/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.32.0 h1:s77OFDvIQeibCmezSnk/q6iAfkdiQaJi4VzroCFrN20=
 golang.org/x/sys v0.32.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
 golang.org/x/term v0.30.0 h1:PQ39fJZ+mfadBm0y5WlL4vlM7Sx1Hgf13sMIY2+QS9Y=