netsync add test case (#365)

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

diff --git a/netsync/sync_manager.go b/netsync/sync_manager.go
index 61645bf..95ffbb4 100644 (file)
--- a/netsync/sync_manager.go
+++ b/netsync/sync_manager.go
@@ -7,6 +7,7 @@ import (
 
        "github.com/vapor/config"
        "github.com/vapor/consensus"
+       dbm "github.com/vapor/database/leveldb"
        "github.com/vapor/event"
        "github.com/vapor/netsync/chainmgr"
        "github.com/vapor/netsync/consensusmgr"
@@ -55,14 +56,14 @@ type SyncManager struct {
 }
 
 // NewSyncManager create sync manager and set switch.
-func NewSyncManager(config *config.Config, chain *protocol.Chain, txPool *protocol.TxPool, dispatcher *event.Dispatcher) (*SyncManager, error) {
+func NewSyncManager(config *config.Config, chain *protocol.Chain, txPool *protocol.TxPool, dispatcher *event.Dispatcher, fastSyncDB dbm.DB) (*SyncManager, error) {
        sw, err := p2p.NewSwitch(config)
        if err != nil {
                return nil, err
        }
        peers := peers.NewPeerSet(sw)
 
-       chainManger, err := chainmgr.NewManager(config, sw, chain, txPool, dispatcher, peers)
+       chainManger, err := chainmgr.NewManager(config, sw, chain, txPool, dispatcher, peers, fastSyncDB)
        if err != nil {
                return nil, err
        }
@@ -100,7 +101,7 @@ func (sm *SyncManager) Stop() {
 
 }
 
-// IsListening check if the bytomd service port is open?
+// IsListening check if the vapord service port is open?
 func (sm *SyncManager) IsListening() bool {
        if sm.config.VaultMode {
                return false