mongodb/mongo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SERVER-113550 Fix ticket exhaustion issue in FLE2 aggregation and explain (#44946)

Browse Source 
GitOrigin-RevId: 4ee7aee06e13a640d400de56108ec676477ca7ef
This commit is contained in:
Erwin Pe 2025-12-08 15:29:56 -05:00 committed by MongoDB Bot
parent e6a4d5e471
commit e49562e8d6
1 changed files with 11 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/mongo/db/fle_crud.cpp
View File

@ -119,6 +119,8 @@ MONGO_FAIL_POINT_DEFINE(fleCrudHangPreDelete);
MONGO_FAIL_POINT_DEFINE(fleCrudHangFindAndModify);
MONGO_FAIL_POINT_DEFINE(fleCrudHangPreFindAndModify);
MONGO_FAIL_POINT_DEFINE(fleCrudPauseNonTxnGetTags);
namespace mongo {
namespace {
std::vector<write_ops::WriteError> singleStatusToWriteErrors(const Status& status) {
@ -2025,15 +2027,10 @@ std::vector<std::vector<FLEEdgeCountInfo>> FLETagNoTXNQuery::getTags(
invariant(!_opCtx->inMultiDocumentTransaction());
// Pop off the current op context so we can get a fresh set of read concern settings
// Allow the thread to be killable. If interrupted, the call to runCommand will fail with the
// interruption.
auto client = _opCtx->getService()->makeClient("FLETagNoTXNQuery");
AlternativeClientRegion clientRegion(client);
auto opCtx = cc().makeOperationContext();
auto as = AuthorizationSession::get(cc());
as->grantInternalAuthorization();
// Push a fresh set of read concern settings for the getTags command.
auto& opCtxReadConcern = repl::ReadConcernArgs::get(_opCtx);
ON_BLOCK_EXIT([&, oldRC = opCtxReadConcern]() { opCtxReadConcern = oldRC; });
opCtxReadConcern = repl::ReadConcernArgs{};
const auto setDollarTenant = nss.tenantId() && gMultitenancySupport;
const auto vts = auth::ValidatedTenancyScope::get(_opCtx);
@ -2051,21 +2048,17 @@ std::vector<std::vector<FLEEdgeCountInfo>> FLETagNoTXNQuery::getTags(
getCountsCmd.setTokens(toTagSets(tokensSets));
getCountsCmd.setQueryType(queryTypeTranslation(type));
DBDirectClient directClient(opCtx.get());
auto request = getCountsCmd.serialize();
if (vts) {
request.validatedTenancyScope = *vts;
}
auto uniqueReply = directClient.runCommand(request);
fleCrudPauseNonTxnGetTags.pauseWhileSet();
auto uniqueReply = DBDirectClient(_opCtx).runCommand(request);
auto response = uniqueReply->getCommandReply();
auto status = getStatusFromWriteCommandReply(response);
uassertStatusOK(status);
auto reply = QECountInfosReply::parse(IDLParserContext("reply"), response);
return toEdgeCounts(reply.getCounts());
uassertStatusOK(getStatusFromWriteCommandReply(response));
return toEdgeCounts(QECountInfosReply::parse(IDLParserContext("reply"), response).getCounts());
}
} // namespace mongo