aboutsummaryrefslogtreecommitdiffstats
path: root/eth
diff options
context:
space:
mode:
authorgary rong <garyrong0905@gmail.com>2018-04-09 20:46:27 +0800
committerFelix Lange <fjl@users.noreply.github.com>2018-04-09 20:46:27 +0800
commit1100e8ba633d968da8ae2caca0a5d0cf48bcfa92 (patch)
tree86bc7ad0a079cd05c065078999618b4dcca616b0 /eth
parent0fac705ed0a2efcda4da9dce8971bbda73f299d5 (diff)
downloaddexon-1100e8ba633d968da8ae2caca0a5d0cf48bcfa92.tar
dexon-1100e8ba633d968da8ae2caca0a5d0cf48bcfa92.tar.gz
dexon-1100e8ba633d968da8ae2caca0a5d0cf48bcfa92.tar.bz2
dexon-1100e8ba633d968da8ae2caca0a5d0cf48bcfa92.tar.lz
dexon-1100e8ba633d968da8ae2caca0a5d0cf48bcfa92.tar.xz
dexon-1100e8ba633d968da8ae2caca0a5d0cf48bcfa92.tar.zst
dexon-1100e8ba633d968da8ae2caca0a5d0cf48bcfa92.zip
eth/downloader: flush state sync data before exit (#16280)
Diffstat (limited to 'eth')
-rw-r--r--eth/downloader/statesync.go18
1 files changed, 13 insertions, 5 deletions
diff --git a/eth/downloader/statesync.go b/eth/downloader/statesync.go
index 4071d0ad9..521ee25a0 100644
--- a/eth/downloader/statesync.go
+++ b/eth/downloader/statesync.go
@@ -274,15 +274,21 @@ func (s *stateSync) Cancel() error {
// receive data from peers, rather those are buffered up in the downloader and
// pushed here async. The reason is to decouple processing from data receipt
// and timeouts.
-func (s *stateSync) loop() error {
+func (s *stateSync) loop() (err error) {
// Listen for new peer events to assign tasks to them
newPeer := make(chan *peerConnection, 1024)
peerSub := s.d.peers.SubscribeNewPeers(newPeer)
defer peerSub.Unsubscribe()
+ defer func() {
+ cerr := s.commit(true)
+ if err == nil {
+ err = cerr
+ }
+ }()
// Keep assigning new tasks until the sync completes or aborts
for s.sched.Pending() > 0 {
- if err := s.commit(false); err != nil {
+ if err = s.commit(false); err != nil {
return err
}
s.assignTasks()
@@ -307,14 +313,14 @@ func (s *stateSync) loop() error {
s.d.dropPeer(req.peer.id)
}
// Process all the received blobs and check for stale delivery
- if err := s.process(req); err != nil {
+ if err = s.process(req); err != nil {
log.Warn("Node data write error", "err", err)
return err
}
req.peer.SetNodeDataIdle(len(req.response))
}
}
- return s.commit(true)
+ return nil
}
func (s *stateSync) commit(force bool) error {
@@ -323,7 +329,9 @@ func (s *stateSync) commit(force bool) error {
}
start := time.Now()
b := s.d.stateDB.NewBatch()
- s.sched.Commit(b)
+ if written, err := s.sched.Commit(b); written == 0 || err != nil {
+ return err
+ }
if err := b.Write(); err != nil {
return fmt.Errorf("DB write error: %v", err)
}