This commit is contained in:
2025-08-26 17:34:05 +06:00
parent bdb6395351
commit 57d6e816fc
3 changed files with 63 additions and 5 deletions

View File

@@ -82,10 +82,10 @@ public:
/*
Менеджер предоставления ресурсов. Управляет ресурс паками
и хранением кешированных ресурсов сервера.
и хранением кешированных ресурсов с сервера.
Интерфейс однопоточный.
Обработка файлов в отдельном потоке
Обработка файлов в отдельном потоке.
*/
class AssetsManager : public IAsyncDestructible {
public:
@@ -153,12 +153,38 @@ public:
// После этого вызова уже нельзя будет обращатся ко внешним ресурсам
void shutdown() {
assert(NeedShutdown);
OffThread.join();
}
bool hasError() {
return IssuedAnError;
}
private:
Logger LOG = "Client>ResourceHandler";
const fs::path CachePath;
const fs::path
CachePath,
PathDatabase = CachePath / "db.sqlite3",
PathFiles = CachePath / "blobs";
static constexpr size_t SMALL_RESOURCE = 1 << 21;
sqlite3 *DB = nullptr; // База хранения кеша меньше 2мб и информации о кеше на диске
sqlite3_stmt
*STMT_DISK_INSERT = nullptr, // Вставка записи о хеше
*STMT_DISK_UPDATE_TIME = nullptr, // Обновить дату последнего использования
*STMT_DISK_REMOVE = nullptr, // Удалить хеш
*STMT_DISK_CONTAINS = nullptr, // Проверка наличия хеша
*STMT_DISK_SUM = nullptr, // Вычисляет занятое место на диске
*STMT_DISK_COUNT = nullptr, // Возвращает количество записей
*STMT_INLINE_INSERT = nullptr, // Вставка ресурса
*STMT_INLINE_GET = nullptr, // Поиск ресурса по хешу
*STMT_INLINE_UPDATE_TIME = nullptr, // Обновить дату последнего использования
*STMT_INLINE_SUM = nullptr, // Размер внутреннего хранилища
*STMT_INLINE_COUNT = nullptr; // Возвращает количество записей
// Полный размер данных на диске (насколько известно)
volatile size_t DatabaseSize = 0;
// Очередь задач на чтение
@@ -179,7 +205,7 @@ private:
TOS::SpinlockObject<Changes_t> Changes;
bool NeedShutdown = false;
bool NeedShutdown = false, IssuedAnError = false;
std::thread OffThread;

View File

@@ -1992,7 +1992,7 @@ struct Resource::InlineMMap {
};
struct Resource::InlinePtr {
std::vector<uint8_t> Data;
std::u8string Data;
Hash_t Hash;
InlinePtr(const uint8_t* data, size_t size) {
@@ -2001,6 +2001,11 @@ struct Resource::InlinePtr {
Hash = sha2::sha256(data, size);
}
InlinePtr(std::u8string&& data) {
Data = std::move(data);
Hash = sha2::sha256((const uint8_t*) Data.data(), Data.size());
}
const std::byte* data() const { return (const std::byte*) Data.data(); }
size_t size() const { return Data.size(); }
};
@@ -2014,10 +2019,28 @@ Resource::Resource(const uint8_t* data, size_t size)
: In(std::make_shared<std::variant<InlineMMap, InlinePtr>>(InlinePtr(data, size)))
{}
Resource::Resource(const std::u8string& data)
: In(std::make_shared<std::variant<InlineMMap, InlinePtr>>(InlinePtr((const uint8_t*) data.data(), data.size())))
{}
Resource::Resource(std::u8string&& data) {
}
const std::byte* Resource::data() const { assert(In); return std::visit<const std::byte*>([](auto& obj){ return obj.data(); }, *In); }
size_t Resource::size() const { assert(In); return std::visit<size_t>([](auto& obj){ return obj.size(); }, *In); }
Hash_t Resource::hash() const { assert(In); return std::visit<Hash_t>([](auto& obj){ return obj.Hash; }, *In); }
Resource Resource::convertToMem() const {
if(InlineMMap* ptr = std::get_if<InlineMMap>(&*In)) {
std::u8string data(ptr->size(), '\0');
std::copy(ptr->data(), ptr->data()+ptr->size(), (std::byte*) data.data());
return Resource(std::move(data));
} else {
return *this;
}
}
auto Resource::operator<=>(const Resource&) const = default;
}

View File

@@ -700,8 +700,11 @@ private:
std::shared_ptr<std::variant<InlineMMap, InlinePtr>> In;
public:
Resource() = default;
Resource(std::filesystem::path path);
Resource(const uint8_t* data, size_t size);
Resource(const std::u8string& data);
Resource(std::u8string&& data);
Resource(const Resource&) = default;
Resource(Resource&&) = default;
@@ -712,6 +715,12 @@ public:
const std::byte* data() const;
size_t size() const;
Hash_t hash() const;
Resource convertToMem() const;
operator bool() const {
return (bool) In;
}
};
}