CRI Provider implementation (#195)

* First commit of CRI provider. Vendor deps not included * First commit of CRI provider. Vendor deps not included * Tidy up comments and format code * vendor grpc, CRI APIs, update protobuf and tidy logging * First commit of CRI provider. Vendor deps not included * Tidy up comments and format code * vendor grpc, CRI APIs, update protobuf and tidy logging * Add README * Fix errors in make test
2018-05-15 08:50:52 -07:00
parent 4b61932ac1
commit c6c89f062f
702 changed files with 456968 additions and 11190 deletions
--- a/providers/cri/client.go
+++ b/providers/cri/client.go
@@ -0,0 +1,181 @@
+package cri
+
+import (
+	"fmt"
+
+	log "github.com/Sirupsen/logrus"
+	"golang.org/x/net/context"
+	criapi "k8s.io/kubernetes/pkg/kubelet/apis/cri/runtime/v1alpha2"
+)
+
+// Call RunPodSandbox on the CRI client
+func runPodSandbox(client criapi.RuntimeServiceClient, config *criapi.PodSandboxConfig) (string, error) {
+	request := &criapi.RunPodSandboxRequest{Config: config}
+	log.Debugf("RunPodSandboxRequest: %v", request)
+	r, err := client.RunPodSandbox(context.Background(), request)
+	log.Debugf("RunPodSandboxResponse: %v", r)
+	if err != nil {
+		return "", err
+	}
+	log.Printf("New pod sandbox created: %v", r.PodSandboxId)
+	return r.PodSandboxId, nil
+}
+
+// Call StopPodSandbox on the CRI client
+func stopPodSandbox(client criapi.RuntimeServiceClient, id string) error {
+	if id == "" {
+		return fmt.Errorf("ID cannot be empty")
+	}
+	request := &criapi.StopPodSandboxRequest{PodSandboxId: id}
+	log.Debugf("StopPodSandboxRequest: %v", request)
+	r, err := client.StopPodSandbox(context.Background(), request)
+	log.Debugf("StopPodSandboxResponse: %v", r)
+	if err != nil {
+		return err
+	}
+
+	log.Printf("Stopped sandbox %s\n", id)
+	return nil
+}
+
+// Call RemovePodSandbox on the CRI client
+func removePodSandbox(client criapi.RuntimeServiceClient, id string) error {
+	if id == "" {
+		return fmt.Errorf("ID cannot be empty")
+	}
+	request := &criapi.RemovePodSandboxRequest{PodSandboxId: id}
+	log.Debugf("RemovePodSandboxRequest: %v", request)
+	r, err := client.RemovePodSandbox(context.Background(), request)
+	log.Debugf("RemovePodSandboxResponse: %v", r)
+	if err != nil {
+		return err
+	}
+	log.Printf("Removed sandbox %s\n", id)
+	return nil
+}
+
+// Call ListPodSandbox on the CRI client
+func getPodSandboxes(client criapi.RuntimeServiceClient) ([]*criapi.PodSandbox, error) {
+	filter := &criapi.PodSandboxFilter{}
+	request := &criapi.ListPodSandboxRequest{
+		Filter: filter,
+	}
+
+	log.Debugf("ListPodSandboxRequest: %v", request)
+	r, err := client.ListPodSandbox(context.Background(), request)
+
+	log.Debugf("ListPodSandboxResponse: %v", r)
+	if err != nil {
+		return nil, err
+	}
+	return r.GetItems(), err
+}
+
+// Call PodSandboxStatus on the CRI client
+func getPodSandboxStatus(client criapi.RuntimeServiceClient, psId string) (*criapi.PodSandboxStatus, error) {
+	if psId == "" {
+		return nil, fmt.Errorf("Pod ID cannot be empty in GPSS")
+	}
+
+	request := &criapi.PodSandboxStatusRequest{
+		PodSandboxId: psId,
+		Verbose:      false,
+	}
+
+	log.Debugf("PodSandboxStatusRequest: %v", request)
+	r, err := client.PodSandboxStatus(context.Background(), request)
+	log.Debugf("PodSandboxStatusResponse: %v", r)
+	if err != nil {
+		return nil, err
+	}
+
+	return r.Status, nil
+}
+
+// Call CreateContainer on the CRI client
+func createContainer(client criapi.RuntimeServiceClient, config *criapi.ContainerConfig, podConfig *criapi.PodSandboxConfig, pId string) (string, error) {
+	request := &criapi.CreateContainerRequest{
+		PodSandboxId:  pId,
+		Config:        config,
+		SandboxConfig: podConfig,
+	}
+	log.Debugf("CreateContainerRequest: %v", request)
+	r, err := client.CreateContainer(context.Background(), request)
+	log.Debugf("CreateContainerResponse: %v", r)
+	if err != nil {
+		return "", err
+	}
+	log.Printf("Container created: %s\n", r.ContainerId)
+	return r.ContainerId, nil
+}
+
+// Call StartContainer on the CRI client
+func startContainer(client criapi.RuntimeServiceClient, cId string) error {
+	if cId == "" {
+		return fmt.Errorf("ID cannot be empty")
+	}
+	request := &criapi.StartContainerRequest{
+		ContainerId: cId,
+	}
+	log.Debugf("StartContainerRequest: %v", request)
+	r, err := client.StartContainer(context.Background(), request)
+	log.Debugf("StartContainerResponse: %v", r)
+	if err != nil {
+		return err
+	}
+	log.Printf("Container started: %s\n", cId)
+	return nil
+}
+
+// Call ContainerStatus on the CRI client
+func getContainerCRIStatus(client criapi.RuntimeServiceClient, cId string) (*criapi.ContainerStatus, error) {
+	if cId == "" {
+		return nil, fmt.Errorf("Container ID cannot be empty in GCCS")
+	}
+
+	request := &criapi.ContainerStatusRequest{
+		ContainerId: cId,
+		Verbose:     false,
+	}
+	log.Debugf("ContainerStatusRequest: %v", request)
+	r, err := client.ContainerStatus(context.Background(), request)
+	log.Debugf("ContainerStatusResponse: %v", r)
+	if err != nil {
+		return nil, err
+	}
+
+	return r.Status, nil
+}
+
+// Call ListContainers on the CRI client
+func getContainersForSandbox(client criapi.RuntimeServiceClient, psId string) ([]*criapi.Container, error) {
+	filter := &criapi.ContainerFilter{}
+	filter.PodSandboxId = psId
+	request := &criapi.ListContainersRequest{
+		Filter: filter,
+	}
+	log.Debugf("ListContainerRequest: %v", request)
+	r, err := client.ListContainers(context.Background(), request)
+	log.Debugf("ListContainerResponse: %v", r)
+	if err != nil {
+		return nil, err
+	}
+	return r.Containers, nil
+}
+
+// Pull and image on the CRI client and return the image ref
+func pullImage(client criapi.ImageServiceClient, image string) (string, error) {
+	request := &criapi.PullImageRequest{
+		Image: &criapi.ImageSpec{
+			Image: image,
+		},
+	}
+	log.Debugf("PullImageRequest: %v", request)
+	r, err := client.PullImage(context.Background(), request)
+	log.Debugf("PullImageResponse: %v", r)
+	if err != nil {
+		return "", err
+	}
+
+	return r.ImageRef, nil
+}