2a8ebc4906
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
127 lines
3.8 KiB
C++
127 lines
3.8 KiB
C++
#include <mln/storage/file_source_manager.hpp>
|
|
#include <mln/storage/resource.hpp>
|
|
#include <mln/storage/resource_options.hpp>
|
|
#include <mln/util/async_request.hpp>
|
|
#include <mln/util/client_options.hpp>
|
|
#include <mln/util/logging.hpp>
|
|
|
|
#include <atomic>
|
|
|
|
#include "file_source.hpp"
|
|
|
|
namespace mln {
|
|
|
|
std::atomic_size_t requestCount{0};
|
|
std::atomic_size_t transferredSize{0};
|
|
std::atomic_bool active{false};
|
|
std::atomic_bool offline{true};
|
|
|
|
ProxyFileSource::ProxyFileSource(std::shared_ptr<FileSource> defaultResourceLoader_,
|
|
const ResourceOptions& resourceOptions_,
|
|
const ClientOptions& clientOptions_)
|
|
: defaultResourceLoader(std::move(defaultResourceLoader_)),
|
|
resourceOptions(resourceOptions_.clone()),
|
|
clientOptions(clientOptions_.clone()) {
|
|
assert(defaultResourceLoader);
|
|
if (offline) {
|
|
std::shared_ptr<FileSource> dbfs = FileSourceManager::get()->getFileSource(
|
|
FileSourceType::Database, resourceOptions_, clientOptions_);
|
|
dbfs->setProperty(READ_ONLY_MODE_KEY, true);
|
|
}
|
|
}
|
|
|
|
ProxyFileSource::~ProxyFileSource() = default;
|
|
|
|
std::unique_ptr<AsyncRequest> ProxyFileSource::request(const Resource& resource, Callback callback) {
|
|
auto transformed = resource;
|
|
|
|
// If offline, force always loading the resource from the cache
|
|
// so we don't make any network request.
|
|
if (offline) {
|
|
transformed.loadingMethod = Resource::LoadingMethod::CacheOnly;
|
|
}
|
|
|
|
// This is needed for compatibility with the style tests that
|
|
// are using local:// instead of http:// which is the schema
|
|
// we support for cached files.
|
|
if (transformed.url.compare(0, 8, "local://") == 0) {
|
|
transformed.url.replace(0, 8, "http://");
|
|
|
|
if (transformed.kind == Resource::Kind::Tile && transformed.tileData) {
|
|
transformed.tileData->urlTemplate.replace(0, 8, "http://");
|
|
}
|
|
}
|
|
|
|
if (transformed.url.compare(0, 22, "http://localhost:2900/") == 0) {
|
|
transformed.url.replace(0, 22, "http://");
|
|
|
|
if (transformed.kind == Resource::Kind::Tile && transformed.tileData) {
|
|
transformed.tileData->urlTemplate.replace(0, 22, "http://");
|
|
}
|
|
}
|
|
|
|
return defaultResourceLoader->request(transformed, [=](Response response) {
|
|
if (transformed.loadingMethod == Resource::LoadingMethod::CacheOnly && response.noContent) {
|
|
if (transformed.kind == Resource::Kind::Tile && transformed.tileData) {
|
|
mln::Log::Info(
|
|
mln::Event::Database,
|
|
"Resource not found in cache: " + transformed.url + "(" + transformed.tileData->urlTemplate + ")");
|
|
} else {
|
|
mln::Log::Info(mln::Event::Database, "Resource not found in cache: " + transformed.url);
|
|
}
|
|
}
|
|
|
|
std::size_t size = response.data != nullptr ? response.data->size() : 0;
|
|
if (active) {
|
|
requestCount++;
|
|
transferredSize += size;
|
|
}
|
|
callback(response);
|
|
});
|
|
}
|
|
|
|
void ProxyFileSource::setResourceOptions(ResourceOptions options) {
|
|
resourceOptions = options;
|
|
}
|
|
|
|
ResourceOptions ProxyFileSource::getResourceOptions() {
|
|
return resourceOptions.clone();
|
|
}
|
|
|
|
void ProxyFileSource::setClientOptions(ClientOptions options) {
|
|
clientOptions = options;
|
|
}
|
|
|
|
ClientOptions ProxyFileSource::getClientOptions() {
|
|
return clientOptions.clone();
|
|
}
|
|
|
|
// static
|
|
void ProxyFileSource::setOffline(bool status) {
|
|
offline = status;
|
|
}
|
|
|
|
// static
|
|
void ProxyFileSource::setTrackingActive(bool active_) {
|
|
active = active_;
|
|
requestCount = 0;
|
|
transferredSize = 0;
|
|
}
|
|
|
|
// static
|
|
bool ProxyFileSource::isTrackingActive() {
|
|
return active;
|
|
}
|
|
|
|
// static
|
|
size_t ProxyFileSource::getRequestCount() {
|
|
return requestCount;
|
|
}
|
|
|
|
// static
|
|
size_t ProxyFileSource::getTransferredSize() {
|
|
return transferredSize;
|
|
}
|
|
|
|
} // namespace mln
|