TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

notary: wait for the service to finish on Shutdown

Browse source
This commit is contained in:
Roman Khimov 2022-07-02 15:50:27 +03:00
parent 0d627c947f
commit bf462a81fe
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pkg/services/notary/notary.go
View file

@ -67,6 +67,7 @@ type (
reqCh chan mempoolevent.Event reqCh chan mempoolevent.Event
blocksCh chan *block.Block blocksCh chan *block.Block
stopCh chan struct{} stopCh chan struct{}
done chan struct{}
} }
// Config represents external configuration for Notary module. // Config represents external configuration for Notary module.
@ -153,6 +154,7 @@ func NewNotary(cfg Config, net netmode.Magic, mp *mempool.Pool, onTransaction fu
reqCh: make(chan mempoolevent.Event), reqCh: make(chan mempoolevent.Event),
blocksCh: make(chan *block.Block), blocksCh: make(chan *block.Block),
stopCh: make(chan struct{}), stopCh: make(chan struct{}),
done: make(chan struct{}),
}, nil }, nil
} }
@ -206,6 +208,7 @@ drainLoop:
} }
close(n.blocksCh) close(n.blocksCh)
close(n.reqCh) close(n.reqCh)
close(n.done)
} }
// Shutdown stops the Notary module. // Shutdown stops the Notary module.
@ -214,6 +217,7 @@ func (n *Notary) Shutdown() {
return return
} }
close(n.stopCh) close(n.stopCh)
<-n.done
} }
// OnNewRequest is a callback method which is called after a new notary request is added to the notary request pool. // OnNewRequest is a callback method which is called after a new notary request is added to the notary request pool.