opensource/podman
1
0
Fork 0
mirror of https://github.com/containers/podman.git synced 2025-06-24 19:42:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

stats: detect closed client connection

Browse Source 
Detect closed client connections and stop streaming.

Fixes: #7521
Signed-off-by: Valentin Rothberg <rothberg@redhat.com>
This commit is contained in:
Valentin Rothberg
2020-09-18 13:56:31 +02:00
parent c6410076b0
commit ae0e4dfd75
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pkg/api/handlers/compat/containers_stats.go
View File

@ -86,7 +86,12 @@ func StatsContainer(w http.ResponseWriter, r *http.Request) {
coder := json.NewEncoder(w)
coder.SetEscapeHTML(true)
for ok := true; ok; ok = query.Stream {
streamLabel: // A label to flatten the scope
select {
case <-r.Context().Done():
logrus.Debugf("Client connection (container stats) cancelled")
default:
// Container stats
stats, err := ctnr.GetContainerStats(stats)
if err != nil {
@ -194,6 +199,10 @@ func StatsContainer(w http.ResponseWriter, r *http.Request) {
flusher.Flush()
}
if !query.Stream {
return
}
preRead = s.Read
bits, err := json.Marshal(s.CPUStats)
if err != nil {
@ -203,10 +212,8 @@ func StatsContainer(w http.ResponseWriter, r *http.Request) {
logrus.Errorf("Unable to unmarshal previous stats: %q", err)
}
// Only sleep when we're streaming.
if query.Stream {
time.Sleep(DefaultStatsPeriod)
}
time.Sleep(DefaultStatsPeriod)
goto streamLabel
}
}