Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / netsync / block_fetcher.go

diff --git a/netsync/block_fetcher.go b/netsync/block_fetcher.go
deleted file mode 100644 (file)
index f7f3fae..0000000
--- a/netsync/block_fetcher.go
+++ /dev/null
@@ -1,95 +0,0 @@
-package netsync
-
-import (
-       log "github.com/sirupsen/logrus"
-       "gopkg.in/karalabe/cookiejar.v2/collections/prque"
-
-       "github.com/vapor/chain"
-       "github.com/vapor/protocol/bc"
-)
-
-const (
-       maxBlockDistance = 64
-       maxMsgSetSize    = 128
-       newBlockChSize   = 64
-)
-
-// blockFetcher is responsible for accumulating block announcements from various peers
-// and scheduling them for retrieval.
-type blockFetcher struct {
-       chain chain.Chain
-       peers *peerSet
-
-       newBlockCh chan *blockMsg
-       queue      *prque.Prque
-       msgSet     map[bc.Hash]*blockMsg
-}
-
-//NewBlockFetcher creates a block fetcher to retrieve blocks of the new mined.
-func newBlockFetcher(chain chain.Chain, peers *peerSet) *blockFetcher {
-       f := &blockFetcher{
-               chain:      chain,
-               peers:      peers,
-               newBlockCh: make(chan *blockMsg, newBlockChSize),
-               queue:      prque.New(),
-               msgSet:     make(map[bc.Hash]*blockMsg),
-       }
-       go f.blockProcessor()
-       return f
-}
-
-func (f *blockFetcher) blockProcessor() {
-       for {
-               height := f.chain.BestBlockHeight()
-               for !f.queue.Empty() {
-                       msg := f.queue.PopItem().(*blockMsg)
-                       if msg.block.Height > height+1 {
-                               f.queue.Push(msg, -float32(msg.block.Height))
-                               break
-                       }
-
-                       f.insert(msg)
-                       delete(f.msgSet, msg.block.Hash())
-               }
-               f.add(<-f.newBlockCh)
-       }
-}
-
-func (f *blockFetcher) add(msg *blockMsg) {
-       bestHeight := f.chain.BestBlockHeight()
-       if len(f.msgSet) > maxMsgSetSize || bestHeight > msg.block.Height || msg.block.Height-bestHeight > maxBlockDistance {
-               return
-       }
-
-       blockHash := msg.block.Hash()
-       if _, ok := f.msgSet[blockHash]; !ok {
-               f.msgSet[blockHash] = msg
-               f.queue.Push(msg, -float32(msg.block.Height))
-               log.WithFields(log.Fields{
-                       "module":       logModule,
-                       "block height": msg.block.Height,
-                       "block hash":   blockHash.String(),
-               }).Debug("blockFetcher receive mine block")
-       }
-}
-
-func (f *blockFetcher) insert(msg *blockMsg) {
-       if _, err := f.chain.ProcessBlock(msg.block); err != nil {
-               peer := f.peers.getPeer(msg.peerID)
-               if peer == nil {
-                       return
-               }
-
-               f.peers.addBanScore(msg.peerID, 20, 0, err.Error())
-               return
-       }
-
-       if err := f.peers.broadcastMinedBlock(msg.block); err != nil {
-               log.WithFields(log.Fields{"module": logModule, "err": err}).Error("blockFetcher fail on broadcast new block")
-               return
-       }
-}
-
-func (f *blockFetcher) processNewBlock(msg *blockMsg) {
-       f.newBlockCh <- msg
-}