tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix walk_path on Windows.

Browse Source 
GitOrigin-RevId: ca6eaea457018869aa476009acc3fd0f26a40999
This commit is contained in:
levlam 2018-01-26 04:06:53 +03:00
parent 648a8eb0af
commit e66cc39f95
3 changed files with 21 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
td/telegram/files/FileLocation.h
View File

@ -1146,10 +1146,17 @@ class FileData {
}
};
inline StringBuilder &operator<<(StringBuilder &sb, const FileData &file_data) {
sb << "[" << tag("name", file_data.name_);
sb << "[" << tag("name", file_data.name_) << " " << file_data.owner_dialog_id_ << " " << tag("size", file_data.size_)
<< tag("expected_size", file_data.expected_size_);
if (!file_data.url_.empty()) {
sb << tag("url", file_data.url_);
}
if (file_data.local_.type() == LocalFileLocation::Type::Full) {
sb << " local " << file_data.local_.full();
}
if (file_data.generate_.type() == GenerateFileLocation::Type::Full) {
sb << " generate " << file_data.generate_.full();
}
if (file_data.remote_.type() == RemoteFileLocation::Type::Full) {
sb << " remote " << file_data.remote_.full();
}

11
td/telegram/files/FileStatsWorker.cpp
View File

@ -50,7 +50,7 @@ Status scan_db(CallbackT &&callback) {
FileData data;
auto status = unserialize(data, value);
if (status.is_error()) {
LOG(ERROR) << "Invalid FileData in db " << tag("value", format::escaped(value));
LOG(ERROR) << "Invalid FileData in the database " << tag("value", format::escaped(value));
return;
}
DbFileInfo info;
@ -67,10 +67,11 @@ Status scan_db(CallbackT &&callback) {
if (path_view.is_relative()) {
info.path = get_files_base_dir(info.file_type) + info.path;
}
// LOG(INFO) << "Found file in the database: " << data << " " << info.path;
info.owner_dialog_id = data.owner_dialog_id_;
info.size = data.size_;
if (info.size == 0 && data.local_.type() == LocalFileLocation::Type::Full) {
LOG(ERROR) << "Unknown size in db";
LOG(ERROR) << "Unknown size in the database";
return;
}
callback(info);
@ -147,7 +148,10 @@ void FileStatsWorker::get_stats(bool need_all_files, bool split_by_owner_dialog_
info.size = fs_info.size;
info.atime_nsec = fs_info.atime_nsec;
info.mtime_nsec = fs_info.mtime_nsec;
full_infos.push_back(info);
// LOG(INFO) << "Found file of size " << info.size << " at " << info.path;
full_infos.push_back(std::move(info));
});
std::unordered_map<size_t, size_t> hash_to_pos;
@ -161,6 +165,7 @@ void FileStatsWorker::get_stats(bool need_all_files, bool split_by_owner_dialog_
if (it == hash_to_pos.end()) {
return;
}
// LOG(INFO) << "Match! " << db_info.path << " from " << db_info.owner_dialog_id;
full_infos[it->second].owner_dialog_id = db_info.owner_dialog_id;
});

6
tdutils/td/utils/port/path.h
View File

@ -212,7 +212,11 @@ Status walk_path_dir(const std::wstring &dir_name, Func &&func) {
template <class Func>
Status walk_path(CSlice path, Func &&func) {
TRY_RESULT(wpath, to_wstring(path));
Slice path_slice = path;
while (!path_slice.empty() && (path_slice.back() == '/' || path_slice.back() == '\\')) {
path_slice.remove_suffix(1);
}
TRY_RESULT(wpath, to_wstring(path_slice));
return detail::walk_path_dir(wpath.c_str(), func);
}