141 lines
3.0 KiB
Go
141 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/docker/containerkit"
|
|
"github.com/docker/containerkit/osutils"
|
|
specs "github.com/opencontainers/runtime-spec/specs-go"
|
|
)
|
|
|
|
func reloadContainer() error {
|
|
dockerContainer := &testConfig{}
|
|
container, err := containerkit.LoadContainer(dockerContainer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// wait for it to exit and get the exit status
|
|
logrus.Info("wait container")
|
|
status, err := container.Wait()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// delete the container after it is done
|
|
logrus.Info("delete container")
|
|
if container.Delete(); err != nil {
|
|
return err
|
|
}
|
|
logrus.Infof("exit status %d", status)
|
|
return nil
|
|
}
|
|
|
|
func runContainer() error {
|
|
// create a new runtime runtime that implements the ExecutionDriver interface
|
|
dockerContainer := &testConfig{}
|
|
|
|
// create a new container
|
|
container, err := containerkit.NewContainer(dockerContainer)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// setup some stdio for our container
|
|
container.Stdin = Stdin("")
|
|
container.Stdout = Stdout("")
|
|
container.Stderr = Stderr("")
|
|
|
|
// go ahead and set the container in the create state and have it ready to start
|
|
logrus.Info("create container")
|
|
if err := container.Create(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// start the user defined process in the container
|
|
logrus.Info("start container")
|
|
if err := container.Start(); err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := 0; i < exec; i++ {
|
|
process, err := container.NewProcess(&specs.Process{
|
|
Args: []string{
|
|
"sh", "-c",
|
|
"echo " + fmt.Sprintf("sup from itteration %d", i),
|
|
},
|
|
Env: env,
|
|
Terminal: false,
|
|
Cwd: "/",
|
|
NoNewPrivileges: true,
|
|
Capabilities: caps,
|
|
})
|
|
|
|
process.Stdin = Stdin(strconv.Itoa(i))
|
|
stdout := Stdout(strconv.Itoa(i))
|
|
|
|
stderr := Stderr(strconv.Itoa(i))
|
|
go io.Copy(os.Stdout, stdout)
|
|
go io.Copy(os.Stdout, stderr)
|
|
process.Stdout = stdout
|
|
process.Stderr = stderr
|
|
|
|
if err := process.Start(); err != nil {
|
|
return err
|
|
}
|
|
procStatus, err := process.Wait()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
logrus.Infof("process %d returned with %d", i, procStatus)
|
|
}
|
|
|
|
if load {
|
|
return nil
|
|
}
|
|
|
|
// wait for it to exit and get the exit status
|
|
logrus.Info("wait container")
|
|
status, err := container.Wait()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// delete the container after it is done
|
|
logrus.Info("delete container")
|
|
if container.Delete(); err != nil {
|
|
return err
|
|
}
|
|
logrus.Infof("exit status %d", status)
|
|
return nil
|
|
}
|
|
|
|
var (
|
|
exec int
|
|
load bool
|
|
reload bool
|
|
)
|
|
|
|
// "Hooks do optional work. Drivers do mandatory work"
|
|
func main() {
|
|
flag.IntVar(&exec, "exec", 0, "run n number of execs")
|
|
flag.BoolVar(&load, "load", false, "reload the container")
|
|
flag.BoolVar(&reload, "reload", false, "reload the container live")
|
|
flag.Parse()
|
|
if err := osutils.SetSubreaper(1); err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
if reload {
|
|
if err := reloadContainer(); err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
return
|
|
}
|
|
if err := runContainer(); err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
}
|