feat(3p/nix): Implement similar group of Worker::Query* handlers

These are the queries that are handled in the confusing case statement
in the old daemon implementation, because they have very similar
structure.

Change-Id: Ie7143354f66cef4336dff8072ede9a56271a7e89
Reviewed-on: https://cl.tvl.fyi/c/depot/+/1228
Tested-by: BuildkiteCI
Reviewed-by: glittershark <grfn@gws.fyi>
This commit is contained in:
Vincent Ambo 2020-07-17 00:11:48 +01:00 committed by tazjin
parent 32735eebdd
commit 5c9af8faee

View file

@ -6,6 +6,7 @@ namespace nix::daemon {
using ::grpc::Status; using ::grpc::Status;
using ::nix::proto::StorePath; using ::nix::proto::StorePath;
using ::nix::proto::StorePaths;
using ::nix::proto::Worker; using ::nix::proto::Worker;
class WorkerServiceImpl final : public Worker::Service { class WorkerServiceImpl final : public Worker::Service {
@ -13,7 +14,7 @@ class WorkerServiceImpl final : public Worker::Service {
WorkerServiceImpl(nix::Store* store) : store_(store) {} WorkerServiceImpl(nix::Store* store) : store_(store) {}
Status IsValidPath(grpc::ServerContext* context, const StorePath* request, Status IsValidPath(grpc::ServerContext* context, const StorePath* request,
nix::proto::IsValidPathResponse* response) { nix::proto::IsValidPathResponse* response) override {
const auto& path = request->path(); const auto& path = request->path();
store_->assertStorePath(path); store_->assertStorePath(path);
response->set_is_valid(store_->isValidPath(path)); response->set_is_valid(store_->isValidPath(path));
@ -22,7 +23,7 @@ class WorkerServiceImpl final : public Worker::Service {
} }
Status HasSubstitutes(grpc::ServerContext* context, const StorePath* request, Status HasSubstitutes(grpc::ServerContext* context, const StorePath* request,
nix::proto::HasSubstitutesResponse* response) { nix::proto::HasSubstitutesResponse* response) override {
const auto& path = request->path(); const auto& path = request->path();
store_->assertStorePath(path); store_->assertStorePath(path);
PathSet res = store_->querySubstitutablePaths({path}); PathSet res = store_->querySubstitutablePaths({path});
@ -31,6 +32,50 @@ class WorkerServiceImpl final : public Worker::Service {
return Status::OK; return Status::OK;
} }
Status QueryReferrers(grpc::ServerContext* context, const StorePath* request,
StorePaths* response) override {
const auto& path = request->path();
store_->assertStorePath(path);
PathSet paths;
store_->queryReferrers(path, paths);
for (const auto& path : paths) {
response->add_paths(path);
}
return Status::OK;
}
Status QueryValidDerivers(grpc::ServerContext* context,
const StorePath* request, StorePaths* response) {
const auto& path = request->path();
store_->assertStorePath(path);
PathSet paths = store_->queryValidDerivers(path);
for (const auto& path : paths) {
response->add_paths(path);
}
return Status::OK;
}
Status QueryDerivationOutputs(grpc::ServerContext* context,
const StorePath* request,
StorePaths* response) override {
const auto& path = request->path();
store_->assertStorePath(path);
PathSet paths = store_->queryDerivationOutputs(path);
for (const auto& path : paths) {
response->add_paths(path);
}
return Status::OK;
}
private: private:
// TODO(tazjin): Who owns the store? // TODO(tazjin): Who owns the store?
nix::Store* store_; nix::Store* store_;