mirror of
https://github.com/bingohuang/docker-labs.git
synced 2025-07-14 10:17:26 +08:00
114 lines
2.7 KiB
Go
114 lines
2.7 KiB
Go
package services
|
|
|
|
import (
|
|
"log"
|
|
"strings"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/container"
|
|
"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 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 CreateExecConnection(id string, ctx context.Context) (string, error) {
|
|
conf := types.ExecConfig{Tty: true, AttachStdin: true, AttachStderr: true, AttachStdout: true, Cmd: []string{"sh"}}
|
|
resp, err := c.ContainerExecCreate(ctx, id, conf)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return resp.ID, nil
|
|
}
|
|
|
|
func AttachExecConnection(execId string, ctx context.Context) (*types.HijackedResponse, error) {
|
|
conf := types.ExecConfig{Tty: true, AttachStdin: true, AttachStderr: true, AttachStdout: true}
|
|
conn, err := c.ContainerExecAttach(ctx, execId, conf)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &conn, nil
|
|
}
|
|
|
|
func ResizeExecConnection(execId string, ctx context.Context, cols, rows uint) error {
|
|
return c.ContainerExecResize(ctx, execId, types.ResizeOptions{Height: rows, Width: cols})
|
|
}
|
|
|
|
func CreateInstance(net string, dindImage string) (*Instance, error) {
|
|
|
|
h := &container.HostConfig{NetworkMode: container.NetworkMode(net), Privileged: true}
|
|
h.Resources.PidsLimit = int64(150)
|
|
h.Resources.Memory = 512 * Megabyte
|
|
t := true
|
|
h.Resources.OomKillDisable = &t
|
|
|
|
conf := &container.Config{Image: dindImage, Tty: true}
|
|
container, err := c.ContainerCreate(context.Background(), conf, h, nil, "")
|
|
|
|
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: strings.Replace(cinfo.Name, "/", "", 1), IP: cinfo.NetworkSettings.Networks[net].IPAddress}, nil
|
|
}
|
|
|
|
func DeleteContainer(id string) error {
|
|
return c.ContainerRemove(context.Background(), id, types.ContainerRemoveOptions{Force: true, RemoveVolumes: true})
|
|
}
|