mirror of
https://github.com/bingohuang/docker-labs.git
synced 2025-07-15 02:37:27 +08:00
dd8f63363e
Previously, if node2 was deleted in a node1, node2, node3 set, the next node creation would fail since it would always try to assign node3 as a name. Now, the first available name is always assigned. Signed-off-by: Antonis Kalipetis <akalipetis@gmail.com>
133 lines
3.2 KiB
Go
133 lines
3.2 KiB
Go
package services
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/api/types/network"
|
|
"github.com/docker/docker/client"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
var c *client.Client
|
|
|
|
const (
|
|
Byte = 1
|
|
Kilobyte = 1024 * Byte
|
|
Megabyte = 1024 * Kilobyte
|
|
)
|
|
|
|
func init() {
|
|
var err error
|
|
c, err = client.NewEnvClient()
|
|
if err != nil {
|
|
// this wont happen if daemon is offline, only for some critical errors
|
|
log.Fatal("Cannot initialize docker client")
|
|
}
|
|
|
|
}
|
|
|
|
func GetContainerStats(id string) (io.ReadCloser, error) {
|
|
stats, err := c.ContainerStats(context.Background(), id, true)
|
|
|
|
return stats.Body, err
|
|
}
|
|
|
|
func GetContainerInfo(id string) (types.ContainerJSON, error) {
|
|
return c.ContainerInspect(context.Background(), id)
|
|
}
|
|
|
|
func CreateNetwork(name string) error {
|
|
opts := types.NetworkCreate{Driver: "overlay", Attachable: true}
|
|
_, err := c.NetworkCreate(context.Background(), name, opts)
|
|
|
|
if err != nil {
|
|
log.Printf("Starting session err [%s]\n", err)
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func DeleteNetwork(id string) error {
|
|
err := c.NetworkRemove(context.Background(), id)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func CreateAttachConnection(id string, ctx context.Context) (*types.HijackedResponse, error) {
|
|
|
|
conf := types.ContainerAttachOptions{true, true, true, true, "ctrl-x,ctrl-x", true}
|
|
conn, err := c.ContainerAttach(ctx, id, conf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &conn, nil
|
|
}
|
|
|
|
func ResizeConnection(name string, cols, rows uint) error {
|
|
return c.ContainerResize(context.Background(), name, types.ResizeOptions{Height: rows, Width: cols})
|
|
}
|
|
|
|
func CreateInstance(session *Session, dindImage string) (*Instance, error) {
|
|
|
|
h := &container.HostConfig{NetworkMode: container.NetworkMode(session.Id), Privileged: true}
|
|
h.Resources.PidsLimit = int64(500)
|
|
h.Resources.Memory = 4092 * Megabyte
|
|
t := true
|
|
h.Resources.OomKillDisable = &t
|
|
|
|
var nodeName string
|
|
var containerName string
|
|
for i := 1; ; i++ {
|
|
nodeName = fmt.Sprintf("node%d", i)
|
|
containerName = fmt.Sprintf("%s_%s", session.Id[:8], nodeName)
|
|
exists := false
|
|
for _, instance := range session.Instances {
|
|
if instance.Name == containerName {
|
|
exists = true
|
|
break
|
|
}
|
|
}
|
|
if !exists {
|
|
break
|
|
}
|
|
}
|
|
conf := &container.Config{Hostname: nodeName, Image: dindImage, Tty: true, OpenStdin: true, AttachStdin: true, AttachStdout: true, AttachStderr: true}
|
|
networkConf := &network.NetworkingConfig{
|
|
map[string]*network.EndpointSettings{
|
|
session.Id: &network.EndpointSettings{Aliases: []string{nodeName}},
|
|
},
|
|
}
|
|
container, err := c.ContainerCreate(context.Background(), conf, h, networkConf, containerName)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = c.ContainerStart(context.Background(), container.ID, types.ContainerStartOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cinfo, err := GetContainerInfo(container.ID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Instance{Name: containerName, Hostname: cinfo.Config.Hostname, IP: cinfo.NetworkSettings.Networks[session.Id].IPAddress}, nil
|
|
}
|
|
|
|
func DeleteContainer(id string) error {
|
|
return c.ContainerRemove(context.Background(), id, types.ContainerRemoveOptions{Force: true, RemoveVolumes: true})
|
|
}
|