status-im
/
go-libp2p-pubsub
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add validation context for cancelation on aborts

This commit is contained in:
vyzo 2018-01-13 18:34:00 +02:00
parent fba445bc6d
commit c95ed28496
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
floodsub.go
View File

@ -377,6 +377,9 @@ func (p *PubSub) pushMsg(subs []*Subscription, src peer.ID, msg *Message) {
// validate performs validation and only sends the message if all validators succeed
func (p *PubSub) validate(subs []*Subscription, src peer.ID, msg *Message) {
ctx, cancel := context.WithCancel(p.ctx)
defer cancel()
results := make([]chan bool, 0, len(subs))
throttle := false
@ -391,10 +394,10 @@ loop:
select {
case sub.validateThrottle <- struct{}{}:
go func(sub *Subscription, msg *Message, rch chan bool) {
rch <- sub.validateMsg(p.ctx, msg)
go func(sub *Subscription, rch chan bool) {
rch <- sub.validateMsg(ctx, msg)
<-sub.validateThrottle
}(sub, msg, rch)
}(sub, rch)
default:
log.Debugf("validation throttled for topic %s", sub.topic)