115 lines
3.0 KiB
C++
115 lines
3.0 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
|
|
//
|
|
// 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/FlatHashMap.h"
|
|
#include "td/utils/StringBuilder.h"
|
|
#include "td/utils/tl_helpers.h"
|
|
|
|
#include <array>
|
|
|
|
namespace td {
|
|
|
|
namespace td_api {
|
|
class storageStatistics;
|
|
class storageStatisticsFast;
|
|
} // namespace td_api
|
|
|
|
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> get_storage_statistics_fast_object() const;
|
|
};
|
|
|
|
class FileStats {
|
|
bool need_all_files_{false};
|
|
bool split_by_owner_dialog_id_{false};
|
|
|
|
using StatByType = std::array<FileTypeStat, MAX_FILE_TYPE>;
|
|
|
|
StatByType stat_by_type_;
|
|
FlatHashMap<DialogId, StatByType, DialogIdHash> stat_by_owner_dialog_id_;
|
|
vector<FullFileInfo> all_files_;
|
|
|
|
void add_impl(const FullFileInfo &info);
|
|
|
|
void add(StatByType &by_type, FileType file_type, int64 size);
|
|
|
|
static FileTypeStat get_nontemp_stat(const StatByType &by_type);
|
|
|
|
static td_api::object_ptr<td_api::storageStatisticsByChat> get_storage_statistics_by_chat_object(
|
|
DialogId dialog_id, const StatByType &stat_by_type_);
|
|
|
|
friend StringBuilder &operator<<(StringBuilder &sb, const FileStats &file_stats);
|
|
|
|
public:
|
|
FileStats(bool need_all_files, bool split_by_owner_dialog_id)
|
|
: need_all_files_(need_all_files), split_by_owner_dialog_id_(split_by_owner_dialog_id) {
|
|
}
|
|
|
|
void add_copy(const FullFileInfo &info);
|
|
|
|
void add(FullFileInfo &&info);
|
|
|
|
void apply_dialog_limit(int32 limit);
|
|
|
|
void apply_dialog_ids(const vector<DialogId> &dialog_ids);
|
|
|
|
tl_object_ptr<td_api::storageStatistics> get_storage_statistics_object() const;
|
|
|
|
vector<DialogId> get_dialog_ids() const;
|
|
|
|
FileTypeStat get_total_nontemp_stat() const;
|
|
|
|
vector<FullFileInfo> get_all_files();
|
|
};
|
|
|
|
StringBuilder &operator<<(StringBuilder &sb, const FileStats &file_stats);
|
|
|
|
} // namespace td
|