| 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) |
| } |
| } |
| } |