forked from lix-project/lix
d1b0909894
This allows readFile() to indicate that a file doesn't exist, and might eliminate some large string copying.
89 lines
2.5 KiB
C++
89 lines
2.5 KiB
C++
#include "binary-cache-store.hh"
|
||
#include "download.hh"
|
||
#include "globals.hh"
|
||
|
||
namespace nix {
|
||
|
||
class HttpBinaryCacheStore : public BinaryCacheStore
|
||
{
|
||
private:
|
||
|
||
Path cacheUri;
|
||
|
||
Pool<Downloader> downloaders;
|
||
|
||
public:
|
||
|
||
HttpBinaryCacheStore(std::shared_ptr<Store> localStore,
|
||
const Path & secretKeyFile, const Path & _cacheUri)
|
||
: BinaryCacheStore(localStore, secretKeyFile)
|
||
, cacheUri(_cacheUri)
|
||
, downloaders(
|
||
std::numeric_limits<size_t>::max(),
|
||
[]() { return makeDownloader(); })
|
||
{
|
||
if (cacheUri.back() == '/')
|
||
cacheUri.pop_back();
|
||
}
|
||
|
||
void init() override
|
||
{
|
||
// FIXME: do this lazily?
|
||
if (!fileExists("nix-cache-info"))
|
||
throw Error(format("‘%s’ does not appear to be a binary cache") % cacheUri);
|
||
}
|
||
|
||
protected:
|
||
|
||
bool fileExists(const std::string & path) override
|
||
{
|
||
try {
|
||
auto downloader(downloaders.get());
|
||
DownloadOptions options;
|
||
options.showProgress = DownloadOptions::no;
|
||
options.head = true;
|
||
downloader->download(cacheUri + "/" + path, options);
|
||
return true;
|
||
} catch (DownloadError & e) {
|
||
/* S3 buckets return 403 if a file doesn't exist and the
|
||
bucket is unlistable, so treat 403 as 404. */
|
||
if (e.error == Downloader::NotFound || e.error == Downloader::Forbidden)
|
||
return false;
|
||
throw;
|
||
}
|
||
}
|
||
|
||
void upsertFile(const std::string & path, const std::string & data) override
|
||
{
|
||
throw Error("uploading to an HTTP binary cache is not supported");
|
||
}
|
||
|
||
std::shared_ptr<std::string> getFile(const std::string & path) override
|
||
{
|
||
auto downloader(downloaders.get());
|
||
DownloadOptions options;
|
||
options.showProgress = DownloadOptions::no;
|
||
try {
|
||
return downloader->download(cacheUri + "/" + path, options).data;
|
||
} catch (DownloadError & e) {
|
||
if (e.error == Downloader::NotFound || e.error == Downloader::Forbidden)
|
||
return 0;
|
||
throw;
|
||
}
|
||
}
|
||
|
||
};
|
||
|
||
static RegisterStoreImplementation regStore([](const std::string & uri) -> std::shared_ptr<Store> {
|
||
if (std::string(uri, 0, 7) != "http://" &&
|
||
std::string(uri, 0, 8) != "https://") return 0;
|
||
auto store = std::make_shared<HttpBinaryCacheStore>(std::shared_ptr<Store>(0),
|
||
settings.get("binary-cache-secret-key-file", string("")),
|
||
uri);
|
||
store->init();
|
||
return store;
|
||
});
|
||
|
||
}
|
||
|