You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
781 B
46 lines
781 B
3 years ago
|
package rpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"go.uber.org/zap"
|
||
|
|
)
|
||
|
|
|
||
|
|
// NotifyAcks notifies engine about received acknowledgements.
|
||
|
|
func (e *Engine) NotifyAcks(ids []int64) {
|
||
|
|
e.mux.Lock()
|
||
|
|
defer e.mux.Unlock()
|
||
|
|
|
||
|
|
for _, id := range ids {
|
||
|
|
ch, ok := e.ack[id]
|
||
|
|
if !ok {
|
||
|
|
e.log.Debug("Acknowledge callback not set", zap.Int64("msg_id", id))
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
close(ch)
|
||
|
|
delete(e.ack, id)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Engine) waitAck(id int64) chan struct{} {
|
||
|
|
e.mux.Lock()
|
||
|
|
defer e.mux.Unlock()
|
||
|
|
|
||
|
|
log := e.log.With(zap.Int64("ack_id", id))
|
||
|
|
if c, found := e.ack[id]; found {
|
||
|
|
log.Warn("Ack already registered")
|
||
|
|
return c
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Debug("Waiting for acknowledge")
|
||
|
|
c := make(chan struct{})
|
||
|
|
e.ack[id] = c
|
||
|
|
return c
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *Engine) removeAck(id int64) {
|
||
|
|
e.mux.Lock()
|
||
|
|
defer e.mux.Unlock()
|
||
|
|
|
||
|
|
delete(e.ack, id)
|
||
|
|
}
|