如何基于用户参数（单值/列表）筛选Nextflow通道数据

问题背景 #

我有一个Nextflow通道first_ch，输出内容如下：

[[A, B, C, D], 1]
[[A, B, C, D], 2]
[[A, B, C], 3]
[[A, B, V], 4]
[[A], 5]
[[A1, B1, D], 7]
[[A1, B1, D], 8]

同时有一个用户自定义参数target_ids，它可以是单个数字（比如4）或者数字列表（比如[4,7,8]）。需要根据这个参数对first_ch做子集化，保留第二个元素和参数匹配的条目，再传递给下游流程。比如输入4时，只保留[[A, B, V], 4]；输入[7,8]时，保留[[A1, B1, D], 7]和[[A1, B1, D], 8]。

尝试过的方法 #

我试过用combine、filter和branch算子，但没得到预期结果，尝试的代码如下：

first_ch.combine(par_ch)
    .branch{ a, b, c, d  ->
        controls: b == d
            return [a, b]
        others: true
            return [a, b]
            }
    .view { }

硬编码值的方式是可行的：

first_ch.filter{ it[1] == '4' }

但把值作为参数或通道传递时，就得不到预期结果。

解决方案 #

方案1：直接使用参数（非通道形式） #

如果参数直接在params中定义，无需转成通道，直接在filter里统一处理单个值和列表的情况即可：

// 定义目标参数，支持单个值或列表
params.target_ids = 4  // 也可以设为 [4,7,8]

// 示例原通道
first_ch = Channel.of(
    [['A','B','C','D'], 1],
    [['A','B','C','D'], 2],
    [['A','B','C'], 3],
    [['A','B','V'], 4],
    [['A'], 5],
    [['A1','B1','D'], 7],
    [['A1','B1','D'], 8]
)

// 子集化处理
filtered_ch = first_ch.filter { entry ->
    // 将参数转为集合，统一适配单个值和列表
    def targets = params.target_ids instanceof Collection ? params.target_ids : [params.target_ids]
    entry[1] in targets
}

// 查看结果
filtered_ch.view()

方案2：参数作为通道传递 #

如果参数必须以通道形式存在（比如par_ch），先把通道内的目标值收集为一个集合，再用该集合过滤原通道：

// 目标参数通道，支持单个值或多个值
par_ch = Channel.of(4)  // 也可以设为 Channel.of(7,8)

// 示例原通道
first_ch = Channel.of(
    [['A','B','C','D'], 1],
    [['A','B','C','D'], 2],
    [['A','B','C'], 3],
    [['A','B','V'], 4],
    [['A'], 5],
    [['A1','B1','D'], 7],
    [['A1','B1','D'], 8]
)

// 收集通道内的所有目标值为一个集合
targets_set = par_ch.collect()

// 结合原通道与目标集合，过滤匹配条目
filtered_ch = first_ch.combine(targets_set)
    .filter { entry, targets -> entry[1] in targets }
    .map { entry, targets -> entry }  // 仅保留原条目

// 查看结果
filtered_ch.view()

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