// StartShell runs a command for a given service
func (a *api) StartShell(config ShellConfig) error {
mounts, err := buildMounts(config.ServicedEndpoint, config.ServiceID, config.Mounts)
if err != nil {
return err
}
command := append([]string{config.Command}, config.Args...)
cfg := shell.ProcessConfig{
ServiceID: config.ServiceID,
IsTTY: config.IsTTY,
SaveAs: config.SaveAs,
Mount: mounts,
Command: utils.ShellQuoteArgs(command),
}
// TODO: change me to use sockets
cmd, err := shell.StartDocker(&cfg, options.Endpoint)
if err != nil {
return fmt.Errorf("failed to connect to service: %s", err)
}
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
return cmd.Run()
}
// RunShell runs a predefined service shell command via the service definition
func (a *api) RunShell(config ShellConfig) error {
client, err := a.connectDAO()
if err != nil {
return err
}
svc, err := a.GetService(config.ServiceID)
if err != nil {
return err
}
getSvc := func(svcID string) (service.Service, error) {
s := service.Service{}
err := client.GetService(svcID, &s)
return s, err
}
findChild := func(svcID, childName string) (service.Service, error) {
s := service.Service{}
err := client.FindChildService(dao.FindChildRequest{svcID, childName}, &s)
return s, err
}
if err := svc.EvaluateRunsTemplate(getSvc, findChild); err != nil {
fmt.Errorf("error evaluating service:%s Runs:%+v error:%s", svc.ID, svc.Runs, err)
}
command, ok := svc.Runs[config.Command]
if !ok {
return fmt.Errorf("command not found for service")
}
mounts, err := buildMounts(config.ServicedEndpoint, config.ServiceID, config.Mounts)
if err != nil {
return err
}
quotedArgs := utils.ShellQuoteArgs(config.Args)
command = strings.Join([]string{command, quotedArgs}, " ")
asUser := "******"
if config.Username != "" && config.Username != "root" {
asUser = fmt.Sprintf("su - %s -c ", config.Username)
}
cfg := shell.ProcessConfig{
ServiceID: config.ServiceID,
IsTTY: config.IsTTY,
SaveAs: config.SaveAs,
Mount: mounts,
Command: asUser + utils.ShellQuoteArg(command),
LogToStderr: config.LogToStderr,
}
cfg.LogStash.Enable = config.LogStash.Enable
cfg.LogStash.SettleTime, err = time.ParseDuration(config.LogStash.SettleTime)
if err != nil {
return err
}
cfg.LogStash.IdleFlushTime, err = time.ParseDuration(config.LogStash.IdleFlushTime)
if err != nil {
return err
}
// TODO: change me to use sockets
cmd, err := shell.StartDocker(&cfg, options.Endpoint)
if err != nil {
return fmt.Errorf("failed to connect to service: %s", err)
}
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
err = cmd.Run()
if _, ok := utils.GetExitStatus(err); !ok {
glog.Fatalf("abnormal termination from shell command: %s", err)
}
dockercli, err := a.connectDocker()
if err != nil {
glog.Fatalf("unable to connect to the docker service: %s", err)
}
exitcode, err := dockercli.WaitContainer(config.SaveAs)
if err != nil {
glog.Fatalf("failure waiting for container: %s", err)
}
container, err := dockercli.InspectContainer(config.SaveAs)
if err != nil {
glog.Fatalf("cannot acquire information about container: %s (%s)", config.SaveAs, err)
}
glog.V(2).Infof("Container ID: %s", container.ID)
switch exitcode {
case 0:
// Commit the container
label := ""
glog.V(0).Infof("Committing container")
if err := client.Commit(container.ID, &label); err != nil {
glog.Fatalf("Error committing container: %s (%s)", container.ID, err)
}
var layers = 0
if err := client.ImageLayerCount(container.Image, &layers); err != nil {
glog.Errorf("Counting layers for image %s", svc.ImageID)
}
if layers > layer.WARN_LAYER_COUNT {
glog.Warningf("Image '%s' number of layers (%d) approaching maximum (%d). Please squash image layers.",
svc.ImageID, layers, layer.MAX_LAYER_COUNT)
}
default:
// Delete the container
if err := dockercli.StopContainer(container.ID, 10); err != nil {
glog.Fatalf("failed to stop container: %s (%s)", container.ID, err)
} else if err := dockercli.RemoveContainer(dockerclient.RemoveContainerOptions{ID: container.ID}); err != nil {
glog.Fatalf("failed to remove container: %s (%s)", container.ID, err)
}
return fmt.Errorf("Command returned non-zero exit code %d. Container not commited.", exitcode)
}
return nil
}