Skip to content

Commit

Permalink
ProgressBar: Fix updating
Browse files Browse the repository at this point in the history
'updateCV.notify_one()' does nothing if the update thread is not
waiting for updateCV (in particular this happens when it is sleeping
on quitCV). So also set a variable to ensure that the update isn't
lost.
  • Loading branch information
edolstra committed Jun 25, 2019
1 parent 8857121 commit 324a5dc
Showing 1 changed file with 15 additions and 11 deletions.
26 changes: 15 additions & 11 deletions src/nix/progress-bar.cc
Expand Up @@ -62,6 +62,7 @@ class ProgressBar : public Logger
uint64_t corruptedPaths = 0, untrustedPaths = 0;

bool active = true;
bool haveUpdate = true;
};

Sync<State> state_;
Expand All @@ -83,7 +84,8 @@ class ProgressBar : public Logger
updateThread = std::thread([&]() {
auto state(state_.lock());
while (state->active) {
state.wait(updateCV);
if (!state->haveUpdate)
state.wait(updateCV);
draw(*state);
state.wait_for(quitCV, std::chrono::milliseconds(50));
}
Expand Down Expand Up @@ -178,7 +180,7 @@ class ProgressBar : public Logger
|| (type == actCopyPath && hasAncestor(*state, actSubstitute, parent)))
i->visible = false;

update();
update(*state);
}

/* Check whether an activity has an ancestore with the specified
Expand Down Expand Up @@ -213,7 +215,7 @@ class ProgressBar : public Logger
state->its.erase(i);
}

update();
update(*state);
}

void result(ActivityId act, ResultType type, const std::vector<Field> & fields) override
Expand All @@ -223,7 +225,7 @@ class ProgressBar : public Logger
if (type == resFileLinked) {
state->filesLinked++;
state->bytesLinked += getI(fields, 0);
update();
update(*state);
}

else if (type == resBuildLogLine) {
Expand All @@ -239,26 +241,26 @@ class ProgressBar : public Logger
info.lastLine = lastLine;
state->activities.emplace_back(info);
i->second = std::prev(state->activities.end());
update();
update(*state);
}
}
}

else if (type == resUntrustedPath) {
state->untrustedPaths++;
update();
update(*state);
}

else if (type == resCorruptedPath) {
state->corruptedPaths++;
update();
update(*state);
}

else if (type == resSetPhase) {
auto i = state->its.find(act);
assert(i != state->its.end());
i->second->phase = getS(fields, 0);
update();
update(*state);
}

else if (type == resProgress) {
Expand All @@ -269,7 +271,7 @@ class ProgressBar : public Logger
actInfo.expected = getI(fields, 1);
actInfo.running = getI(fields, 2);
actInfo.failed = getI(fields, 3);
update();
update(*state);
}

else if (type == resSetExpected) {
Expand All @@ -281,17 +283,19 @@ class ProgressBar : public Logger
state->activitiesByType[type].expected -= j;
j = getI(fields, 1);
state->activitiesByType[type].expected += j;
update();
update(*state);
}
}

void update()
void update(State & state)
{
state.haveUpdate = true;
updateCV.notify_one();
}

void draw(State & state)
{
state.haveUpdate = false;
if (!state.active) return;

std::string line;
Expand Down

0 comments on commit 324a5dc

Please sign in to comment.