From 712b17a01909f3e99ac1e36efd86a3100a5efb31 Mon Sep 17 00:00:00 2001
From: Zsolt Felfoldi <zsfelfoldi@gmail.com>
Date: Fri, 5 May 2017 21:25:06 +0200
Subject: [PATCH] core: added updateMsg to ChainSectionProcessorBackend
 interface

---
 core/chain_processor.go | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/core/chain_processor.go b/core/chain_processor.go
index 7112c3047c..28335001f6 100644
--- a/core/chain_processor.go
+++ b/core/chain_processor.go
@@ -59,6 +59,7 @@ type ChainSectionProcessorBackend interface {
 	Process(idx uint64) bool
 	GetStored() uint64
 	SetStored(count uint64)
+	UpdateMsg(done, all uint64)
 }
 
 // NewChainSectionProcessor creates a new  ChainSectionProcessor
@@ -80,6 +81,7 @@ func (csp *ChainSectionProcessor) updateLoop() {
 	tryUpdate := make(chan struct{}, 1)
 	updating := false
 	var targetCnt uint64
+	updateMsg := false
 
 	for {
 		select {
@@ -93,6 +95,10 @@ func (csp *ChainSectionProcessor) updateLoop() {
 		case <-tryUpdate:
 			csp.lock.Lock()
 			if targetCnt > csp.stored {
+				if !updateMsg && targetCnt > csp.stored+1 {
+					updateMsg = true
+					csp.backend.UpdateMsg(csp.stored, targetCnt)
+				}
 				csp.calcValid = true
 				csp.calcIdx = csp.stored
 
@@ -103,12 +109,16 @@ func (csp *ChainSectionProcessor) updateLoop() {
 				if ok && csp.calcValid {
 					csp.stored = csp.calcIdx + 1
 					csp.backend.SetStored(csp.stored)
-
+					if updateMsg {
+						csp.backend.UpdateMsg(csp.stored, targetCnt)
+						if csp.stored >= targetCnt {
+							updateMsg = false
+						}
+					}
 					csp.lastForwarded = csp.stored*csp.sectionSize - 1
 					for _, cp := range csp.childProcessors {
 						cp.NewHead(csp.lastForwarded, false)
 					}
-
 				}
 				csp.calcValid = false
 			}