builder/docker: lock around Login

This commit is contained in:
Mitchell Hashimoto 2014-09-05 15:06:07 -07:00
parent e9611dfb11
commit 622b9f459a
2 changed files with 19 additions and 5 deletions

View File

@ -20,10 +20,11 @@ type Driver interface {
// Import imports a container from a tar file
Import(path, repo string) (string, error)
// Login
// Login. This will lock the driver from performing another Login
// until Logout is called. Therefore, any users MUST call Logout.
Login(repo, email, username, password string) error
// Logout
// Logout. This can only be called if Login succeeded.
Logout(repo string) error
// Pull should pull down the given image.

View File

@ -3,17 +3,21 @@ package docker
import (
"bytes"
"fmt"
"github.com/mitchellh/packer/packer"
"io"
"log"
"os"
"os/exec"
"strings"
"sync"
"github.com/mitchellh/packer/packer"
)
type DockerDriver struct {
Ui packer.Ui
Tpl *packer.ConfigTemplate
l sync.Mutex
}
func (d *DockerDriver) DeleteImage(id string) error {
@ -110,6 +114,8 @@ func (d *DockerDriver) Import(path string, repo string) (string, error) {
}
func (d *DockerDriver) Login(repo, email, user, pass string) error {
d.l.Lock()
args := []string{"login"}
if email != "" {
args = append(args, "-e", email)
@ -125,7 +131,12 @@ func (d *DockerDriver) Login(repo, email, user, pass string) error {
}
cmd := exec.Command("docker", args...)
return runAndStream(cmd, d.Ui)
err := runAndStream(cmd, d.Ui)
if err != nil {
d.l.Unlock()
}
return err
}
func (d *DockerDriver) Logout(repo string) error {
@ -135,7 +146,9 @@ func (d *DockerDriver) Logout(repo string) error {
}
cmd := exec.Command("docker", args...)
return runAndStream(cmd, d.Ui)
err := runAndStream(cmd, d.Ui)
d.l.Unlock()
return err
}
func (d *DockerDriver) Pull(image string) error {