forked from lix-project/nix-eval-jobs
Replace internal attr string representation with an array
This ensures correct handling of attrnames with a dot in them and will not throw errors about illegal attrnames. Additionally this escapes any attributes containing dots in the JSON output and adds another field called `attrPath` which contains the attribute path as a list. Example output: ``` { "attr": "hello", "attrPath": [ "hello" ], "drvPath": "/nix/store/n204jib73z55cp9s0rmw1c5v5q528j7v-hello-2.12.drv", "name": "hello-2.12", "outputs": { "out": "/nix/store/h59dfk7dwrn7d2csykh9z9xm2miqmrnz-hello-2.12" }, "system": "x86_64-linux" } ```
This commit is contained in:
parent
2e58354a7e
commit
61c9f4cfcc
|
@ -28,6 +28,7 @@
|
||||||
#include <nlohmann/json.hpp>
|
#include <nlohmann/json.hpp>
|
||||||
|
|
||||||
using namespace nix;
|
using namespace nix;
|
||||||
|
using namespace nlohmann;
|
||||||
|
|
||||||
typedef enum { evalAuto, evalImpure, evalPure } pureEval;
|
typedef enum { evalAuto, evalImpure, evalPure } pureEval;
|
||||||
|
|
||||||
|
@ -232,6 +233,17 @@ static void to_json(nlohmann::json & json, const Drv & drv) {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::string attrPathJoin(json input) {
|
||||||
|
return std::accumulate(input.begin(), input.end(), std::string(),
|
||||||
|
[](std::string ss, std::string s) {
|
||||||
|
// Escape token if containing dots
|
||||||
|
if (s.find(".") != std::string::npos) {
|
||||||
|
s = "\"" + s + "\"";
|
||||||
|
}
|
||||||
|
return ss.empty() ? s : ss + "." + s;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
static void worker(
|
static void worker(
|
||||||
EvalState & state,
|
EvalState & state,
|
||||||
Bindings & autoArgs,
|
Bindings & autoArgs,
|
||||||
|
@ -247,14 +259,15 @@ static void worker(
|
||||||
auto s = readLine(from.get());
|
auto s = readLine(from.get());
|
||||||
if (s == "exit") break;
|
if (s == "exit") break;
|
||||||
if (!hasPrefix(s, "do ")) abort();
|
if (!hasPrefix(s, "do ")) abort();
|
||||||
std::string attrPath(s, 3);
|
auto path = json::parse(s.substr(3));
|
||||||
|
auto attrPathS = attrPathJoin(path);
|
||||||
|
|
||||||
debug("worker process %d at '%s'", getpid(), attrPath);
|
debug("worker process %d at '%s'", getpid(), path);
|
||||||
|
|
||||||
/* Evaluate it and send info back to the collector. */
|
/* Evaluate it and send info back to the collector. */
|
||||||
nlohmann::json reply = nlohmann::json{ { "attr", attrPath } };
|
json reply = json{ {"attr", attrPathS }, {"attrPath", path} };
|
||||||
try {
|
try {
|
||||||
auto vTmp = findAlongAttrPath(state, attrPath, autoArgs, *vRoot).first;
|
auto vTmp = findAlongAttrPath(state, attrPathS, autoArgs, *vRoot).first;
|
||||||
|
|
||||||
auto v = state.allocValue();
|
auto v = state.allocValue();
|
||||||
state.autoCallFunction(autoArgs, *vTmp, *v);
|
state.autoCallFunction(autoArgs, *vTmp, *v);
|
||||||
|
@ -281,14 +294,10 @@ static void worker(
|
||||||
else if (v->type() == nAttrs)
|
else if (v->type() == nAttrs)
|
||||||
{
|
{
|
||||||
auto attrs = nlohmann::json::array();
|
auto attrs = nlohmann::json::array();
|
||||||
bool recurse = attrPath == ""; // Dont require `recurseForDerivations = true;` for top-level attrset
|
bool recurse = path.size() == 0; // Dont require `recurseForDerivations = true;` for top-level attrset
|
||||||
|
|
||||||
for (auto & i : v->attrs->lexicographicOrder()) {
|
for (auto & i : v->attrs->lexicographicOrder()) {
|
||||||
std::string name(i->name);
|
std::string name(i->name);
|
||||||
if (name.find('.') != std::string::npos || name.find(' ') != std::string::npos) {
|
|
||||||
printError("skipping job with illegal name '%s'", name);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
attrs.push_back(name);
|
attrs.push_back(name);
|
||||||
|
|
||||||
if (name == "recurseForDerivations") {
|
if (name == "recurseForDerivations") {
|
||||||
|
@ -305,7 +314,7 @@ static void worker(
|
||||||
else if (v->type() == nNull)
|
else if (v->type() == nNull)
|
||||||
;
|
;
|
||||||
|
|
||||||
else throw TypeError("attribute '%s' is %s, which is not supported", attrPath, showType(*v));
|
else throw TypeError("attribute '%s' is %s, which is not supported", path, showType(*v));
|
||||||
|
|
||||||
} catch (EvalError & e) {
|
} catch (EvalError & e) {
|
||||||
auto err = e.info();
|
auto err = e.info();
|
||||||
|
@ -380,9 +389,9 @@ struct Proc {
|
||||||
|
|
||||||
struct State
|
struct State
|
||||||
{
|
{
|
||||||
std::set<std::string> todo{""};
|
std::set<json> todo = json::array({ json::array() });
|
||||||
std::set<std::string> active;
|
std::set<json> active;
|
||||||
std::exception_ptr exc;
|
std::exception_ptr exc;
|
||||||
};
|
};
|
||||||
|
|
||||||
std::function<void()> collector(Sync<State> & state_, std::condition_variable & wakeup) {
|
std::function<void()> collector(Sync<State> & state_, std::condition_variable & wakeup) {
|
||||||
|
@ -402,12 +411,12 @@ std::function<void()> collector(Sync<State> & state_, std::condition_variable &
|
||||||
proc_ = std::nullopt;
|
proc_ = std::nullopt;
|
||||||
continue;
|
continue;
|
||||||
} else if (s != "next") {
|
} else if (s != "next") {
|
||||||
auto json = nlohmann::json::parse(s);
|
auto json = json::parse(s);
|
||||||
throw Error("worker error: %s", (std::string) json["error"]);
|
throw Error("worker error: %s", (std::string) json["error"]);
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Wait for a job name to become available. */
|
/* Wait for a job name to become available. */
|
||||||
std::string attrPath;
|
json attrPath;
|
||||||
|
|
||||||
while (true) {
|
while (true) {
|
||||||
checkInterrupt();
|
checkInterrupt();
|
||||||
|
@ -426,18 +435,19 @@ std::function<void()> collector(Sync<State> & state_, std::condition_variable &
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Tell the worker to evaluate it. */
|
/* Tell the worker to evaluate it. */
|
||||||
writeLine(proc->to.get(), "do " + attrPath);
|
writeLine(proc->to.get(), "do " + attrPath.dump());
|
||||||
|
|
||||||
/* Wait for the response. */
|
/* Wait for the response. */
|
||||||
auto respString = readLine(proc->from.get());
|
auto respString = readLine(proc->from.get());
|
||||||
auto response = nlohmann::json::parse(respString);
|
auto response = json::parse(respString);
|
||||||
|
|
||||||
/* Handle the response. */
|
/* Handle the response. */
|
||||||
StringSet newAttrs;
|
std::vector<json> newAttrs;
|
||||||
if (response.find("attrs") != response.end()) {
|
if (response.find("attrs") != response.end()) {
|
||||||
for (auto & i : response["attrs"]) {
|
for (auto & i : response["attrs"]) {
|
||||||
auto s = (attrPath.empty() ? "" : attrPath + ".") + (std::string) i;
|
json newAttr = json(response["attrPath"]);
|
||||||
newAttrs.insert(s);
|
newAttr.emplace_back(i);
|
||||||
|
newAttrs.push_back(newAttr);
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
auto state(state_.lock());
|
auto state(state_.lock());
|
||||||
|
@ -450,8 +460,9 @@ std::function<void()> collector(Sync<State> & state_, std::condition_variable &
|
||||||
{
|
{
|
||||||
auto state(state_.lock());
|
auto state(state_.lock());
|
||||||
state->active.erase(attrPath);
|
state->active.erase(attrPath);
|
||||||
for (auto & s : newAttrs)
|
for (auto p : newAttrs) {
|
||||||
state->todo.insert(s);
|
state->todo.insert(p);
|
||||||
|
}
|
||||||
wakeup.notify_all();
|
wakeup.notify_all();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -31,4 +31,6 @@
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
|
"dotted.attr" = pkgs.hello;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -23,7 +23,7 @@ def common_test(extra_args: List[str]) -> None:
|
||||||
)
|
)
|
||||||
|
|
||||||
results = [json.loads(r) for r in res.stdout.split("\n") if r]
|
results = [json.loads(r) for r in res.stdout.split("\n") if r]
|
||||||
assert len(results) == 4
|
assert len(results) == 5
|
||||||
|
|
||||||
built_job = results[0]
|
built_job = results[0]
|
||||||
assert built_job["attr"] == "builtJob"
|
assert built_job["attr"] == "builtJob"
|
||||||
|
@ -32,14 +32,18 @@ def common_test(extra_args: List[str]) -> None:
|
||||||
assert built_job["drvPath"].endswith(".drv")
|
assert built_job["drvPath"].endswith(".drv")
|
||||||
assert built_job["meta"]['broken'] is False
|
assert built_job["meta"]['broken'] is False
|
||||||
|
|
||||||
recurse_drv = results[1]
|
dotted_job = results[1]
|
||||||
|
assert dotted_job["attr"] == "\"dotted.attr\""
|
||||||
|
assert dotted_job["attrPath"] == [ "dotted.attr" ]
|
||||||
|
|
||||||
|
recurse_drv = results[2]
|
||||||
assert recurse_drv["attr"] == "recurse.drvB"
|
assert recurse_drv["attr"] == "recurse.drvB"
|
||||||
assert recurse_drv["name"] == "drvB"
|
assert recurse_drv["name"] == "drvB"
|
||||||
|
|
||||||
recurse_recurse_bool = results[2]
|
recurse_recurse_bool = results[3]
|
||||||
assert "error" in recurse_recurse_bool
|
assert "error" in recurse_recurse_bool
|
||||||
|
|
||||||
substituted_job = results[3]
|
substituted_job = results[4]
|
||||||
assert substituted_job["attr"] == "substitutedJob"
|
assert substituted_job["attr"] == "substitutedJob"
|
||||||
assert substituted_job["name"].startswith("hello-")
|
assert substituted_job["name"].startswith("hello-")
|
||||||
assert substituted_job["meta"]['broken'] is False
|
assert substituted_job["meta"]['broken'] is False
|
||||||
|
|
Loading…
Reference in a new issue