forked from goodrain/rainbond
/
container_client.go
115 lines (99 loc) · 3.31 KB
/
container_client.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package sources
import (
"context"
"fmt"
"time"
"github.com/docker/docker/api/types"
dockercli "github.com/docker/docker/client"
"github.com/sirupsen/logrus"
runtimeapi "k8s.io/cri-api/pkg/apis/runtime/v1"
)
const (
// ContainerRuntimeDocker docker runtime
ContainerRuntimeDocker = "docker"
// ContainerRuntimeContainerd containerd runtime
ContainerRuntimeContainerd = "containerd"
// DefaultDockerSock docker runtime endpoint
DefaultDockerSock = "/var/run/dockershim.sock"
// DefaultContainerdSock containerd runtime endpoint
DefaultContainerdSock = "/run/containerd/containerd.sock"
)
const (
// CONTAINER_ACTION_START is start container event action
CONTAINER_ACTION_START = "start"
// CONTAINER_ACTION_STOP is stop container event action
CONTAINER_ACTION_STOP = "stop"
// CONTAINER_ACTION_CREATE is create container event action
CONTAINER_ACTION_CREATE = "create"
// CONTAINER_ACTION_DESTROY is destroy container event action
CONTAINER_ACTION_DESTROY = "destroy"
// CONTAINER_ACTION_DIE is die container event action
CONTAINER_ACTION_DIE = "die"
)
type ContainerDesc struct {
ContainerRuntime string
// Info is extra information of the Container. The key could be arbitrary string, and
// value should be in json format. The information could include anything useful for
// debug, e.g. pid for linux container based container runtime.
// It should only be returned non-empty when Verbose is true.
Info map[string]string
*runtimeapi.ContainerStatus
// Docker container json
*types.ContainerJSON
}
func (c *ContainerDesc) GetLogPath() string {
if c.ContainerRuntime == ContainerRuntimeDocker {
return c.ContainerJSON.LogPath
}
return c.ContainerStatus.GetLogPath()
}
func (c *ContainerDesc) GetId() string {
if c.ContainerRuntime == ContainerRuntimeDocker {
return c.ContainerJSON.ID
}
return c.ContainerStatus.GetId()
}
// ContainerImageCli container image client
type ContainerImageCli interface {
ListContainers() ([]*runtimeapi.Container, error)
InspectContainer(containerID string) (*ContainerDesc, error)
WatchContainers(ctx context.Context, cchan chan ContainerEvent) error
GetRuntimeClient() (*runtimeapi.RuntimeServiceClient, error)
GetDockerClient() (*dockercli.Client, error)
}
// ClientFactory client factory
type ClientFactory interface {
NewClient(endpoint string, timeout time.Duration) (ContainerImageCli, error)
}
// NewContainerImageClient new container image client
func NewContainerImageClient(containerRuntime, endpoint string, timeout time.Duration) (c ContainerImageCli, err error) {
logrus.Infof("create container client runtime %s endpoint %s", containerRuntime, endpoint)
switch containerRuntime {
case ContainerRuntimeDocker:
factory := &dockerClientFactory{}
c, err = factory.NewClient(
endpoint, timeout,
)
case ContainerRuntimeContainerd:
factory := &containerdClientFactory{}
c, err = factory.NewClient(
endpoint, timeout,
)
return
default:
err = fmt.Errorf("unknown runtime %s", containerRuntime)
return
}
return
}
// ContainerEvent container event
type ContainerEvent struct {
Action string
Container *ContainerDesc
}
func CacheContainer(cchan chan ContainerEvent, cs ...ContainerEvent) {
for _, container := range cs {
logrus.Debugf("found a container %s %s", container.Container.GetMetadata().GetName(), container.Action)
cchan <- container
}
}