tdlight/td/telegram/files/FileStats.h

89 lines
2.1 KiB
C
Raw Normal View History

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019
//
// 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/FileLocation.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 T>
void store(const FileTypeStat &stat, T &storer) {
using ::td::store;
store(stat.size, storer);
store(stat.cnt, storer);
}
template <class T>
void parse(FileTypeStat &stat, T &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 db_size{0};
FileStatsFast(int64 size, int32 count, int64 db_size) : size(size), count(count), db_size(db_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(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(StatByType &by_type, FileType file_type, int64 size);
};
StringBuilder &operator<<(StringBuilder &sb, const FileStats &file_stats);
} // namespace td