84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
package docker
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
dockertypes "github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/api/types/filters"
|
|
"github.com/docker/go-connections/nat"
|
|
"github.com/ehazlett/element/api/types"
|
|
"github.com/ehazlett/element/runtime"
|
|
"github.com/sirupsen/logrus"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func (d *Docker) Create(spec *types.RuntimeSpec) (runtime.Container, error) {
|
|
if spec.Labels == nil {
|
|
spec.Labels = map[string]string{}
|
|
}
|
|
labels := spec.Labels
|
|
// insert element labels
|
|
labels[elementRuntimeLabel] = "docker"
|
|
|
|
p := strconv.Itoa(int(spec.Port))
|
|
port, err := nat.NewPort(spec.Protocol, p)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
containerConfig := &container.Config{
|
|
Image: spec.Image,
|
|
Labels: spec.Labels,
|
|
ExposedPorts: nat.PortSet{
|
|
port: struct{}{},
|
|
},
|
|
}
|
|
|
|
hostConfig := &container.HostConfig{
|
|
PortBindings: nat.PortMap{
|
|
port: []nat.PortBinding{
|
|
{
|
|
HostIP: "0.0.0.0",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
logrus.Debugf("%+v", containerConfig)
|
|
|
|
// create
|
|
resp, err := d.client.ContainerCreate(context.Background(), containerConfig, hostConfig, nil, "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// start
|
|
if err := d.client.ContainerStart(context.Background(), resp.ID, dockertypes.ContainerStartOptions{}); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
optFilters := filters.NewArgs()
|
|
optFilters.Add("id", resp.ID)
|
|
optFilters.Add("label", elementRuntimeLabel)
|
|
|
|
containers, err := d.client.ContainerList(context.Background(), dockertypes.ContainerListOptions{
|
|
Filters: optFilters,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(containers) == 0 {
|
|
return nil, fmt.Errorf("error creating container: no container found after start")
|
|
}
|
|
|
|
container := containers[0]
|
|
endpoint := getContainerEndpoint(container)
|
|
|
|
return Container{
|
|
id: container.ID,
|
|
endpoint: endpoint,
|
|
}, nil
|
|
}
|