Function ConnectDaemons

integration/main_test.go:89–134

ConnectDaemons connect cri plugin and containerd, and initialize the clients.

()

// ConnectDaemons connect cri plugin and containerd, and initialize the clients.
func ConnectDaemons() error {
	fail := func(format string, args ...any) error {
		err := fmt.Errorf(format, args...)
		if cleanupErr := DisconnectDaemons(); cleanupErr != nil {
			return errors.Join(err, cleanupErr)
		}
		return err
	}

	var err error
	runtimeService, err = remote.NewRuntimeService(*criEndpoint, timeout)
	if err != nil {
		return fail("failed to create runtime service: %w", err)
	}
	runtimeService2, err = remote.NewRuntimeService(*criEndpoint, timeout)
	if err != nil {
		return fail("failed to create runtime service: %w", err)
	}
	imageService, err = remote.NewImageService(*criEndpoint, timeout)
	if err != nil {
		return fail("failed to create image service: %w", err)
	}
	// Since CRI grpc client doesn't have `WithBlock` specified, we
	// need to check whether it is actually connected.
	// TODO(#6069) Use grpc options to block on connect and remove for this list containers request.
	_, err = runtimeService.ListContainers(&runtime.ContainerFilter{})
	if err != nil {
		return fail("failed to list containers: %w", err)
	}
	_, err = runtimeService2.ListContainers(&runtime.ContainerFilter{})
	if err != nil {
		return fail("failed to list containers: %w", err)
	}
	_, err = imageService.ListImages(&runtime.ImageFilter{})
	if err != nil {
		return fail("failed to list images: %w", err)
	}
	// containerdEndpoint is the same with criEndpoint now
	containerdEndpoint = strings.TrimPrefix(*criEndpoint, "unix://")
	containerdEndpoint = strings.TrimPrefix(containerdEndpoint, "npipe:")
	containerdClient, err = containerd.New(containerdEndpoint, containerd.WithDefaultNamespace(k8sNamespace))
	if err != nil {
		return fail("failed to connect containerd: %w", err)
	}
	return nil
}

// DisconnectDaemons tears down the shared CRI and containerd clients.
func DisconnectDaemons() error {