67 lines
2.4 KiB
Go
67 lines
2.4 KiB
Go
//go:build !windows && !linux
|
|
// +build !windows,!linux
|
|
|
|
/*
|
|
Copyright The containerd Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package server
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/containerd/containerd"
|
|
"github.com/containerd/containerd/containers"
|
|
"github.com/containerd/typeurl"
|
|
runtimespec "github.com/opencontainers/runtime-spec/specs-go"
|
|
runtime "k8s.io/cri-api/pkg/apis/runtime/v1"
|
|
|
|
containerstore "github.com/containerd/containerd/pkg/cri/store/container"
|
|
)
|
|
|
|
// UpdateContainerResources updates ContainerConfig of the container.
|
|
func (c *criService) UpdateContainerResources(ctx context.Context, r *runtime.UpdateContainerResourcesRequest) (retRes *runtime.UpdateContainerResourcesResponse, retErr error) {
|
|
container, err := c.containerStore.Get(r.GetContainerId())
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to find container: %w", err)
|
|
}
|
|
// Update resources in status update transaction, so that:
|
|
// 1) There won't be race condition with container start.
|
|
// 2) There won't be concurrent resource update to the same container.
|
|
if err := container.Status.Update(func(status containerstore.Status) (containerstore.Status, error) {
|
|
return status, nil
|
|
}); err != nil {
|
|
return nil, fmt.Errorf("failed to update resources: %w", err)
|
|
}
|
|
return &runtime.UpdateContainerResourcesResponse{}, nil
|
|
}
|
|
|
|
// TODO: Copied from container_update_resources.go because that file is not built for darwin.
|
|
// updateContainerSpec updates container spec.
|
|
func updateContainerSpec(ctx context.Context, cntr containerd.Container, spec *runtimespec.Spec) error {
|
|
any, err := typeurl.MarshalAny(spec)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to marshal spec %+v: %w", spec, err)
|
|
}
|
|
if err := cntr.Update(ctx, func(ctx context.Context, client *containerd.Client, c *containers.Container) error {
|
|
c.Spec = any
|
|
return nil
|
|
}); err != nil {
|
|
return fmt.Errorf("failed to update container spec: %w", err)
|
|
}
|
|
return nil
|
|
}
|