mirror of
https://github.com/NixOS/nix
synced 2025-07-06 00:51:47 +02:00
d972f9e2e2
The existing header is a bit too big. Now the following use-cases are separated, and get their own headers: - Using or implementing an arbitrary store: remaining `store-api.hh` This is closer to just being about the `Store` (and `StoreConfig`) classes, as one would expect. - Opening a store from a textual description: `store-open.hh` Opening an aribtrary store implementation like this requires some sort of store registration mechanism to exists, but the caller doesn't need to know how it works. This just exposes the functions which use such a mechanism, without exposing the mechanism itself - Registering a store implementation: `store-registration.hh` This requires understanding how the mechanism actually works, and the mechanism in question involves templated machinery in headers we rather not expose to things that don't need it, as it would slow down compilation for no reason.
137 lines
3.5 KiB
C++
137 lines
3.5 KiB
C++
#include "nix/store/local-binary-cache-store.hh"
|
|
#include "nix/store/globals.hh"
|
|
#include "nix/store/nar-info-disk-cache.hh"
|
|
#include "nix/util/signals.hh"
|
|
#include "nix/store/store-registration.hh"
|
|
|
|
#include <atomic>
|
|
|
|
namespace nix {
|
|
|
|
LocalBinaryCacheStoreConfig::LocalBinaryCacheStoreConfig(
|
|
std::string_view scheme,
|
|
PathView binaryCacheDir,
|
|
const StoreReference::Params & params)
|
|
: Store::Config{params}
|
|
, BinaryCacheStoreConfig{params}
|
|
, binaryCacheDir(binaryCacheDir)
|
|
{
|
|
}
|
|
|
|
|
|
std::string LocalBinaryCacheStoreConfig::doc()
|
|
{
|
|
return
|
|
#include "local-binary-cache-store.md"
|
|
;
|
|
}
|
|
|
|
|
|
struct LocalBinaryCacheStore :
|
|
virtual BinaryCacheStore
|
|
{
|
|
using Config = LocalBinaryCacheStoreConfig;
|
|
|
|
ref<Config> config;
|
|
|
|
LocalBinaryCacheStore(ref<Config> config)
|
|
: Store{*config}
|
|
, BinaryCacheStore{*config}
|
|
, config{config}
|
|
{
|
|
init();
|
|
}
|
|
|
|
void init() override;
|
|
|
|
std::string getUri() override
|
|
{
|
|
return "file://" + config->binaryCacheDir;
|
|
}
|
|
|
|
protected:
|
|
|
|
bool fileExists(const std::string & path) override;
|
|
|
|
void upsertFile(const std::string & path,
|
|
std::shared_ptr<std::basic_iostream<char>> istream,
|
|
const std::string & mimeType) override
|
|
{
|
|
auto path2 = config->binaryCacheDir + "/" + path;
|
|
static std::atomic<int> counter{0};
|
|
Path tmp = fmt("%s.tmp.%d.%d", path2, getpid(), ++counter);
|
|
AutoDelete del(tmp, false);
|
|
StreamToSourceAdapter source(istream);
|
|
writeFile(tmp, source);
|
|
std::filesystem::rename(tmp, path2);
|
|
del.cancel();
|
|
}
|
|
|
|
void getFile(const std::string & path, Sink & sink) override
|
|
{
|
|
try {
|
|
readFile(config->binaryCacheDir + "/" + path, sink);
|
|
} catch (SysError & e) {
|
|
if (e.errNo == ENOENT)
|
|
throw NoSuchBinaryCacheFile("file '%s' does not exist in binary cache", path);
|
|
throw;
|
|
}
|
|
}
|
|
|
|
StorePathSet queryAllValidPaths() override
|
|
{
|
|
StorePathSet paths;
|
|
|
|
for (auto & entry : DirectoryIterator{config->binaryCacheDir}) {
|
|
checkInterrupt();
|
|
auto name = entry.path().filename().string();
|
|
if (name.size() != 40 ||
|
|
!hasSuffix(name, ".narinfo"))
|
|
continue;
|
|
paths.insert(parseStorePath(
|
|
storeDir + "/" + name.substr(0, name.size() - 8)
|
|
+ "-" + MissingName));
|
|
}
|
|
|
|
return paths;
|
|
}
|
|
|
|
std::optional<TrustedFlag> isTrustedClient() override
|
|
{
|
|
return Trusted;
|
|
}
|
|
};
|
|
|
|
void LocalBinaryCacheStore::init()
|
|
{
|
|
createDirs(config->binaryCacheDir + "/nar");
|
|
createDirs(config->binaryCacheDir + "/" + realisationsPrefix);
|
|
if (config->writeDebugInfo)
|
|
createDirs(config->binaryCacheDir + "/debuginfo");
|
|
createDirs(config->binaryCacheDir + "/log");
|
|
BinaryCacheStore::init();
|
|
}
|
|
|
|
bool LocalBinaryCacheStore::fileExists(const std::string & path)
|
|
{
|
|
return pathExists(config->binaryCacheDir + "/" + path);
|
|
}
|
|
|
|
StringSet LocalBinaryCacheStoreConfig::uriSchemes()
|
|
{
|
|
if (getEnv("_NIX_FORCE_HTTP") == "1")
|
|
return {};
|
|
else
|
|
return {"file"};
|
|
}
|
|
|
|
ref<Store> LocalBinaryCacheStoreConfig::openStore() const {
|
|
return make_ref<LocalBinaryCacheStore>(ref{
|
|
// FIXME we shouldn't actually need a mutable config
|
|
std::const_pointer_cast<LocalBinaryCacheStore::Config>(shared_from_this())
|
|
});
|
|
}
|
|
|
|
static RegisterStoreImplementation<LocalBinaryCacheStore::Config> regLocalBinaryCacheStore;
|
|
|
|
}
|