使用jq工具在Bash脚本中构建多层级JSON

在Shell脚本编程中，处理JSON数据是一项常见任务。jq作为一款强大的命令行JSON处理器，能够帮助我们高效地构建和操作JSON数据结构。本文将通过一个实际案例，讲解如何使用jq在Bash脚本中动态构建包含数组的多层级JSON结构。

问题背景

开发者尝试从一个文本文件中读取数据，并将其转换为特定结构的JSON格式。核心需求包括：

1. 解析包含时间戳和站点信息的多行文本
2. 将时间戳转换为可读格式
3. 将站点信息组织为JSON数组
4. 动态地向数组中添加元素

初始实现分析

原始脚本采用了以下方法：

1. 使用readarray读取文件内容到数组
2. 遍历数组元素，通过正则匹配识别不同类型的数据
3. 对于时间戳，转换为日期格式并添加到JSON对象
4. 对于站点信息，尝试创建节点数组并添加MAC地址和接口信息

关键挑战

在实现过程中，开发者遇到了向JSON数组中动态添加元素的困难。具体表现为：

1. 直接使用变量作为数组索引（.nodes[$count]）无法正常工作
2. 尝试使用数组长度作为索引（.nodes[.nodes | length]）也未达到预期效果

解决方案

jq提供了多种向数组添加元素的方法，以下是几种有效的实现方式：

方法一：使用数组拼接运算符

JSON=$(echo $JSON | jq --arg mac "${mac}" '.nodes += [{"mac": $mac}]')

这种方法简洁明了，通过+=运算符将包含新元素的对象追加到数组末尾。

方法二：使用数组构造语法

JSON=$(echo $JSON | jq --arg mac "${mac}" '.nodes |= . + [{"mac": $mac}]')

这种语法使用|=更新运算符，将原数组与新构造的数组合并。

方法三：使用变量索引

如果需要显式使用计数变量作为索引，可以这样实现：

JSON=$(echo $JSON | jq --arg mac "${mac}" --argjson count "$count" '.nodes[.nodes|length] = {"mac": $mac}')

完整改进方案

结合上述方法，改进后的脚本核心部分如下：

#!/bin/bash

JSON=$(jq -n '{}')
readarray -t array < ./station.sample

for e in "${array[@]}"
do
    if echo "$e" | grep -Eq '^[0-9]{10}' >/dev/null; then
        timestamp=$e
        datetime=$(date +'%Y-%m-%d %H:%M:%S' -d "@$e")
        JSON=$(jq -n --argjson json "$JSON" \
            --arg timestamp "$timestamp" \
            --arg datetime "$datetime" \
            '$json + ($ARGS.named|del(.json))')
    fi

    if echo "$e" | grep '^Station ' >/dev/null; then
        mac=$(echo "$e" | awk '{print $2}')
        interface=$(echo "$e" | awk '{print $4}' | rev | cut -c2- | rev)
        
        JSON=$(jq -n --argjson json "$JSON" \
            --arg mac "$mac" \
            --arg interface "$interface" \
            '$json | (.nodes += [{"mac": $mac, "interface": $interface}])')
    fi
done

性能优化建议

1. 减少jq调用次数：每次调用jq都会产生进程创建开销，可以考虑将多次操作合并到单个jq命令中
2. 使用--argjson传递复杂数据结构
3. 对于大型数据集，考虑使用jq的流式处理功能

扩展应用

这种技术可以应用于多种场景：

1. 日志文件转JSON
2. 系统监控数据收集
3. 网络设备配置导出
4. API响应数据处理

通过掌握jq在Shell脚本中的高级用法，开发者可以更高效地处理复杂的JSON数据转换任务。