56 lines
1.7 KiB
Go
56 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/kubernetes-incubator/cri-o/oci"
|
|
rspec "github.com/opencontainers/runtime-spec/specs-go"
|
|
"github.com/sirupsen/logrus"
|
|
"golang.org/x/net/context"
|
|
pb "k8s.io/kubernetes/pkg/kubelet/apis/cri/v1alpha1/runtime"
|
|
)
|
|
|
|
// UpdateContainerResources updates ContainerConfig of the container.
|
|
func (s *Server) UpdateContainerResources(ctx context.Context, req *pb.UpdateContainerResourcesRequest) (resp *pb.UpdateContainerResourcesResponse, err error) {
|
|
const operation = "update_container_resources"
|
|
defer func() {
|
|
recordOperation(operation, time.Now())
|
|
recordError(operation, err)
|
|
}()
|
|
logrus.Debugf("UpdateContainerResources %+v", req)
|
|
|
|
c, err := s.GetContainerFromRequest(req.GetContainerId())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
state := s.Runtime().ContainerStatus(c)
|
|
if !(state.Status == oci.ContainerStateRunning || state.Status == oci.ContainerStateCreated) {
|
|
return nil, fmt.Errorf("container %s is not running or created state: %s", c.ID(), state.Status)
|
|
}
|
|
|
|
resources := toOCIResources(req.GetLinux())
|
|
if err := s.Runtime().UpdateContainer(c, resources); err != nil {
|
|
return nil, err
|
|
}
|
|
return &pb.UpdateContainerResourcesResponse{}, nil
|
|
}
|
|
|
|
// toOCIResources converts CRI resource constraints to OCI.
|
|
func toOCIResources(r *pb.LinuxContainerResources) *rspec.LinuxResources {
|
|
return &rspec.LinuxResources{
|
|
CPU: &rspec.LinuxCPU{
|
|
Shares: proto.Uint64(uint64(r.GetCpuShares())),
|
|
Quota: proto.Int64(r.GetCpuQuota()),
|
|
Period: proto.Uint64(uint64(r.GetCpuPeriod())),
|
|
Cpus: r.GetCpusetCpus(),
|
|
Mems: r.GetCpusetMems(),
|
|
},
|
|
Memory: &rspec.LinuxMemory{
|
|
Limit: proto.Int64(r.GetMemoryLimitInBytes()),
|
|
},
|
|
// TODO(runcom): OOMScoreAdj is missing
|
|
}
|
|
}
|