Add ElapsedTime method to JobProgressMetrics (#1660)

pkg/sources/job_progress.go

@@ -284,3 +284,16 @@ func (m JobProgressMetrics) PercentComplete() int {
 	}
 	return int(num * 100 / den)
 }
+
+// ElapsedTime is a convenience method that provides the elapsed time the job
+// has been running. If it hasn't started yet, 0 is returned. If it has
+// finished, the total time is returned.
+func (m JobProgressMetrics) ElapsedTime() time.Duration {
+	if m.StartTime.IsZero() {
+		return 0
+	}
+	if m.EndTime.IsZero() {
+		return time.Since(m.StartTime)
+	}
+	return m.EndTime.Sub(m.StartTime)
+}

pkg/sources/job_progress_test.go

@@ -115,3 +115,14 @@ func TestJobProgressDone(t *testing.T) {
 		assert.FailNow(t, "done should not block for a nil job")
 	}
 }
+
+func TestJobProgressElapsedTime(t *testing.T) {
+	metrics := JobProgressMetrics{}
+	assert.Equal(t, time.Duration(0), metrics.ElapsedTime())
+
+	metrics.StartTime = time.Now()
+	assert.Greater(t, metrics.ElapsedTime(), time.Duration(0))
+
+	metrics.EndTime = metrics.StartTime.Add(1 * time.Hour)
+	assert.Equal(t, metrics.ElapsedTime(), 1*time.Hour)
+}