Notify syncFinished to syncStateListener when peer is synced (#400)

horizontalsystems · Jun 25, 2019 · a4378f7 · a4378f7
1 parent 9cd93c4
commit a4378f7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/BitcoinCore/BitcoinCore/Blocks/InitialBlockDownload.swift b/BitcoinCore/BitcoinCore/Blocks/InitialBlockDownload.swift
@@ -125,6 +125,7 @@ public class InitialBlockDownload {
         syncedPeers.append(peer)
 
         subject.onNext(.onPeerSynced(peer: peer))
+        syncStateListener.syncFinished()
     }
 
     private func setPeerNotSynced(_ peer: IPeer) {

diff --git a/BitcoinCore/BitcoinCore/Core/KitStateProvider.swift b/BitcoinCore/BitcoinCore/Core/KitStateProvider.swift
@@ -32,6 +32,10 @@ extension KitStateProvider: ISyncStateListener {
         currentBestBlockHeight = height
     }
 
+    func syncFinished() {
+        syncState = .synced
+    }
+
     func currentBestBlockHeightUpdated(height: Int32, maxBlockHeight: Int32) {
         if currentBestBlockHeight < height {
             currentBestBlockHeight = height

diff --git a/BitcoinCore/BitcoinCore/Core/Protocols.swift b/BitcoinCore/BitcoinCore/Core/Protocols.swift
@@ -223,6 +223,7 @@ protocol IConnectionTimeoutManager: class {
 }
 
 protocol ISyncStateListener: class {
+    func syncFinished()
     func syncStarted()
     func syncStopped()
     func initialBestBlockHeightUpdated(height: Int32)