diff options
Diffstat (limited to 'src/mongo/s/sharding_task_executor.cpp')
-rw-r--r-- | src/mongo/s/sharding_task_executor.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/mongo/s/sharding_task_executor.cpp b/src/mongo/s/sharding_task_executor.cpp index ee15fdf3a6b..e7401272645 100644 --- a/src/mongo/s/sharding_task_executor.cpp +++ b/src/mongo/s/sharding_task_executor.cpp @@ -111,11 +111,13 @@ StatusWith<TaskExecutor::CallbackHandle> ShardingTaskExecutor::scheduleWorkAt( } StatusWith<TaskExecutor::CallbackHandle> ShardingTaskExecutor::scheduleRemoteCommand( - const RemoteCommandRequest& request, const RemoteCommandCallbackFn& cb) { + const RemoteCommandRequest& request, + const RemoteCommandCallbackFn& cb, + const transport::BatonHandle& baton) { // schedule the user's callback if there is not opCtx if (!request.opCtx) { - return _executor->scheduleRemoteCommand(request, cb); + return _executor->scheduleRemoteCommand(request, cb, baton); } boost::optional<RemoteCommandRequest> newRequest; @@ -201,7 +203,7 @@ StatusWith<TaskExecutor::CallbackHandle> ShardingTaskExecutor::scheduleRemoteCom } }; - return _executor->scheduleRemoteCommand(newRequest ? *newRequest : request, shardingCb); + return _executor->scheduleRemoteCommand(newRequest ? *newRequest : request, shardingCb, baton); } void ShardingTaskExecutor::cancel(const CallbackHandle& cbHandle) { |