docker.go

package util

import (
	"os"

	dockerClient "github.com/fsouza/go-dockerclient"
	"github.com/golang/glog"
)

const defaultDockerEndpoint = "unix:///var/run/docker.sock"

// RequireDocker ensures that a new docker client can be created and that a ListImages command can be run on the client
// or it fails with glog.Fatal
func RequireDocker() {
	client, err := NewDockerClient()

	if err != nil {
		glog.Fatalf("unable to create docker client for testing: %v", err)
	}

	//simple test to make sure you can take action with the client
	_, err = client.ListImages(dockerClient.ListImagesOptions{All: false})

	if err != nil {
		glog.Fatalf("unable to create docker client for testing: %v", err)
	}
}

// newDockerClient creates a docker client using the env var DOCKER_ENDPOINT or, if not supplied, uses the default
// docker endpoint /var/run/docker.sock
func NewDockerClient() (*dockerClient.Client, error) {
	endpoint := os.Getenv("DOCKER_ENDPOINT")

	if len(endpoint) == 0 {
		endpoint = defaultDockerEndpoint
	}

	return dockerClient.NewClient(endpoint)
}