mirror of
https://github.com/NixOS/nix
synced 2025-06-29 19:03:16 +02:00
Add "nix verify-paths" command
Unlike "nix-store --verify-path", this command verifies signatures in addition to store path contents, is multi-threaded (especially useful when verifying binary caches), and has a progress indicator. Example use: $ nix verify-paths --store https://cache.nixos.org -r $(type -p thunderbird) ... [17/132 checked] checking ‘/nix/store/rawakphadqrqxr6zri2rmnxh03gqkrl3-autogen-5.18.6’
This commit is contained in:
parent
0ebe69dc67
commit
784ee35c80
11 changed files with 432 additions and 2 deletions
81
src/libutil/thread-pool.cc
Normal file
81
src/libutil/thread-pool.cc
Normal file
|
@ -0,0 +1,81 @@
|
|||
#include "thread-pool.hh"
|
||||
|
||||
namespace nix {
|
||||
|
||||
ThreadPool::ThreadPool(size_t _nrThreads)
|
||||
: nrThreads(_nrThreads)
|
||||
{
|
||||
if (!nrThreads) {
|
||||
nrThreads = std::thread::hardware_concurrency();
|
||||
if (!nrThreads) nrThreads = 1;
|
||||
}
|
||||
}
|
||||
|
||||
void ThreadPool::enqueue(const work_t & t)
|
||||
{
|
||||
auto state_(state.lock());
|
||||
state_->left.push(t);
|
||||
wakeup.notify_one();
|
||||
}
|
||||
|
||||
void ThreadPool::process()
|
||||
{
|
||||
printMsg(lvlDebug, format("starting pool of %d threads") % nrThreads);
|
||||
|
||||
std::vector<std::thread> workers;
|
||||
|
||||
for (size_t n = 0; n < nrThreads; n++)
|
||||
workers.push_back(std::thread([&]() {
|
||||
bool first = true;
|
||||
|
||||
while (true) {
|
||||
work_t work;
|
||||
{
|
||||
auto state_(state.lock());
|
||||
if (state_->exception) return;
|
||||
if (!first) {
|
||||
assert(state_->pending);
|
||||
state_->pending--;
|
||||
}
|
||||
first = false;
|
||||
while (state_->left.empty()) {
|
||||
if (!state_->pending) {
|
||||
wakeup.notify_all();
|
||||
return;
|
||||
}
|
||||
if (state_->exception) return;
|
||||
state_.wait(wakeup);
|
||||
}
|
||||
work = state_->left.front();
|
||||
state_->left.pop();
|
||||
state_->pending++;
|
||||
}
|
||||
|
||||
try {
|
||||
work();
|
||||
} catch (std::exception & e) {
|
||||
auto state_(state.lock());
|
||||
if (state_->exception)
|
||||
printMsg(lvlError, format("error: %s") % e.what());
|
||||
else {
|
||||
state_->exception = std::current_exception();
|
||||
wakeup.notify_all();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}));
|
||||
|
||||
for (auto & thr : workers)
|
||||
thr.join();
|
||||
|
||||
{
|
||||
auto state_(state.lock());
|
||||
if (state_->exception)
|
||||
std::rethrow_exception(state_->exception);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
Loading…
Add table
Add a link
Reference in a new issue