根据元素所属的类标记元素

Question

我有一个元素流 object_count.json 看起来像：

{
    "name":"apple",
    "count":10
}
{
    "name":"potato",
    "count":18
}
{
    "name":"stone",
    "count:7
}

与流分开，我有两个文件，每个文件都列出了其类对象的名称：

cat fruits.txt
  apple
  peach

cat vegetables.txt
  potato
  tomato

如何构造一个流，其中来自object_count.json 的每个对象都带有其各自的类名？即：

{
    "name":"apple",
    "count":10
    "class":"fruit"
}
{
    "name":"potato",
    "count":18
    "class":"vegetable"
}
{
    "name":"stone",
    "count":7
    "class":"other"
}

我们可以假设object_count.json 中的每个对象名称只有一个条目。

---

我最喜欢这样的解决方案：

• 适用于 3、4 和更多类 - 每个仍然列在单独的文件中；如果每个类都是硬编码的就可以了
• object_count.json 被视为输入流，而 {fruits|vegetables}.txt 分别作为单独的 --rawfile 参数提供

JQ 报告的版本：

 $ jq --version
 jq-1.6

---

我设法将类文件作为命令行参数提供，但未能找到合适的 join / in 表达式来解决问题：

cat object_count.json | jq -n --rawfile fruits "fruits.txt" '($fruits | split("\n") | map(select(. != "")) | sort) as $frts | inputs'

Answer 1

根据更新后的问题，这里有一个避免slurping object_count.json 内容的解决方案。

dictionary.jq

def trim: sub("^ +"; "") | sub(" +$"; "");

def dictionary(s):
  reduce (s | trim | select(length>0)) as $in ({};
     (input_filename | sub(".txt";"") | sub("s$";"")) as $class
     | .[$in] = $class );

dictionary(inputs) 

调用

jq --argfile class <(jq -n -R -f dictionary.jq *.txt)  '
 .class = ($class[.name] // "other")'  object_count.json

后记

可能需要使用“其他”作为默认类名，但如果不是，那么在这种情况下，最好顺其自然并使用 JSON 的 null 作为对象的默认值键。

Answer 2

[这是对原始问题的回应。它允许任意多个“类”文件，并且只需要一次调用 jq。对于不涉及“slurping”JSON 流的解决方案，请参阅本页其他地方。]

这里的棘手部分是管理未知数量的 .txt 文件。在这里，我们假设我们可以编写 '*.txt' 来捕获它们，因此 jq 的调用将如下所示：

jq -n -R -f program.jq --slurpfile input object_count.json *.txt

为了构建字典，我们将使用以下辅助函数：

def trim: sub("^ +"; "") | sub(" +$"; "");

# construct the dictionary
def dictionary(s):
  reduce (s | trim | select(length>0)) as $in ({};
     (input_filename | sub(".txt";"") | sub("s$";"")) as $class
     | .[$in] = $class );

现在任务减少到以下三行：

dictionary(inputs) as $class
| $input[]
| .class = ($class[.name] // "other")