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
}
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 {
if len(job.Args) != 1 {
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")
}
_, err = job.Stdout.Write(b)
if err != nil {
// On error, evict the listener
utils.Errorf("%s", err)
srv.Lock()
delete(srv.listeners, from)
srv.Unlock()
return err
}
return nil
return err
}
listener := make(chan utils.JSONMessage)
@ -241,8 +242,9 @@ func (srv *Server) Events(job *engine.Job) engine.Status {
continue
}
if err != nil {
job.Error(err)
return engine.StatusErr
// On error, evict the listener
srv.EvictListener(from)
return job.Error(err)
}
}
}
@ -260,6 +262,8 @@ func (srv *Server) Events(job *engine.Job) engine.Status {
continue
}
if err != nil {
// On error, evict the listener
srv.EvictListener(from)
return job.Error(err)
}
case <-timeout.C: