Drain callback channel when try to deregister it

callback() is blocked on sending into channel otherwise.
This commit is contained in:
Yuriy Taraday 2018-02-05 01:28:14 +04:00
parent 0c23de2f7e
commit 07dbf8603a

16
rpc.go
View File

@ -191,10 +191,10 @@ func (l *Libvirt) listen() {
func (l *Libvirt) callback(id uint32, res response) { func (l *Libvirt) callback(id uint32, res response) {
l.cm.Lock() l.cm.Lock()
c, ok := l.callbacks[id] c, ok := l.callbacks[id]
l.cm.Unlock()
if ok { if ok {
c <- res c <- res
} }
l.cm.Unlock()
} }
// route sends incoming packets to their listeners. // route sends incoming packets to their listeners.
@ -281,7 +281,19 @@ func (l *Libvirt) register(id uint32, c chan response) {
// deregister destroys a method response callback // deregister destroys a method response callback
func (l *Libvirt) deregister(id uint32) { func (l *Libvirt) deregister(id uint32) {
l.cm.Lock() lockChan := make(chan bool)
go func() {
l.cm.Lock()
lockChan <- true
}()
Loop:
for {
select {
case _ = <-l.callbacks[id]:
case _ = <-lockChan:
break Loop
}
}
close(l.callbacks[id]) close(l.callbacks[id])
delete(l.callbacks, id) delete(l.callbacks, id)
l.cm.Unlock() l.cm.Unlock()