| package main |
| |
| import ( |
| "encoding/json" |
| "fmt" |
| "net/http" |
| "os/exec" |
| "runtime" |
| "strconv" |
| "strings" |
| "time" |
| |
| "github.com/docker/docker/pkg/integration/checker" |
| "github.com/docker/engine-api/types" |
| "github.com/docker/engine-api/types/versions" |
| "github.com/go-check/check" |
| ) |
| |
| var expectedNetworkInterfaceStats = strings.Split("rx_bytes rx_dropped rx_errors rx_packets tx_bytes tx_dropped tx_errors tx_packets", " ") |
| |
| func (s *DockerSuite) TestApiStatsNoStreamGetCpu(c *check.C) { |
| testRequires(c, DaemonIsLinux) |
| out, _ := dockerCmd(c, "run", "-d", "busybox", "/bin/sh", "-c", "while true;do echo 'Hello'; usleep 100000; done") |
| |
| id := strings.TrimSpace(out) |
| c.Assert(waitRun(id), checker.IsNil) |
| |
| resp, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/stats?stream=false", id), nil, "") |
| c.Assert(err, checker.IsNil) |
| c.Assert(resp.StatusCode, checker.Equals, http.StatusOK) |
| c.Assert(resp.Header.Get("Content-Type"), checker.Equals, "application/json") |
| |
| var v *types.Stats |
| err = json.NewDecoder(body).Decode(&v) |
| c.Assert(err, checker.IsNil) |
| body.Close() |
| |
| var cpuPercent = 0.0 |
| cpuDelta := float64(v.CPUStats.CPUUsage.TotalUsage - v.PreCPUStats.CPUUsage.TotalUsage) |
| systemDelta := float64(v.CPUStats.SystemUsage - v.PreCPUStats.SystemUsage) |
| cpuPercent = (cpuDelta / systemDelta) * float64(len(v.CPUStats.CPUUsage.PercpuUsage)) * 100.0 |
| |
| c.Assert(cpuPercent, check.Not(checker.Equals), 0.0, check.Commentf("docker stats with no-stream get cpu usage failed: was %v", cpuPercent)) |
| } |
| |
| func (s *DockerSuite) TestApiStatsStoppedContainerInGoroutines(c *check.C) { |
| testRequires(c, DaemonIsLinux) |
| out, _ := dockerCmd(c, "run", "-d", "busybox", "/bin/sh", "-c", "echo 1") |
| id := strings.TrimSpace(out) |
| |
| getGoRoutines := func() int { |
| _, body, err := sockRequestRaw("GET", fmt.Sprintf("/info"), nil, "") |
| c.Assert(err, checker.IsNil) |
| info := types.Info{} |
| err = json.NewDecoder(body).Decode(&info) |
| c.Assert(err, checker.IsNil) |
| body.Close() |
| return info.NGoroutines |
| } |
| |
| // When the HTTP connection is closed, the number of goroutines should not increase. |
| routines := getGoRoutines() |
| _, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/stats", id), nil, "") |
| c.Assert(err, checker.IsNil) |
| body.Close() |
| |
| t := time.After(30 * time.Second) |
| for { |
| select { |
| case <-t: |
| c.Assert(getGoRoutines(), checker.LessOrEqualThan, routines) |
| return |
| default: |
| if n := getGoRoutines(); n <= routines { |
| return |
| } |
| time.Sleep(200 * time.Millisecond) |
| } |
| } |
| } |
| |
| func (s *DockerSuite) TestApiStatsNetworkStats(c *check.C) { |
| testRequires(c, SameHostDaemon) |
| testRequires(c, DaemonIsLinux) |
| |
| out, _ := runSleepingContainer(c) |
| id := strings.TrimSpace(out) |
| c.Assert(waitRun(id), checker.IsNil) |
| |
| // Retrieve the container address |
| contIP := findContainerIP(c, id, "bridge") |
| numPings := 4 |
| |
| var preRxPackets uint64 |
| var preTxPackets uint64 |
| var postRxPackets uint64 |
| var postTxPackets uint64 |
| |
| // Get the container networking stats before and after pinging the container |
| nwStatsPre := getNetworkStats(c, id) |
| for _, v := range nwStatsPre { |
| preRxPackets += v.RxPackets |
| preTxPackets += v.TxPackets |
| } |
| |
| countParam := "-c" |
| if runtime.GOOS == "windows" { |
| countParam = "-n" // Ping count parameter is -n on Windows |
| } |
| pingout, err := exec.Command("ping", contIP, countParam, strconv.Itoa(numPings)).CombinedOutput() |
| if err != nil && runtime.GOOS == "linux" { |
| // If it fails then try a work-around, but just for linux. |
| // If this fails too then go back to the old error for reporting. |
| // |
| // The ping will sometimes fail due to an apparmor issue where it |
| // denies access to the libc.so.6 shared library - running it |
| // via /lib64/ld-linux-x86-64.so.2 seems to work around it. |
| pingout2, err2 := exec.Command("/lib64/ld-linux-x86-64.so.2", "/bin/ping", contIP, "-c", strconv.Itoa(numPings)).CombinedOutput() |
| if err2 == nil { |
| pingout = pingout2 |
| err = err2 |
| } |
| } |
| c.Assert(err, checker.IsNil) |
| pingouts := string(pingout[:]) |
| nwStatsPost := getNetworkStats(c, id) |
| for _, v := range nwStatsPost { |
| postRxPackets += v.RxPackets |
| postTxPackets += v.TxPackets |
| } |
| |
| // Verify the stats contain at least the expected number of packets (account for ARP) |
| expRxPkts := 1 + preRxPackets + uint64(numPings) |
| expTxPkts := 1 + preTxPackets + uint64(numPings) |
| c.Assert(postTxPackets, checker.GreaterOrEqualThan, expTxPkts, |
| check.Commentf("Reported less TxPackets than expected. Expected >= %d. Found %d. %s", expTxPkts, postTxPackets, pingouts)) |
| c.Assert(postRxPackets, checker.GreaterOrEqualThan, expRxPkts, |
| check.Commentf("Reported less Txbytes than expected. Expected >= %d. Found %d. %s", expRxPkts, postRxPackets, pingouts)) |
| } |
| |
| func (s *DockerSuite) TestApiStatsNetworkStatsVersioning(c *check.C) { |
| testRequires(c, SameHostDaemon) |
| testRequires(c, DaemonIsLinux) |
| |
| out, _ := runSleepingContainer(c) |
| id := strings.TrimSpace(out) |
| c.Assert(waitRun(id), checker.IsNil) |
| |
| for i := 17; i <= 21; i++ { |
| apiVersion := fmt.Sprintf("v1.%d", i) |
| statsJSONBlob := getVersionedStats(c, id, apiVersion) |
| if versions.LessThan(apiVersion, "v1.21") { |
| c.Assert(jsonBlobHasLTv121NetworkStats(statsJSONBlob), checker.Equals, true, |
| check.Commentf("Stats JSON blob from API %s %#v does not look like a <v1.21 API stats structure", apiVersion, statsJSONBlob)) |
| } else { |
| c.Assert(jsonBlobHasGTE121NetworkStats(statsJSONBlob), checker.Equals, true, |
| check.Commentf("Stats JSON blob from API %s %#v does not look like a >=v1.21 API stats structure", apiVersion, statsJSONBlob)) |
| } |
| } |
| } |
| |
| func getNetworkStats(c *check.C, id string) map[string]types.NetworkStats { |
| var st *types.StatsJSON |
| |
| _, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/stats?stream=false", id), nil, "") |
| c.Assert(err, checker.IsNil) |
| |
| err = json.NewDecoder(body).Decode(&st) |
| c.Assert(err, checker.IsNil) |
| body.Close() |
| |
| return st.Networks |
| } |
| |
| // getVersionedStats returns stats result for the |
| // container with id using an API call with version apiVersion. Since the |
| // stats result type differs between API versions, we simply return |
| // map[string]interface{}. |
| func getVersionedStats(c *check.C, id string, apiVersion string) map[string]interface{} { |
| stats := make(map[string]interface{}) |
| |
| _, body, err := sockRequestRaw("GET", fmt.Sprintf("/%s/containers/%s/stats?stream=false", apiVersion, id), nil, "") |
| c.Assert(err, checker.IsNil) |
| defer body.Close() |
| |
| err = json.NewDecoder(body).Decode(&stats) |
| c.Assert(err, checker.IsNil, check.Commentf("failed to decode stat: %s", err)) |
| |
| return stats |
| } |
| |
| func jsonBlobHasLTv121NetworkStats(blob map[string]interface{}) bool { |
| networkStatsIntfc, ok := blob["network"] |
| if !ok { |
| return false |
| } |
| networkStats, ok := networkStatsIntfc.(map[string]interface{}) |
| if !ok { |
| return false |
| } |
| for _, expectedKey := range expectedNetworkInterfaceStats { |
| if _, ok := networkStats[expectedKey]; !ok { |
| return false |
| } |
| } |
| return true |
| } |
| |
| func jsonBlobHasGTE121NetworkStats(blob map[string]interface{}) bool { |
| networksStatsIntfc, ok := blob["networks"] |
| if !ok { |
| return false |
| } |
| networksStats, ok := networksStatsIntfc.(map[string]interface{}) |
| if !ok { |
| return false |
| } |
| for _, networkInterfaceStatsIntfc := range networksStats { |
| networkInterfaceStats, ok := networkInterfaceStatsIntfc.(map[string]interface{}) |
| if !ok { |
| return false |
| } |
| for _, expectedKey := range expectedNetworkInterfaceStats { |
| if _, ok := networkInterfaceStats[expectedKey]; !ok { |
| return false |
| } |
| } |
| } |
| return true |
| } |
| |
| func (s *DockerSuite) TestApiStatsContainerNotFound(c *check.C) { |
| testRequires(c, DaemonIsLinux) |
| |
| status, _, err := sockRequest("GET", "/containers/nonexistent/stats", nil) |
| c.Assert(err, checker.IsNil) |
| c.Assert(status, checker.Equals, http.StatusNotFound) |
| |
| status, _, err = sockRequest("GET", "/containers/nonexistent/stats?stream=0", nil) |
| c.Assert(err, checker.IsNil) |
| c.Assert(status, checker.Equals, http.StatusNotFound) |
| } |
| |
| func (s *DockerSuite) TestApiStatsNoStreamConnectedContainers(c *check.C) { |
| testRequires(c, DaemonIsLinux) |
| |
| out1, _ := runSleepingContainer(c) |
| id1 := strings.TrimSpace(out1) |
| c.Assert(waitRun(id1), checker.IsNil) |
| |
| out2, _ := runSleepingContainer(c, "--net", "container:"+id1) |
| id2 := strings.TrimSpace(out2) |
| c.Assert(waitRun(id2), checker.IsNil) |
| |
| ch := make(chan error) |
| go func() { |
| resp, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/stats?stream=false", id2), nil, "") |
| defer body.Close() |
| if err != nil { |
| ch <- err |
| } |
| if resp.StatusCode != http.StatusOK { |
| ch <- fmt.Errorf("Invalid StatusCode %v", resp.StatusCode) |
| } |
| if resp.Header.Get("Content-Type") != "application/json" { |
| ch <- fmt.Errorf("Invalid 'Content-Type' %v", resp.Header.Get("Content-Type")) |
| } |
| var v *types.Stats |
| if err := json.NewDecoder(body).Decode(&v); err != nil { |
| ch <- err |
| } |
| ch <- nil |
| }() |
| |
| select { |
| case err := <-ch: |
| c.Assert(err, checker.IsNil, check.Commentf("Error in stats remote API: %v", err)) |
| case <-time.After(15 * time.Second): |
| c.Fatalf("Stats did not return after timeout") |
| } |
| } |
