提交 1a37c0a8 作者: Steven Allen

avoid publishing if notification system has been shut down

(will deadlock)

License: MIT
Signed-off-by: 's avatarSteven Allen <steven@stebalien.com>
上级 2baa3312
...@@ -36,6 +36,16 @@ type impl struct { ...@@ -36,6 +36,16 @@ type impl struct {
} }
func (ps *impl) Publish(block blocks.Block) { func (ps *impl) Publish(block blocks.Block) {
ps.wg.Add(1)
defer ps.wg.Done()
select {
case <-ps.cancel:
// Already shutdown, bail.
return
default:
}
ps.wrapped.Pub(block, block.Cid().KeyString()) ps.wrapped.Pub(block, block.Cid().KeyString())
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论