122 lines
3.7 KiB
C++
122 lines
3.7 KiB
C++
#define BOOST_ASIO_HAS_IO_URING 1
|
||
#include "BinaryResourceManager.hpp"
|
||
#include <memory>
|
||
#include <boost/system.hpp>
|
||
#include <boost/asio.hpp>
|
||
#include <boost/asio/stream_file.hpp>
|
||
#include <TOSLib.hpp>
|
||
|
||
|
||
namespace LV::Server {
|
||
|
||
|
||
|
||
BinaryResourceManager::BinaryResourceManager(asio::io_context &ioc,
|
||
std::shared_ptr<ResourceFile> zeroResource)
|
||
: AsyncObject(ioc), ZeroResource(std::move(zeroResource))
|
||
{
|
||
}
|
||
|
||
BinaryResourceManager::~BinaryResourceManager() {
|
||
|
||
}
|
||
|
||
void BinaryResourceManager::recheckResources() {
|
||
|
||
}
|
||
|
||
ResourceId_t BinaryResourceManager::mapUriToId(const std::string &uri) {
|
||
UriParse parse = parseUri(uri);
|
||
if(parse.Protocol != "assets")
|
||
MAKE_ERROR("Неизвестный протокол ресурса '" << parse.Protocol << "'. Полный путь: " << parse.Orig);
|
||
|
||
return getResource_Assets(parse.Path);
|
||
}
|
||
|
||
void BinaryResourceManager::needResourceResponse(const std::vector<ResourceId_t> &resources) {
|
||
UpdatedResources.lock_write()->insert(resources.end(), resources.begin(), resources.end());
|
||
}
|
||
|
||
void BinaryResourceManager::update(float dtime) {
|
||
if(UpdatedResources.no_lock_readable().empty())
|
||
return;
|
||
|
||
auto lock = UpdatedResources.lock_write();
|
||
for(ResourceId_t resId : *lock) {
|
||
std::shared_ptr<ResourceFile> &objRes = PreparedInformation[resId];
|
||
if(objRes)
|
||
continue;
|
||
|
||
auto iter = ResourcesInfo.find(resId);
|
||
if(iter == ResourcesInfo.end()) {
|
||
objRes = ZeroResource;
|
||
} else {
|
||
objRes = iter->second->Loaded;
|
||
}
|
||
}
|
||
}
|
||
|
||
BinaryResourceManager::UriParse BinaryResourceManager::parseUri(const std::string &uri) {
|
||
size_t pos = uri.find("://");
|
||
|
||
if(pos == std::string::npos)
|
||
return {uri, "assets", uri};
|
||
else
|
||
return {uri, uri.substr(0, pos), uri.substr(pos+3)};
|
||
}
|
||
|
||
ResourceId_t BinaryResourceManager::getResource_Assets(std::string path) {
|
||
size_t pos = path.find("/");
|
||
|
||
if(pos == std::string::npos)
|
||
MAKE_ERROR("Не действительный путь assets: '" << path << "'");
|
||
|
||
std::string domain = path.substr(0, pos);
|
||
std::string inDomainPath = path.substr(pos+1);
|
||
|
||
ResourceId_t &resId = KnownResource[path];
|
||
if(!resId)
|
||
resId = NextId++;
|
||
|
||
std::shared_ptr<Resource> &res = ResourcesInfo[resId];
|
||
if(!res) {
|
||
res = std::make_shared<Resource>();
|
||
res->Loaded = ZeroResource;
|
||
|
||
auto iter = Domains.find("domain");
|
||
if(iter == Domains.end()) {
|
||
UpdatedResources.lock_write()->push_back(resId);
|
||
} else {
|
||
res->IsLoading = true;
|
||
co_spawn(checkResource_Assets(resId, iter->second / inDomainPath, res));
|
||
}
|
||
}
|
||
|
||
return resId;
|
||
}
|
||
|
||
coro<> BinaryResourceManager::checkResource_Assets(ResourceId_t id, fs::path path, std::shared_ptr<Resource> res) {
|
||
try {
|
||
asio::stream_file fd(IOC, path, asio::stream_file::flags::read_only);
|
||
|
||
if(fd.size() > 1024*1024*16)
|
||
MAKE_ERROR("Превышен лимит размера файла: " << fd.size() << " > " << 1024*1024*16);
|
||
|
||
std::shared_ptr<ResourceFile> file = std::make_shared<ResourceFile>();
|
||
file->Data.resize(fd.size());
|
||
co_await asio::async_read(fd, asio::mutable_buffer(file->Data.data(), file->Data.size()));
|
||
file->calcHash();
|
||
res->LastError.clear();
|
||
} catch(const std::exception &exc) {
|
||
res->LastError = exc.what();
|
||
res->IsLoading = false;
|
||
|
||
if(const boost::system::system_error *errc = dynamic_cast<const boost::system::system_error*>(&exc); errc && errc->code() == asio::error::operation_aborted)
|
||
co_return;
|
||
}
|
||
|
||
res->IsLoading = false;
|
||
UpdatedResources.lock_write()->push_back(id);
|
||
}
|
||
|
||
} |