/
containers.go
83 lines (68 loc) · 1.8 KB
/
containers.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
package containers
import (
"context"
"errors"
"log"
"strings"
"github.com/docker/docker/api/types"
"github.com/docker/docker/api/types/filters"
"github.com/docker/docker/client"
"github.com/hibare/go-container-status/internal/utils"
)
var (
ErrUnhealthyContainers = "unhealthy containers found"
ErrNoContainersFound = "no containers found"
)
type Container struct {
Name []string
State string
Status string
Image string
}
func ContainerStatus(containerName string) ([]Container, error) {
ctx := context.Background()
foundContainers := []Container{}
containerFavorableConditions := []string{"running", "healthy"}
containerName = strings.Replace(containerName, "\n", "", -1)
containerName = strings.Replace(containerName, "\r", "", -1)
log.Printf("Checking status for container %s", containerName)
cli, err := client.NewClientWithOpts()
if err != nil {
log.Fatal(err)
return foundContainers, err
}
defer cli.Close()
cli.NegotiateAPIVersion(ctx)
options := types.ContainerListOptions{
All: true,
Filters: filters.NewArgs(
filters.Arg("name", containerName),
),
}
containers, err := cli.ContainerList(ctx, options)
if err != nil {
log.Fatal(err)
return foundContainers, err
}
unhealthyContainers := false
if len(containers) > 0 {
for _, container := range containers {
foundContainers = append(foundContainers, Container{
Name: container.Names,
State: container.State,
Status: container.Status,
Image: container.Image,
})
if !utils.SliceContains(container.State, containerFavorableConditions) {
unhealthyContainers = true
}
}
if unhealthyContainers {
return foundContainers, errors.New(ErrUnhealthyContainers)
} else {
return foundContainers, nil
}
} else {
return foundContainers, errors.New(ErrNoContainersFound)
}
}