delroth/nix-eval-jobs
0
0
Fork 0
forked from lix-project/nix-eval-jobs
Code Pull requests Activity

Switch to streaming output (using line delimited JSON)

Browse source
This commit is contained in:
adisbladis 2021-08-25 12:43:35 -05:00
parent 0c93e00635
commit a6171c9a62
No known key found for this signature in database
GPG key ID: 110BFAD44C6249B7
2 changed files with 17 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/nix-eval-jobs.cc
View file

@ -156,6 +156,7 @@ static void worker(
/* Evaluate it and send info back to the master. */
nlohmann::json reply;
reply["attr"] = attrPath;
try {
auto vTmp = findAlongAttrPath(state, attrPath, autoArgs, *vRoot).first;
@ -215,7 +216,8 @@ static void worker(
reply["job"] = std::move(job);
}
else if (v->type() == nAttrs) {
else if (v->type() == nAttrs)
{
auto attrs = nlohmann::json::array();
StringSet ss;
for (auto & i : v->attrs->lexicographicOrder()) {
@ -287,7 +289,6 @@ int main(int argc, char * * argv)
{
std::set<std::string> todo{""};
std::set<std::string> active;
nlohmann::json jobs;
std::exception_ptr exc;
};
@ -369,26 +370,19 @@ int main(int argc, char * * argv)
writeLine(to.get(), "do " + attrPath);
/* Wait for the response. */
auto response = nlohmann::json::parse(readLine(from.get()));
auto respString = readLine(from.get());
auto response = nlohmann::json::parse(respString);
/* Handle the response. */
StringSet newAttrs;
if (response.find("job") != response.end()) {
auto state(state_.lock());
state->jobs[attrPath] = response["job"];
}
if (response.find("attrs") != response.end()) {
for (auto & i : response["attrs"]) {
auto s = (attrPath.empty() ? "" : attrPath + ".") + (std::string) i;
newAttrs.insert(s);
}
}
if (response.find("error") != response.end()) {
} else {
auto state(state_.lock());
state->jobs[attrPath]["error"] = response["error"];
std::cout << respString << "\n";
}
/* Add newly discovered job names to the queue. */
@ -419,6 +413,5 @@ int main(int argc, char * * argv)
if (state->exc)
std::rethrow_exception(state->exc);
std::cout << state->jobs.dump(2) << "\n";
});
}

14
tests/test_eval.py
View file

@ -21,12 +21,18 @@ def common_test(extra_args: List[str]) -> None:
check=True,
stdout=subprocess.PIPE,
)
data = json.loads(res.stdout)
assert data["builtJob"]["nixName"] == "job1"
assert "out" in data["builtJob"]["outputs"]
results = [json.loads(r) for r in res.stdout.split("\n") if r]
assert len(results) == 2
assert data["substitutedJob"]["nixName"].startswith("hello-")
built_job = results[0]
assert built_job["attr"] == "builtJob"
assert built_job["job"]["nixName"] == "job1"
substituted_job = results[1]
assert substituted_job["attr"] == "substitutedJob"
assert substituted_job["job"]["nixName"].startswith("hello-")
def test_flake() -> None: