summaryrefslogtreecommitdiff
path: root/render-test/file_source.cpp
blob: 0968f1d2f017d229fab7f8ebb07f949d23887b48 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#include <mbgl/storage/resource_options.hpp>

#include "file_source.hpp"

namespace mbgl {

std::atomic_size_t requestCount{0};
std::atomic_size_t transferredSize{0};
std::atomic_bool active{false};

ProxyFileSource::ProxyFileSource(const std::string& cachePath,
                                 const std::string& assetPath,
                                 bool supportCacheOnlyRequests_)
    : DefaultFileSource(cachePath, assetPath, supportCacheOnlyRequests_) {}

ProxyFileSource::ProxyFileSource(const std::string& cachePath,
                                 std::unique_ptr<FileSource>&& assetFileSource_,
                                 bool supportCacheOnlyRequests_)
    : DefaultFileSource(cachePath, std::move(assetFileSource_), supportCacheOnlyRequests_) {}

ProxyFileSource::~ProxyFileSource() = default;

std::unique_ptr<AsyncRequest> ProxyFileSource::request(const Resource& resource, Callback callback) {
    auto result = DefaultFileSource::request(resource, [=](Response response) {
        std::size_t size = response.data != nullptr ? response.data->size() : 0;
        if (active) {
            requestCount++;
            transferredSize += size;
        }
        callback(response);
    });
    return result;
}

std::shared_ptr<FileSource> FileSource::createPlatformFileSource(const ResourceOptions& options) {
    auto fileSource = std::make_shared<ProxyFileSource>(
        options.cachePath(), options.assetPath(), options.supportsCacheOnlyRequests());
    fileSource->setAccessToken(options.accessToken());
    fileSource->setAPIBaseURL(options.baseURL());
    return fileSource;
}

// 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 mbgl