github/containerd
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

add cpu-shares to ctr

Browse Source 
This allows the cpu shares to be modified via ctr.

Signed-off-by: Michael Crosby <michael@thepasture.io>
This commit is contained in:
Michael Crosby 2021-08-09 12:21:21 -04:00
parent c7e4747cfb
commit d2f3b71468
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/ctr/commands/run/run_unix.go
View File

@ -70,6 +70,11 @@ var platformRunFlags = []cli.Flag{
Usage: "set the CFS cpu quota", Usage: "set the CFS cpu quota",
Value: 0.0, Value: 0.0,
}, },
cli.IntFlag{
Name: "cpu-shares",
Usage: "set the cpu shares",
Value: 1024,
},
cli.BoolFlag{ cli.BoolFlag{
Name: "cni", Name: "cni",
Usage: "enable cni networking for the container", Usage: "enable cni networking for the container",
@ -226,6 +231,10 @@ func NewContainer(ctx gocontext.Context, client *containerd.Client, context *cli
opts = append(opts, oci.WithCPUCFS(quota, period)) opts = append(opts, oci.WithCPUCFS(quota, period))
} }
if shares := context.Int("cpu-shares"); shares > 0 {
opts = append(opts, oci.WithCPUShares(uint64(shares)))
}
quota := context.Int64("cpu-quota") quota := context.Int64("cpu-quota")
period := context.Uint64("cpu-period") period := context.Uint64("cpu-period")
if quota != -1 || period != 0 { if quota != -1 || period != 0 {

8
oci/spec_opts_nonlinux.go
View File

@ -36,3 +36,11 @@ var WithAllCurrentCapabilities = func(ctx context.Context, client Client, c *con
var WithAllKnownCapabilities = func(ctx context.Context, client Client, c *containers.Container, s *Spec) error { var WithAllKnownCapabilities = func(ctx context.Context, client Client, c *containers.Container, s *Spec) error {
return WithCapabilities(nil)(ctx, client, c, s) return WithCapabilities(nil)(ctx, client, c, s)
} }
// WithCPUShares sets the container's cpu shares
//nolint: deadcode, unused
func WithCPUShares(shares uint64) SpecOpts {
return func(ctx context.Context, _ Client, c *containers.Container, s *Spec) error {
return nil
}
}