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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#include <mbgl/storage/default_file_source.hpp>
#include <mbgl/storage/asset_file_source.hpp>
#include <mbgl/storage/online_file_source.hpp>
#include <mbgl/storage/sqlite_cache.hpp>
#include <mbgl/platform/platform.hpp>
#include <mbgl/util/url.hpp>
#include <mbgl/util/work_request.hpp>
#include <cassert>
namespace {
const std::string assetProtocol = "asset://";
bool isAssetURL(const std::string& url) {
return std::equal(assetProtocol.begin(), assetProtocol.end(), url.begin());
}
} // namespace
namespace mbgl {
class DefaultFileSource::Impl {
public:
Impl(const std::string& cachePath, const std::string& assetRoot)
: assetFileSource(assetRoot),
cache(SQLiteCache::getShared(cachePath)) {
}
AssetFileSource assetFileSource;
std::shared_ptr<SQLiteCache> cache;
OnlineFileSource onlineFileSource;
};
DefaultFileSource::DefaultFileSource(const std::string& cachePath, const std::string& assetRoot)
: impl(std::make_unique<DefaultFileSource::Impl>(cachePath, assetRoot)) {
}
DefaultFileSource::~DefaultFileSource() = default;
void DefaultFileSource::setAccessToken(const std::string& accessToken) {
impl->onlineFileSource.setAccessToken(accessToken);
}
std::string DefaultFileSource::getAccessToken() const {
return impl->onlineFileSource.getAccessToken();
}
void DefaultFileSource::setMaximumCacheSize(uint64_t size) {
impl->cache->setMaximumCacheSize(size);
}
void DefaultFileSource::setMaximumCacheEntrySize(uint64_t size) {
impl->cache->setMaximumCacheEntrySize(size);
}
SQLiteCache& DefaultFileSource::getCache() {
return *impl->cache;
}
class DefaultFileRequest : public FileRequest {
public:
DefaultFileRequest(Resource resource, FileSource::Callback callback, DefaultFileSource::Impl* impl) {
cacheRequest = impl->cache->get(resource, [=](std::shared_ptr<Response> cacheResponse) mutable {
cacheRequest.reset();
if (cacheResponse) {
resource.priorModified = cacheResponse->modified;
resource.priorExpires = cacheResponse->expires;
resource.priorEtag = cacheResponse->etag;
}
onlineRequest = impl->onlineFileSource.request(resource, [=] (Response onlineResponse) {
impl->cache->put(resource, onlineResponse);
callback(onlineResponse);
});
// Do this last because it may result in deleting this DefaultFileRequest.
if (cacheResponse) {
callback(*cacheResponse);
}
});
}
std::unique_ptr<WorkRequest> cacheRequest;
std::unique_ptr<FileRequest> onlineRequest;
};
std::unique_ptr<FileRequest> DefaultFileSource::request(const Resource& resource, Callback callback) {
if (isAssetURL(resource.url)) {
return impl->assetFileSource.request(resource, callback);
} else {
return std::make_unique<DefaultFileRequest>(resource, callback, impl.get());
}
}
} // namespace mbgl
|
