stakey/vspd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use shutdownCtx for dcrdWithNotifs

Browse Source
This commit is contained in:
jholdstock 2020-09-21 11:17:31 +01:00 committed by David Hill
parent f06d4cd8ba
commit c82b08b760
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
background/background.go
View File

@ -279,7 +279,7 @@ func (n *NotificationHandler) Close() error {
func connectNotifier(shutdownCtx context.Context, dcrdWithNotifs rpc.DcrdConnect) error { func connectNotifier(shutdownCtx context.Context, dcrdWithNotifs rpc.DcrdConnect) error {
notifierClosed = make(chan struct{}) notifierClosed = make(chan struct{})
dcrdClient, err := dcrdWithNotifs.Client(context.Background(), netParams) dcrdClient, err := dcrdWithNotifs.Client(shutdownCtx, netParams)
if err != nil { if err != nil {
return err return err
} }
@ -295,6 +295,9 @@ func connectNotifier(shutdownCtx context.Context, dcrdWithNotifs rpc.DcrdConnect
// notifier is closed. // notifier is closed.
select { select {
case <-shutdownCtx.Done(): case <-shutdownCtx.Done():
// A shutdown signal has been received - close the client with the
// notification handler to prevent further notifications from being
// received.
dcrdWithNotifs.Close() dcrdWithNotifs.Close()
return nil return nil
case <-notifierClosed: case <-notifierClosed: