DataWeave代码验证与优化：多员工记录聚合逻辑适配问询

场景与需求 #

处理同部门员工对象数组，需按指定布尔逻辑聚合记录：

1. 先通过oldfreshHire、newPosition、newemployeeId计算value
2. 再用value、oldPosition、oldemployeeId计算最终value1
3. 执行流程：oldPosition为当前索引记录的position，newPosition为下一条记录的position；前两条记录的value1作为下一轮的oldfreshHire，与第三条记录继续计算，以此类推

测试输入数据 #

[
  {
    "employeeId": 100,
    "firstName": "John",
    "lastName": "Chen",
    "division": "2024-DOA09-1111",
    "email": "john.chen@xxxx.com",
    "freshHire": false,
    "position": "M"
  },
  {
    "employeeId": 101,
    "firstName": "Ameya",
    "lastName": "Job",
    "division": "2024-DOA09-1111",
    "email": "ameya.job@xxx.com",
    "freshHire": false,
    "position": "M"
  },
  {
    "employeeId": 102,
    "firstName": "Pat",
    "lastName": "Fay",
    "division": "2024-DOA09-1111",
    "email": "pat.fey@xxx.com",
    "freshHire": false,
    "position": "M"
  }
]

指定布尔逻辑规则 #

Boolean value = if(oldfreshHire == true) true else (if((newPosition == "M") and !isEmpty(newemployeeId)) true else false)
value1= if(value == true) true else (if((oldPosition == "M") and !isEmpty(oldemployeeId)) true else false)

现有DataWeave代码 #

%dw 2.0
fun combine(val,old, new) = (if(if(val) true else (if((new.position == 'C') and !(isEmpty(new.employeeId)) ) true else false)) true else (if((old.position == 'D') and !(isEmpty(old.employeeId)) ) true else false))
output application/json
---
(payload map ((item, index) -> if(sizeOf(payload) == 2) (combine(payload[index].freshHire,payload[index], payload[index+1])) else (combine((combine(payload[index].freshHire,payload[index], payload[index+1])), payload[index+1], payload[index+2]))))[0]

用户诉求 #

当前3条记录下代码输出符合预期（false），但担忧记录数量增多时代码失效，请求验证代码正确性并提供优化方案

代码正确性验证 #

现有代码存在严重缺陷，无法适配记录数量变化的场景：

1. 硬编码长度判断：仅处理记录数为2或3的情况，当记录数>3时，会直接访问index+2导致数组越界错误；记录数<2时也会报错
2. 逻辑规则不匹配：指定规则中newPosition判断为"M"、oldPosition判断为"M"，但代码中写的是"C"和"D"，当前测试数据刚好未暴露该错误
3. 遍历逻辑混乱：用map遍历所有元素后取[0]，当记录数>2时，每个索引都会执行嵌套计算，逻辑完全错误

优化方案 #

核心思路 #

使用DataWeave的reduce函数实现累积式迭代计算，适配任意长度的输入数组；同时严格对齐指定的布尔逻辑，提升代码可读性与可维护性

优化后的代码 #

%dw 2.0
// 单步计算函数：输入上一轮结果、当前员工、下一个员工，返回本轮计算后的value1
fun calculateNextValue(prevFreshHire: Boolean, current: Object, next: Object): Boolean = 
    var value = if (prevFreshHire) true 
                else (next.position == "M" and !isEmpty(next.employeeId))
    var value1 = if (value) true 
                 else (current.position == "M" and !isEmpty(current.employeeId))
    value1

output application/json
---
// 边界处理：记录数<2时返回false（可根据业务需求调整）
if (sizeOf(payload) < 2) false
else 
    payload[0 to -2] reduce ((current, accumulator) -> 
        var next = payload[indexOf(payload, current) + 1]
        calculateNextValue(accumulator, current, next)
    ) 
    initialValue: payload[0].freshHire

代码说明 #

1. 逻辑严格对齐：完全按照指定规则实现value和value1的计算，修正了原代码中position判断的错误
2. 适配任意长度：用reduce遍历数组的前n-1条记录，每次取当前记录与下一条记录计算，累积结果作为下一轮的oldfreshHire，支持2条及以上任意数量的记录
3. 边界处理：增加记录数<2的判断，避免索引越界问题
4. 可读性提升：将计算逻辑封装为独立函数，变量命名清晰，避免嵌套if的混乱

测试验证 #

用用户提供的3条测试数据运行优化后的代码，输出为false，与预期一致；增加更多符合条件的记录时，代码仍能正确执行聚合计算

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