kotovalexarian-likes-github/moby--moby
1
0
Fork 0
mirror of https://github.com/moby/moby.git synced 2022-11-09 12:21:53 -05:00
Code Releases Activity

fix event removal

Browse source 
Docker-DCO-1.1-Signed-off-by: Victor Vieux <vieux@docker.com> (github: vieux)
This commit is contained in:
Victor Vieux 2014-05-09 18:48:55 +00:00
parent 0f6fe22833
commit 670564d070
1 changed files with 15 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
server/server.go
View file

@ -195,6 +195,15 @@ func (srv *Server) ContainerKill(job *engine.Job) engine.Status {
return engine.StatusOK return engine.StatusOK
} }
func (srv *Server) EvictListener(from string) {
srv.Lock()
if old, ok := srv.listeners[from]; ok {
delete(srv.listeners, from)
close(old)
}
srv.Unlock()
}
func (srv *Server) Events(job *engine.Job) engine.Status { func (srv *Server) Events(job *engine.Job) engine.Status {
if len(job.Args) != 1 { if len(job.Args) != 1 {
return job.Errorf("Usage: %s FROM", job.Name) return job.Errorf("Usage: %s FROM", job.Name)
@ -212,15 +221,7 @@ func (srv *Server) Events(job *engine.Job) engine.Status {
return fmt.Errorf("JSON error") return fmt.Errorf("JSON error")
} }
_, err = job.Stdout.Write(b) _, err = job.Stdout.Write(b)
if err != nil { return err
// On error, evict the listener
utils.Errorf("%s", err)
srv.Lock()
delete(srv.listeners, from)
srv.Unlock()
return err
}
return nil
} }
listener := make(chan utils.JSONMessage) listener := make(chan utils.JSONMessage)
@ -241,8 +242,9 @@ func (srv *Server) Events(job *engine.Job) engine.Status {
continue continue
} }
if err != nil { if err != nil {
job.Error(err) // On error, evict the listener
return engine.StatusErr srv.EvictListener(from)
return job.Error(err)
} }
} }
} }
@ -260,6 +262,8 @@ func (srv *Server) Events(job *engine.Job) engine.Status {
continue continue
} }
if err != nil { if err != nil {
// On error, evict the listener
srv.EvictListener(from)
return job.Error(err) return job.Error(err)
} }
case <-timeout.C: case <-timeout.C: