loader: fix load progress (#89)

pingcap · Mar 23, 2019 · a8e91c0 · a8e91c0
1 parent b57feb4
commit a8e91c0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/loader/loader.go b/loader/loader.go
@@ -483,6 +483,10 @@ func (l *Loader) IsFreshTask() (bool, error) {
 
 // Restore begins the restore process.
 func (l *Loader) Restore(ctx context.Context) error {
+	// reset some counter used to calculate progress
+	l.totalDataSize.Set(0)
+	l.finishedDataSize.Set(0) // reset before load from checkpoint
+
 	if err := l.prepare(); err != nil {
 		log.Errorf("[loader] scan dir[%s] failed, err[%v]", l.cfg.Dir, err)
 		return errors.Trace(err)