diff --git a/src/go/mongolib/profiler/profiler_test.go b/src/go/mongolib/profiler/profiler_test.go index 4dc37c75..21e4ebba 100644 --- a/src/go/mongolib/profiler/profiler_test.go +++ b/src/go/mongolib/profiler/profiler_test.go @@ -317,6 +317,8 @@ func TestCalcTotalStats(t *testing.T) { if err != nil && !tutil.ShouldUpdateSamples() { t.Fatalf("cannot load expected results: %s", err.Error()) } + want.FirstSeen = want.FirstSeen.UTC() + want.LastSeen = want.LastSeen.UTC() iter := pmgomock.NewMockIterManager(ctrl) gomock.InOrder( @@ -347,6 +349,8 @@ func TestCalcTotalStats(t *testing.T) { fmt.Printf("cannot update samples: %s", err.Error()) } } + s.FirstSeen = s.FirstSeen.UTC() + s.LastSeen = s.LastSeen.UTC() if !reflect.DeepEqual(s, want) { t.Errorf("Invalid stats.\nGot:%#v\nWant: %#v\n", s, want) } diff --git a/src/go/mongolib/stats/stats.go b/src/go/mongolib/stats/stats.go index 31af1e99..75e4bb36 100644 --- a/src/go/mongolib/stats/stats.go +++ b/src/go/mongolib/stats/stats.go @@ -101,11 +101,10 @@ func (s *Stats) Add(doc proto.SystemProfile) error { qiac.NReturned = append(qiac.NReturned, float64(doc.Nreturned)) qiac.QueryTime = append(qiac.QueryTime, float64(doc.Millis)) qiac.ResponseLength = append(qiac.ResponseLength, float64(doc.ResponseLength)) - var zeroTime time.Time - if qiac.FirstSeen == zeroTime || qiac.FirstSeen.After(doc.Ts) { + if qiac.FirstSeen.IsZero() || qiac.FirstSeen.After(doc.Ts) { qiac.FirstSeen = doc.Ts } - if qiac.LastSeen == zeroTime || qiac.LastSeen.Before(doc.Ts) { + if qiac.LastSeen.IsZero() || qiac.LastSeen.Before(doc.Ts) { qiac.LastSeen = doc.Ts }