exit collect when we get EOF

Signed-off-by: Xianglin Gao <xlgao@zju.edu.cn>
Upstream-commit: e03b55a2f5ce2c55699f0181350534a9e2034461
Component: engine

components/engine/cli/command/container/stats_helpers.go

@@ -155,11 +155,13 @@ func collect(ctx context.Context, s *formatter.ContainerStats, cli client.APICli
 				waitFirst.Done()
 			}
 		case err := <-u:
+			s.SetError(err)
+			if err == io.EOF {
+				break
+			}
 			if err != nil {
-				s.SetError(err)
 				continue
 			}
-			s.SetError(nil)
 			// if this is the first stat you get, release WaitGroup
 			if !getFirst {
 				getFirst = true