Mirroristas/kaspad
2
0
Fork 0
mirror of https://github.com/kaspanet/kaspad.git synced 2025-11-24 06:25:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

DownloadHeaders: Instead of using doneChan - close blockHeadersMessageChan. (#1523)

Browse Source
This commit is contained in:
Svarog 2021-02-11 16:59:31 +02:00 committed by GitHub
parent 05d1671f3a
commit 6db296b98b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
app/protocol/flows/blockrelay/ibd.go
View File

@ -221,7 +221,6 @@ func (flow *handleRelayInvsFlow) downloadHeaders(highestSharedBlockHash *externa
// headers // headers
blockHeadersMessageChan := make(chan *appmessage.BlockHeadersMessage, 2) blockHeadersMessageChan := make(chan *appmessage.BlockHeadersMessage, 2)
errChan := make(chan error) errChan := make(chan error)
doneChan := make(chan interface{})
spawn("handleRelayInvsFlow-downloadHeaders", func() { spawn("handleRelayInvsFlow-downloadHeaders", func() {
for { for {
blockHeadersMessage, doneIBD, err := flow.receiveHeaders() blockHeadersMessage, doneIBD, err := flow.receiveHeaders()
@ -230,7 +229,7 @@ func (flow *handleRelayInvsFlow) downloadHeaders(highestSharedBlockHash *externa
return return
} }
if doneIBD { if doneIBD {
doneChan <- struct{}{} close(blockHeadersMessageChan)
return return
} }
@ -246,7 +245,10 @@ func (flow *handleRelayInvsFlow) downloadHeaders(highestSharedBlockHash *externa
for { for {
select { select {
case blockHeadersMessage := <-blockHeadersMessageChan: case blockHeadersMessage, ok := <-blockHeadersMessageChan:
if !ok {
return nil
}
for _, header := range blockHeadersMessage.BlockHeaders { for _, header := range blockHeadersMessage.BlockHeaders {
err = flow.processHeader(header) err = flow.processHeader(header)
if err != nil { if err != nil {
@ -255,8 +257,6 @@ func (flow *handleRelayInvsFlow) downloadHeaders(highestSharedBlockHash *externa
} }
case err := <-errChan: case err := <-errChan:
return err return err
case <-doneChan:
return nil
} }
} }
} }