132 lines
2.9 KiB
Go
132 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/docker/docker/integration-cli/cli"
|
|
"gotest.tools/v3/assert"
|
|
)
|
|
|
|
type DockerBenchmarkSuite struct {
|
|
ds *DockerSuite
|
|
}
|
|
|
|
func (s *DockerBenchmarkSuite) TearDownTest(ctx context.Context, c *testing.T) {
|
|
s.ds.TearDownTest(ctx, c)
|
|
}
|
|
|
|
func (s *DockerBenchmarkSuite) OnTimeout(c *testing.T) {
|
|
s.ds.OnTimeout(c)
|
|
}
|
|
|
|
func (s *DockerBenchmarkSuite) BenchmarkConcurrentContainerActions(c *testing.B) {
|
|
maxConcurrency := runtime.GOMAXPROCS(0)
|
|
numIterations := c.N
|
|
outerGroup := &sync.WaitGroup{}
|
|
outerGroup.Add(maxConcurrency)
|
|
chErr := make(chan error, numIterations*2*maxConcurrency)
|
|
|
|
for i := 0; i < maxConcurrency; i++ {
|
|
go func() {
|
|
defer outerGroup.Done()
|
|
innerGroup := &sync.WaitGroup{}
|
|
innerGroup.Add(2)
|
|
|
|
go func() {
|
|
defer innerGroup.Done()
|
|
for i := 0; i < numIterations; i++ {
|
|
args := []string{"run", "-d", "busybox"}
|
|
args = append(args, sleepCommandForDaemonPlatform()...)
|
|
out, _, err := dockerCmdWithError(args...)
|
|
if err != nil {
|
|
chErr <- fmt.Errorf(out)
|
|
return
|
|
}
|
|
|
|
id := strings.TrimSpace(out)
|
|
tmpDir, err := os.MkdirTemp("", "docker-concurrent-test-"+id)
|
|
if err != nil {
|
|
chErr <- err
|
|
return
|
|
}
|
|
defer os.RemoveAll(tmpDir)
|
|
out, _, err = dockerCmdWithError("cp", id+":/tmp", tmpDir)
|
|
if err != nil {
|
|
chErr <- fmt.Errorf(out)
|
|
return
|
|
}
|
|
|
|
out, _, err = dockerCmdWithError("kill", id)
|
|
if err != nil {
|
|
chErr <- fmt.Errorf(out)
|
|
}
|
|
|
|
out, _, err = dockerCmdWithError("start", id)
|
|
if err != nil {
|
|
chErr <- fmt.Errorf(out)
|
|
}
|
|
|
|
out, _, err = dockerCmdWithError("kill", id)
|
|
if err != nil {
|
|
chErr <- fmt.Errorf(out)
|
|
}
|
|
|
|
// don't do an rm -f here since it can potentially ignore errors from the graphdriver
|
|
out, _, err = dockerCmdWithError("rm", id)
|
|
if err != nil {
|
|
chErr <- fmt.Errorf(out)
|
|
}
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
defer innerGroup.Done()
|
|
for i := 0; i < numIterations; i++ {
|
|
out, _, err := dockerCmdWithError("ps")
|
|
if err != nil {
|
|
chErr <- fmt.Errorf(out)
|
|
}
|
|
}
|
|
}()
|
|
|
|
innerGroup.Wait()
|
|
}()
|
|
}
|
|
|
|
outerGroup.Wait()
|
|
close(chErr)
|
|
|
|
for err := range chErr {
|
|
assert.NilError(c, err)
|
|
}
|
|
}
|
|
|
|
func (s *DockerBenchmarkSuite) BenchmarkLogsCLIRotateFollow(c *testing.B) {
|
|
out := cli.DockerCmd(c, "run", "-d", "--log-opt", "max-size=1b", "--log-opt", "max-file=10", "busybox", "sh", "-c", "while true; do usleep 50000; echo hello; done").Combined()
|
|
id := strings.TrimSpace(out)
|
|
ch := make(chan error, 1)
|
|
go func() {
|
|
ch <- nil
|
|
out, _, _ := dockerCmdWithError("logs", "-f", id)
|
|
// if this returns at all, it's an error
|
|
ch <- fmt.Errorf(out)
|
|
}()
|
|
|
|
<-ch
|
|
select {
|
|
case <-time.After(30 * time.Second):
|
|
// ran for 30 seconds with no problem
|
|
return
|
|
case err := <-ch:
|
|
if err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
}
|
|
}
|