Code clean for nri_api.
Signed-off-by: yanggang <gang.yang@daocloud.io>
This commit is contained in:
		@@ -32,7 +32,6 @@ import (
 | 
			
		||||
	"github.com/containerd/containerd/v2/pkg/errdefs"
 | 
			
		||||
	"github.com/containerd/log"
 | 
			
		||||
	"github.com/containerd/typeurl/v2"
 | 
			
		||||
	"github.com/opencontainers/runtime-spec/specs-go"
 | 
			
		||||
	runtimespec "github.com/opencontainers/runtime-spec/specs-go"
 | 
			
		||||
	"github.com/opencontainers/runtime-tools/generate"
 | 
			
		||||
	cri "k8s.io/cri-api/pkg/apis/runtime/v1"
 | 
			
		||||
@@ -117,7 +116,7 @@ func (a *API) RemovePodSandbox(ctx context.Context, criPod *sstore.Sandbox) erro
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *API) CreateContainer(ctx context.Context, ctrs *containers.Container, spec *specs.Spec) (*api.ContainerAdjustment, error) {
 | 
			
		||||
func (a *API) CreateContainer(ctx context.Context, ctrs *containers.Container, spec *runtimespec.Spec) (*api.ContainerAdjustment, error) {
 | 
			
		||||
	ctr := a.nriContainer(ctrs, spec)
 | 
			
		||||
 | 
			
		||||
	criPod, err := a.cri.SandboxStore().Get(ctr.GetPodSandboxID())
 | 
			
		||||
@@ -256,7 +255,7 @@ func (a *API) RemoveContainer(ctx context.Context, criPod *sstore.Sandbox, criCt
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *API) UndoCreateContainer(ctx context.Context, criPod *sstore.Sandbox, id string, spec *specs.Spec) {
 | 
			
		||||
func (a *API) UndoCreateContainer(ctx context.Context, criPod *sstore.Sandbox, id string, spec *runtimespec.Spec) {
 | 
			
		||||
	if a.IsDisabled() {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
@@ -318,7 +317,7 @@ func (a *API) WithContainerAdjustment() containerd.NewContainerOpts {
 | 
			
		||||
	)
 | 
			
		||||
 | 
			
		||||
	return func(ctx context.Context, _ *containerd.Client, c *containers.Container) error {
 | 
			
		||||
		spec := &specs.Spec{}
 | 
			
		||||
		spec := &runtimespec.Spec{}
 | 
			
		||||
		if err := json.Unmarshal(c.Spec.GetValue(), spec); err != nil {
 | 
			
		||||
			return fmt.Errorf("failed to unmarshal container OCI Spec for NRI: %w", err)
 | 
			
		||||
		}
 | 
			
		||||
@@ -463,14 +462,14 @@ func (a *API) EvictContainer(ctx context.Context, e *api.ContainerEviction) erro
 | 
			
		||||
 | 
			
		||||
type criPodSandbox struct {
 | 
			
		||||
	*sstore.Sandbox
 | 
			
		||||
	spec *specs.Spec
 | 
			
		||||
	spec *runtimespec.Spec
 | 
			
		||||
	pid  uint32
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *API) nriPodSandbox(pod *sstore.Sandbox) *criPodSandbox {
 | 
			
		||||
	criPod := &criPodSandbox{
 | 
			
		||||
		Sandbox: pod,
 | 
			
		||||
		spec:    &specs.Spec{},
 | 
			
		||||
		spec:    &runtimespec.Spec{},
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if pod == nil || pod.Container == nil {
 | 
			
		||||
@@ -490,13 +489,10 @@ func (a *API) nriPodSandbox(pod *sstore.Sandbox) *criPodSandbox {
 | 
			
		||||
	criPod.pid = task.Pid()
 | 
			
		||||
	spec, err := task.Spec(ctx)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.L.WithError(err).Errorf("failed to get spec for sandbox container %s",
 | 
			
		||||
				pod.Container.ID())
 | 
			
		||||
		}
 | 
			
		||||
		log.L.WithError(err).Errorf("failed to get spec for sandbox container %s",
 | 
			
		||||
			pod.Container.ID())
 | 
			
		||||
		return criPod
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	criPod.spec = spec
 | 
			
		||||
 | 
			
		||||
	return criPod
 | 
			
		||||
@@ -644,12 +640,12 @@ func (p *criPodSandbox) GetPid() uint32 {
 | 
			
		||||
type criContainer struct {
 | 
			
		||||
	api  *API
 | 
			
		||||
	ctrs *containers.Container
 | 
			
		||||
	spec *specs.Spec
 | 
			
		||||
	spec *runtimespec.Spec
 | 
			
		||||
	meta *cstore.Metadata
 | 
			
		||||
	pid  uint32
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *API) nriContainer(ctr interface{}, spec *specs.Spec) *criContainer {
 | 
			
		||||
func (a *API) nriContainer(ctr interface{}, spec *runtimespec.Spec) *criContainer {
 | 
			
		||||
	switch c := ctr.(type) {
 | 
			
		||||
	case *cstore.Container:
 | 
			
		||||
		ctx := ctrdutil.NamespacedContext()
 | 
			
		||||
@@ -662,7 +658,7 @@ func (a *API) nriContainer(ctr interface{}, spec *specs.Spec) *criContainer {
 | 
			
		||||
		spec, err := ctrd.Spec(ctx)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			log.L.WithError(err).Errorf("failed to get OCI Spec for container %s", ctrd.ID())
 | 
			
		||||
			spec = &specs.Spec{}
 | 
			
		||||
			spec = &runtimespec.Spec{}
 | 
			
		||||
		}
 | 
			
		||||
		task, err := ctrd.Task(ctx, nil)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
@@ -703,7 +699,7 @@ func (a *API) nriContainer(ctr interface{}, spec *specs.Spec) *criContainer {
 | 
			
		||||
	return &criContainer{
 | 
			
		||||
		api:  a,
 | 
			
		||||
		meta: &cstore.Metadata{},
 | 
			
		||||
		spec: &specs.Spec{},
 | 
			
		||||
		spec: &runtimespec.Spec{},
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user