27 lines
485 B
Go
27 lines
485 B
Go
|
//go:build linux
|
||
|
|
|
||
|
|
package client
|
||
|
|
|
||
|
|
import (
|
||
|
|
"golang.org/x/sys/unix"
|
||
|
|
|
||
|
|
"repositories.action2quare.com/ayo/gocommon/logger"
|
||
|
|
)
|
||
|
|
|
||
|
|
func set_affinity(pid int, cpu int) {
|
||
|
|
var cpuset unix.CPUSet
|
||
|
|
err := unix.SchedGetaffinity(pid, &cpuset)
|
||
|
|
if err != nil {
|
||
|
|
logger.Println("SchedGetaffinity failed :", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
count := cpuset.Count()
|
||
|
|
cpuset.Zero()
|
||
|
|
cpuset.Set(cpu % count)
|
||
|
|
|
||
|
|
err = unix.SchedSetaffinity(pid, &cpuset)
|
||
|
|
if err != nil {
|
||
|
|
logger.Println("SchedSetaffinity failed :", err)
|
||
|
|
}
|
||
|
|
}
|