-
Notifications
You must be signed in to change notification settings - Fork 0
/
docker.go
56 lines (47 loc) · 1.51 KB
/
docker.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
package docker
import (
"os"
"github.com/fsouza/go-dockerclient"
"github.com/golang/glog"
"github.com/spf13/pflag"
)
// Helper contains all the valid config options for connecting to Docker from
// a command line.
type Helper struct {
}
// NewHelper creates a Flags object with the default values set. Use this
// to use consistent Docker client loading behavior from different contexts.
func NewHelper() *Helper {
return &Helper{}
}
// InstallFlags installs the Docker flag helper into a FlagSet with the default
// options and default values from the Helper object.
func (_ *Helper) InstallFlags(flags *pflag.FlagSet) {
}
// GetClient returns a valid Docker client, the address of the client, or an error
// if the client couldn't be created.
func (_ *Helper) GetClient() (*docker.Client, string, error) {
addr := getDockerEndpoint("")
client, err := docker.NewClient(addr)
return client, addr, err
}
// GetClientOrExit returns a valid Docker client and the address of the client,
// or prints an error and exits.
func (h *Helper) GetClientOrExit() (*docker.Client, string) {
client, addr, err := h.GetClient()
if err != nil {
glog.Fatalf("ERROR: Couldn't connect to Docker at %s.\n%v\n.", addr, err)
}
return client, addr
}
func getDockerEndpoint(dockerEndpoint string) string {
var endpoint string
if len(dockerEndpoint) > 0 {
endpoint = dockerEndpoint
} else if len(os.Getenv("DOCKER_HOST")) > 0 {
endpoint = os.Getenv("DOCKER_HOST")
} else {
endpoint = "unix:///var/run/docker.sock"
}
return endpoint
}