CUE语言中如何在列表推导式中使用外部作用域变量？

问题描述 #

在CUE-lang中，尝试在列表推导式内引用外部作用域的布尔变量parallelize时，会触发如下错误：

experiments.0.setup.0.cpus: non-concrete value bool in operand to ==:

即使已经在外部作用域明确赋值parallelize: true，但直接将参数替换为字面量{parallelize: true}则能正常运行。复现代码如下：

#GetCPUSet: {
    parallelize: bool

    if parallelize == true {
        cpuSet: "1-2"
    }
    if parallelize == false {
        cpuSet: "3-4"
    }
}

_experimentInput: "test_A1,test_A2|test_B2"
_experiments:      strings.Split(_experimentInput, "|")

parallelize: true

experiments: [
    for i, l in _experiments {
        tests: "\(l)"
        setup: [{
            _cpuset: #GetCPUSet & {parallelize: parallelize}
            cpus:    _cpuset
        }]
    },
]

原因分析 #

CUE的列表推导式在求值阶段，内部对外部变量的引用会被延迟解析。这导致#GetCPUSet中的条件判断无法及时获取到parallelize的具体值——尽管外部已经赋值，但在推导式内部该变量仍处于未完全具体化的状态，无法用于等式运算。

解决方案 #

方法一：绑定外部变量到推导式局部变量 #

在推导式内部显式将外部变量赋值给局部变量，确保其在推导式上下文内是具体值：

experiments: [
    for i, l in _experiments {
        localParallelize: parallelize
        tests: "\(l)"
        setup: [{
            _cpuset: #GetCPUSet & {parallelize: localParallelize}
            cpus:    _cpuset
        }]
    },
]

方法二：提前具体化模板实例 #

在列表推导式外部预先创建#GetCPUSet的具体实例，推导式内直接引用该实例：

_cpusetTemplate: #GetCPUSet & {parallelize: parallelize}

experiments: [
    for i, l in _experiments {
        tests: "\(l)"
        setup: [{
            cpus:    _cpusetTemplate
        }]
    },
]

方法三：简化条件逻辑写法 #

使用CUE的单条件表达式简化#GetCPUSet的定义，减少作用域解析带来的问题：

#GetCPUSet: {
    parallelize: bool
    cpuSet: if parallelize { "1-2" } else { "3-4" }
}

内容的提问来源于stack exchange，提问作者Augusto