diff options
Diffstat (limited to 'src/mongo/db/query/telemetry.h')
| -rw-r--r-- | src/mongo/db/query/telemetry.h | 28 |
1 files changed, 13 insertions, 15 deletions
diff --git a/src/mongo/db/query/telemetry.h b/src/mongo/db/query/telemetry.h index 6fd59884ac4..3a2f41a7e4e 100644 --- a/src/mongo/db/query/telemetry.h +++ b/src/mongo/db/query/telemetry.h @@ -163,33 +163,31 @@ std::unique_ptr<TelemetryStore> resetTelemetryStore(const ServiceContext* servic bool isTelemetryEnabled(const ServiceContext* serviceCtx); /** - * Should we collect telemetry for a request? The decision is made based on the feature flag and - * telemetry parameters such as rate limiting. + * Register a request for telemetry collection. The telemetry machinery may decide not to collect + * anything but this should be called for all requests. The decision is made based on the feature + * flag and telemetry parameters such as rate limiting. * - * If the return value is a telemetry key in the form of BSONObj, this indicates the telemetry - * should be collected. Otherwise, telemetry should not be collected. + * The caller is still responsible for subsequently calling collectTelemetry() once the request is + * completed. * * Note that calling this affects internal state. It should be called once for each request for * which telemetry may be collected. */ -boost::optional<BSONObj> shouldCollectTelemetry(const AggregateCommandRequest& request, - const OperationContext* opCtx); +void registerAggRequest(const AggregateCommandRequest& request, OperationContext* opCtx); -boost::optional<BSONObj> shouldCollectTelemetry(const FindCommandRequest& request, - const NamespaceString& collection, - const OperationContext* opCtx); +void registerFindRequest(const FindCommandRequest& request, + const NamespaceString& collection, + OperationContext* opCtx); -boost::optional<BSONObj> shouldCollectTelemetry(const OperationContext* opCtx, - const BSONObj& telemetryKey); +void registerGetMoreRequest(OperationContext* opCtx, const PlanExplainer& planExplainer); + +void recordExecution(const OperationContext* opCtx, const OpDebug& opDebug, bool isFle); /** * Collect telemetry for the operation identified by `key`. The `isExec` flag should be set if it's * the beginning of execution (first batch) of results and not set for subsequent getMore() calls. */ -void collectTelemetry(const ServiceContext* serviceCtx, - const BSONObj& key, - const OpDebug& opDebug, - bool isExec); +void collectTelemetry(const OperationContext* opCtx, const OpDebug& opDebug); } // namespace telemetry } // namespace mongo |