mirror of
https://github.com/lzh-1625/go_process_manager.git
synced 2025-10-31 11:26:49 +08:00
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
package cgroup_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
_ "github.com/lzh-1625/go_process_manager/boot"
|
|
"github.com/lzh-1625/go_process_manager/internal/app/constants"
|
|
"github.com/lzh-1625/go_process_manager/internal/app/model"
|
|
"github.com/lzh-1625/go_process_manager/internal/app/service"
|
|
|
|
"github.com/containerd/cgroups/v3/cgroup1"
|
|
"github.com/opencontainers/runtime-spec/specs-go"
|
|
)
|
|
|
|
func TestCgroup(t *testing.T) {
|
|
period := uint64(100000) // 100ms = 100000微秒
|
|
// 设置 CPU 配额为 20% (20ms)
|
|
quota := int64(20000 * 8) // 20ms = 20000微秒
|
|
control, err := cgroup1.New(cgroup1.StaticPath("/test"), &specs.LinuxResources{
|
|
CPU: &specs.LinuxCPU{
|
|
Period: &period,
|
|
Quota: "a,
|
|
},
|
|
Memory: &specs.LinuxMemory{},
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer control.Delete()
|
|
p, err := service.ProcessCtlService.RunNewProcess(model.Process{
|
|
Name: "test",
|
|
Cmd: "bash",
|
|
Cwd: `/root`,
|
|
TermType: constants.TERMINAL_PTY,
|
|
})
|
|
if err != nil {
|
|
t.FailNow()
|
|
}
|
|
control.AddProc(uint64(p.Pid))
|
|
time.Sleep(time.Second * 20)
|
|
p.Kill()
|
|
control.Delete()
|
|
}
|