45e855f89d
I can't maintain anymore this amount of features while keeping the library constantly updated and without bugs. Every merge was taking me multiple hours of revisioning the code. I give up. From this commit onwards TDLight will only have small useful customizations that are easy to maintain. Now the people relying on the OptimizeMemory method can restart the session every N hours to free up the memory. The real way to keep a low memory usage must involve a huge refactoring to allow the unloading of the caches into the sqlite database, similar to what's already happening with messages data. Only Levlam has the ability to implement this without needing to merge the upstream everytime.
198 lines
5.1 KiB
C++
198 lines
5.1 KiB
C++
//
|
|
// Copyright Andrea Cavalli (nospam@warp.ovh) 2020
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
#include "td/telegram/MemoryManager.h"
|
|
|
|
#include "td/telegram/secret_api.h"
|
|
#include "td/telegram/td_api.h"
|
|
#include "td/telegram/telegram_api.h"
|
|
|
|
#include "td/telegram/AccessRights.h"
|
|
#include "td/telegram/AuthManager.h"
|
|
#include "td/telegram/ConfigManager.h"
|
|
#include "td/telegram/ConfigShared.h"
|
|
#include "td/telegram/ContactsManager.h"
|
|
#include "td/telegram/DialogId.h"
|
|
#include "td/telegram/Document.h"
|
|
#include "td/telegram/DocumentsManager.h"
|
|
#include "td/telegram/FileReferenceManager.h"
|
|
#include "td/telegram/files/FileLocation.h"
|
|
#include "td/telegram/files/FileManager.h"
|
|
#include "td/telegram/WebPagesManager.h"
|
|
#include "td/telegram/StickersManager.h"
|
|
#include "td/telegram/VideoNotesManager.h"
|
|
#include "td/telegram/VideosManager.h"
|
|
#include "td/telegram/AudiosManager.h"
|
|
#include "td/telegram/AnimationsManager.h"
|
|
#include "td/telegram/GroupCallManager.h"
|
|
#include "td/telegram/BackgroundManager.h"
|
|
#include "td/telegram/PollManager.h"
|
|
#include "td/telegram/InlineQueriesManager.h"
|
|
#include "td/telegram/files/FileType.h"
|
|
#include "td/telegram/Global.h"
|
|
#include "td/telegram/LanguagePackManager.h"
|
|
#include "td/telegram/logevent/LogEvent.h"
|
|
#include "td/telegram/MessagesManager.h"
|
|
#include "td/telegram/misc.h"
|
|
#include "td/telegram/net/DcId.h"
|
|
#include "td/telegram/net/MtprotoHeader.h"
|
|
#include "td/telegram/Td.h"
|
|
#include "td/telegram/TdDb.h"
|
|
|
|
#include "td/actor/MultiPromise.h"
|
|
#include "td/actor/PromiseFuture.h"
|
|
#include "td/actor/SleepActor.h"
|
|
|
|
#include "td/db/SqliteKeyValue.h"
|
|
#include "td/db/SqliteKeyValueAsync.h"
|
|
|
|
#include "td/utils/format.h"
|
|
#include "td/utils/logging.h"
|
|
#include "td/utils/misc.h"
|
|
#include "td/utils/PathView.h"
|
|
#include "td/utils/Random.h"
|
|
#include "td/utils/Slice.h"
|
|
#include "td/utils/Time.h"
|
|
#include "td/utils/tl_helpers.h"
|
|
#include "td/utils/utf8.h"
|
|
|
|
#include <algorithm>
|
|
#include <limits>
|
|
#include <type_traits>
|
|
#include <unordered_set>
|
|
#include <numeric>
|
|
|
|
namespace td {
|
|
|
|
tl_object_ptr<td_api::memoryStatistics> MemoryStats::get_memory_statistics_object() const {
|
|
return make_tl_object<td_api::memoryStatistics>(debug);
|
|
}
|
|
|
|
MemoryManager::MemoryManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) {
|
|
}
|
|
|
|
void MemoryManager::start_up() {
|
|
}
|
|
|
|
void MemoryManager::tear_down() {
|
|
parent_.reset();
|
|
}
|
|
|
|
void MemoryManager::get_memory_stats(bool full, Promise<MemoryStats> promise) const {
|
|
vector<string> output = {"{"};
|
|
|
|
output.push_back("\"memory_stats\":{");
|
|
|
|
output.push_back("\"messages_manager_\":{");
|
|
td_->messages_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"contacts_manager_\":{");
|
|
td_->contacts_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"web_pages_manager_\":{");
|
|
td_->web_pages_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"stickers_manager_\":{");
|
|
td_->stickers_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"documents_manager_\":{");
|
|
td_->documents_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"video_notes_manager_\":{");
|
|
td_->video_notes_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"videos_manager_\":{");
|
|
td_->videos_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"audios_manager_\":{");
|
|
td_->audios_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"animations_manager_\":{");
|
|
td_->animations_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"file_manager_\":{");
|
|
td_->file_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"file_reference_manager_\":{");
|
|
td_->file_reference_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"group_call_manager_\":{");
|
|
td_->group_call_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"background_manager_\":{");
|
|
td_->background_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"inline_queries_manager_\":{");
|
|
td_->inline_queries_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
|
|
output.push_back(",");
|
|
|
|
output.push_back("\"poll_manager_\":{");
|
|
td_->poll_manager_->memory_stats(output);
|
|
output.push_back("}");
|
|
|
|
output.push_back("}");
|
|
|
|
output.push_back("}");
|
|
|
|
string s;
|
|
s = accumulate(output.begin(), output.end(), s);
|
|
auto value = MemoryStats(s);
|
|
|
|
promise.set_value(std::move(value));
|
|
}
|
|
|
|
void MemoryManager::get_current_state(vector<td_api::object_ptr<td_api::Update>> &updates) const {
|
|
if (td_->auth_manager_->is_bot()) {
|
|
return;
|
|
}
|
|
|
|
// Never return updates
|
|
}
|
|
|
|
} // namespace td
|