kotatogram/kotatogram-desktop
kotatogram/kotatogram-desktop
1
0
Fork 0
Code Issues Releases 45 Packages Wiki Activity

Send dependent requests in bunches.

Browse source
This commit is contained in:
John Preston 2021-07-06 11:31:01 +03:00
parent e28fb1211e
commit a0540e0486
1 changed files with 35 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
Telegram/SourceFiles/mtproto/mtp_instance.cpp
View file

@ -1017,14 +1017,30 @@ void Instance::Private::unregisterRequest(mtpRequestId requestId) {
_requestsByDc.erase(requestId);
}
{
QMutexLocker locker(&_dependentRequestsLock);
for (auto i = begin(_dependentRequests); i != end(_dependentRequests);) {
if (i->first == requestId) {
i = _dependentRequests.erase(i);
} else if (i->second == requestId) {
const auto resendingId = i->first;
i = _dependentRequests.erase(i);
auto toRemove = base::flat_set<mtpRequestId>();
auto toResend = base::flat_set<mtpRequestId>();
toRemove.emplace(requestId);
QMutexLocker locker(&_dependentRequestsLock);
auto handling = 0;
do {
handling = toResend.size();
for (const auto [resendingId, afterId] : _dependentRequests) {
if (toRemove.contains(afterId)) {
toRemove.emplace(resendingId);
toResend.emplace(resendingId);
}
}
} while (handling != toResend.size());
for (const auto removingId : toRemove) {
_dependentRequests.remove(removingId);
}
locker.unlock();
for (const auto resendingId : toResend) {
if (const auto shiftedDcId = queryRequestByDc(resendingId)) {
SerializedRequest request;
{
@ -1036,13 +1052,7 @@ void Instance::Private::unregisterRequest(mtpRequestId requestId) {
}
request = it->second;
}
request->after = SerializedRequest();
const auto session = getSession(qAbs(*shiftedDcId));
request->needsLayer = true;
session->sendPrepared(request);
}
} else {
++i;
getSession(qAbs(*shiftedDcId))->sendPrepared(request);
}
}
}
@ -1405,9 +1415,7 @@ bool Instance::Private::onErrorDefault(
}
if (!request->after) {
const auto session = getSession(qAbs(dcWithShift));
request->needsLayer = true;
session->sendPrepared(request);
getSession(qAbs(dcWithShift))->sendPrepared(request);
} else {
QMutexLocker locker(&_dependentRequestsLock);
_dependentRequests.emplace(requestId, request->after->requestId);