#pragma once ///@file #include #include #include "store-api.hh" #include "gc-store.hh" #include "log-store.hh" namespace nix { class Pipe; class Pid; struct FdSink; struct FdSource; template class Pool; struct RemoteStoreConfig : virtual StoreConfig { using StoreConfig::StoreConfig; const Setting maxConnections{this, 1, "max-connections", "Maximum number of concurrent connections to the Nix daemon."}; const Setting maxConnectionAge{this, std::numeric_limits::max(), "max-connection-age", "Maximum age of a connection before it is closed."}; }; /** * \todo RemoteStore is a misnomer - should be something like * DaemonStore. */ class RemoteStore : public virtual RemoteStoreConfig, public virtual Store, public virtual GcStore, public virtual LogStore { public: RemoteStore(const Params & params); /* Implementations of abstract store API methods. */ bool isValidPathUncached(const StorePath & path) override; StorePathSet queryValidPaths(const StorePathSet & paths, SubstituteFlag maybeSubstitute = NoSubstitute) override; StorePathSet queryAllValidPaths() override; std::shared_ptr queryPathInfoUncached(const StorePath & path) override; void queryReferrers(const StorePath & path, StorePathSet & referrers) override; StorePathSet queryValidDerivers(const StorePath & path) override; StorePathSet queryDerivationOutputs(const StorePath & path) override; std::map> queryPartialDerivationOutputMap(const StorePath & path, Store * evalStore = nullptr) override; std::optional queryPathFromHashPart(const std::string & hashPart) override; StorePathSet querySubstitutablePaths(const StorePathSet & paths) override; void querySubstitutablePathInfos(const StorePathCAMap & paths, SubstitutablePathInfos & infos) override; /** * Add a content-addressable store path. `dump` will be drained. */ ref addCAToStore( Source & dump, std::string_view name, ContentAddressMethod caMethod, HashType hashType, const StorePathSet & references, RepairFlag repair); /** * Add a content-addressable store path. Does not support references. `dump` will be drained. */ StorePath addToStoreFromDump(Source & dump, std::string_view name, FileIngestionMethod method = FileIngestionMethod::Recursive, HashType hashAlgo = htSHA256, RepairFlag repair = NoRepair, const StorePathSet & references = StorePathSet()) override; void addToStore(const ValidPathInfo & info, Source & nar, RepairFlag repair, CheckSigsFlag checkSigs) override; void addMultipleToStore( Source & source, RepairFlag repair, CheckSigsFlag checkSigs) override; void addMultipleToStore( PathsSource & pathsToCopy, Activity & act, RepairFlag repair, CheckSigsFlag checkSigs) override; StorePath addTextToStore( std::string_view name, std::string_view s, const StorePathSet & references, RepairFlag repair) override; void registerDrvOutput(const Realisation & info) override; std::shared_ptr queryRealisationUncached(const DrvOutput &) override; void buildPaths(const std::vector & paths, BuildMode buildMode, std::shared_ptr evalStore) override; std::vector buildPathsWithResults( const std::vector & paths, BuildMode buildMode, std::shared_ptr evalStore) override; BuildResult buildDerivation(const StorePath & drvPath, const BasicDerivation & drv, BuildMode buildMode) override; void ensurePath(const StorePath & path) override; void addTempRoot(const StorePath & path) override; Roots findRoots(bool censor) override; void collectGarbage(const GCOptions & options, GCResults & results) override; void optimiseStore() override; bool verifyStore(bool checkContents, RepairFlag repair) override; /** * The default instance would schedule the work on the client side, but * for consistency with `buildPaths` and `buildDerivation` it should happen * on the remote side. * * We make this fail for now so we can add implement this properly later * without it being a breaking change. */ void repairPath(const StorePath & path) override { unsupported("repairPath"); } void addSignatures(const StorePath & storePath, const StringSet & sigs) override; void queryMissing(const std::vector & targets, StorePathSet & willBuild, StorePathSet & willSubstitute, StorePathSet & unknown, uint64_t & downloadSize, uint64_t & narSize) override; void addBuildLog(const StorePath & drvPath, std::string_view log) override; std::optional getVersion() override; void connect() override; unsigned int getProtocol() override; std::optional isTrustedClient() override; struct Connection; ref openConnectionWrapper(); protected: virtual ref openConnection() = 0; void initConnection(Connection & conn); ref> connections; virtual void setOptions(Connection & conn); void setOptions() override; struct ConnectionHandle; ConnectionHandle getConnection(); friend struct ConnectionHandle; virtual ref getFSAccessor() override; virtual void narFromPath(const StorePath & path, Sink & sink) override; private: std::atomic_bool failed{false}; void copyDrvsFromEvalStore( const std::vector & paths, std::shared_ptr evalStore); }; }