diff options
Diffstat (limited to 'src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h')
-rw-r--r-- | src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h b/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h index a898e9fca73..cc66423e231 100644 --- a/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h +++ b/src/mongo/db/s/sharding_data_transform_metrics_test_fixture.h @@ -63,11 +63,11 @@ public: return _uuid; } - virtual int64_t getHighEstimateRemainingTimeMillis() const override { + virtual boost::optional<Milliseconds> getHighEstimateRemainingTimeMillis() const override { return _timeRemainingHigh; } - virtual int64_t getLowEstimateRemainingTimeMillis() const override { + virtual boost::optional<Milliseconds> getLowEstimateRemainingTimeMillis() const override { return _timeRemainingLow; } @@ -82,8 +82,8 @@ public: private: UUID _uuid; Date_t _startTime; - int64_t _timeRemainingHigh; - int64_t _timeRemainingLow; + Milliseconds _timeRemainingHigh; + Milliseconds _timeRemainingLow; ShardingDataTransformMetrics::Role _role; }; @@ -186,6 +186,10 @@ protected: return &clock.value(); } + ShardingDataTransformCumulativeMetrics* getCumulativeMetrics() { + return _cumulativeMetrics.get(); + } + using SpecialIndexBehaviorMap = stdx::unordered_map<int, std::function<void()>>; const SpecialIndexBehaviorMap kNoSpecialBehavior{}; SpecialIndexBehaviorMap registerAtIndex(int index, const ObserverMock* mock) { |