Merge pull request #29 from github/status-eta

Adding ETA starting at 2% progress
github · May 5, 2016 · 619be65 · 619be65
2 parents ee673e2 + 6528010
commit 619be65
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/go/logic/migrator.go b/go/logic/migrator.go
@@ -550,6 +550,18 @@ func (this *Migrator) printStatus() {
 	eta := "N/A"
 	if isThrottled, throttleReason := this.migrationContext.IsThrottled(); isThrottled {
 		eta = fmt.Sprintf("throttled, %s", throttleReason)
+	} else if progressPct > 100.0 {
+		eta = "Due"
+	} else if progressPct >= 2.0 {
+		elapsedRowCopySeconds := this.migrationContext.ElapsedRowCopyTime().Seconds()
+		totalExpectedSeconds := elapsedRowCopySeconds * float64(rowsEstimate) / float64(totalRowsCopied)
+		etaSeconds := totalExpectedSeconds - elapsedRowCopySeconds
+		etaDuration := time.Duration(etaSeconds) * time.Second
+		if etaDuration >= 0 {
+			eta = base.PrettifyDurationOutput(etaDuration)
+		} else {
+			eta = "Due"
+		}
 	}
 	status := fmt.Sprintf("Copy: %d/%d %.1f%%; Applied: %d; Backlog: %d/%d; Elapsed: %+v(copy), %+v(total); ETA: %s",
 		totalRowsCopied, rowsEstimate, progressPct,