mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(tvix): Wrap remaining RPCs in HandleExceptions

Browse source 
Wrap the BuildPaths and AddTextToStore RPC handlers in HandleExceptions.
These were missed in the original pass due to a merge.

Change-Id: Ie5be45e6098fba7a2b6b1c1be81578cb742c2880
Reviewed-on: https://cl.tvl.fyi/c/depot/+/1689
Tested-by: BuildkiteCI
Reviewed-by: kanepyork <rikingcoding@gmail.com>
This commit is contained in:
Griffin Smith 2020-08-08 14:30:55 -04:00 committed by glittershark
parent 3115113854
commit 053a138002
1 changed files with 29 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
third_party/nix/src/nix-daemon/nix-daemon-proto.cc vendored
View file

@ -256,35 +256,43 @@ class WorkerServiceImpl final : public WorkerService::Service {
Status AddTextToStore(grpc::ServerContext*,
const nix::proto::AddTextToStoreRequest* request,
nix::proto::StorePath* response) override {
PathSet references;
for (const auto& ref : request->references()) {
references.insert(ref);
}
auto path =
store_->addTextToStore(request->name(), request->content(), references);
response->set_path(path);
return Status::OK;
return HandleExceptions(
[&]() -> Status {
PathSet references;
for (const auto& ref : request->references()) {
references.insert(ref);
}
auto path = store_->addTextToStore(request->name(),
request->content(), references);
response->set_path(path);
return Status::OK;
},
__FUNCTION__);
}
Status BuildPaths(grpc::ServerContext*,
const nix::proto::BuildPathsRequest* request,
google::protobuf::Empty*) override {
PathSet drvs;
for (const auto& drv : request->drvs()) {
drvs.insert(drv);
}
auto mode = BuildModeFrom(request->mode());
return HandleExceptions(
[&]() -> Status {
PathSet drvs;
for (const auto& drv : request->drvs()) {
drvs.insert(drv);
}
auto mode = BuildModeFrom(request->mode());
if (!mode.has_value()) {
return Status(grpc::StatusCode::INTERNAL, "Invalid build mode");
}
if (!mode.has_value()) {
return Status(grpc::StatusCode::INTERNAL, "Invalid build mode");
}
// TODO(grfn): If mode is repair and not trusted, we need to return an error
// here (but we can't yet because we don't know anything about trusted
// users)
store_->buildPaths(drvs, mode.value());
// TODO(grfn): If mode is repair and not trusted, we need to return an
// error here (but we can't yet because we don't know anything about
// trusted users)
store_->buildPaths(drvs, mode.value());
return Status::OK;
return Status::OK;
},
__FUNCTION__);
}
Status AddTempRoot(grpc::ServerContext*, const nix::proto::StorePath* request,