fix gracefull shutdown issue

j75689 · Apr 22, 2022 · 734d17f · 734d17f
1 parent a73aad6
commit 734d17f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/remote_state_verifier.go b/core/remote_state_verifier.go
@@ -121,6 +121,10 @@ func (vm *remoteVerifyManager) mainLoop() {
 		case <-pruneTicker.C:
 			vm.taskLock.Lock()
 			for hash, task := range vm.tasks {
+				if vm.bc.insertStopped() {
+					close(task.terminalCh)
+					continue
+				}
 				if vm.bc.CurrentHeader().Number.Cmp(task.blockHeader.Number) == 1 &&
 					vm.bc.CurrentHeader().Number.Uint64()-task.blockHeader.Number.Uint64() > pruneHeightDiff {
 					delete(vm.tasks, hash)
@@ -136,7 +140,6 @@ func (vm *remoteVerifyManager) mainLoop() {
 				vt.messageCh <- message
 			}
 			vm.taskLock.RUnlock()
-
 		// System stopped
 		case <-vm.bc.quit:
 			vm.taskLock.RLock()