// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-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) // #pragma once #include "td/telegram/DialogId.h" #include "td/telegram/files/FileType.h" #include "td/utils/common.h" #include "td/utils/StringBuilder.h" #include "td/utils/tl_helpers.h" #include <array> #include <unordered_map> namespace td { namespace td_api { class storageStatistics; class storageStatisticsFast; } // namespace td_api } // namespace td namespace td { struct FileTypeStat { int64 size{0}; int32 cnt{0}; }; template <class StorerT> void store(const FileTypeStat &stat, StorerT &storer) { using ::td::store; store(stat.size, storer); store(stat.cnt, storer); } template <class ParserT> void parse(FileTypeStat &stat, ParserT &parser) { using ::td::parse; parse(stat.size, parser); parse(stat.cnt, parser); } struct FullFileInfo { FileType file_type; string path; DialogId owner_dialog_id; int64 size; uint64 atime_nsec; uint64 mtime_nsec; }; struct FileStatsFast { int64 size{0}; int32 count{0}; int64 database_size{0}; int64 language_pack_database_size{0}; int64 log_size{0}; FileStatsFast(int64 size, int32 count, int64 database_size, int64 language_pack_database_size, int64 log_size) : size(size) , count(count) , database_size(database_size) , language_pack_database_size(language_pack_database_size) , log_size(log_size) { } tl_object_ptr<td_api::storageStatisticsFast> as_td_api() const; }; struct FileStats { bool need_all_files{false}; bool split_by_owner_dialog_id{false}; using StatByType = std::array<FileTypeStat, file_type_size>; StatByType stat_by_type; std::unordered_map<DialogId, StatByType, DialogIdHash> stat_by_owner_dialog_id; std::vector<FullFileInfo> all_files; void add_copy(const FullFileInfo &info); void add(FullFileInfo &&info); void apply_dialog_limit(int32 limit); tl_object_ptr<td_api::storageStatistics> as_td_api() const; std::vector<DialogId> get_dialog_ids() const; FileTypeStat get_total_nontemp_stat() const; private: void add_impl(const FullFileInfo &info); void add(StatByType &by_type, FileType file_type, int64 size); }; StringBuilder &operator<<(StringBuilder &sb, const FileStats &file_stats); } // namespace td