Groovy字符串出现次数统计问题：脚本未达预期输出

需求与输入输出 #

需求 #

在Groovy中统计指定字符串的出现次数，将统计结果替换原JSON中的result字段，新增TP（对应Talent Partner的次数）和Partner字段。

输入JSON #

[
{
"Org" : "Test",
"result": "Talent Partner,  Partner, Partner"
}
]

预期输出JSON #

[
{
"Org" : "Test",
"TP": "1",
"Partner" : "2"
}
]

原脚本问题分析 #

原脚本存在4个关键问题：

• 循环条件错误：数组长度判断需用NumberOfElements.length，直接比较数组对象会导致循环无法正确执行。
• 未处理空格：split后的元素包含前置空格（如" Partner"），直接与"Partner"匹配会失败，统计结果为0。
• 属性重复覆盖：两次设置同一个document.dynamic.userdefined.NoOfTP属性，后赋值的Partner会覆盖之前的NoOfTP，且未将统计结果写入目标JSON对象。
• 未处理JSON数组：输入是JSON数组，直接访问jsonDoc.result会抛出异常，需先取数组第一个元素jsonDoc[0]。

修复后的脚本 #

import groovy.json.JsonSlurper
import groovy.json.JsonOutput

def slurper = new JsonSlurper();
def jsonDoc = slurper.parseText(doc);
// 取数组第一个元素
def item = jsonDoc[0]
def NoOfTP = 0;
def PartnerCount = 0;
// 分割后清理每个元素的前后空格
def NumberOfElements = item.result.split(',').collect { it.trim() };
// 用增强for循环简化遍历逻辑
for (element in NumberOfElements) {
    if (element == 'Talent Partner') {
        NoOfTP++
    } else if (element == 'Partner') {
        PartnerCount++
    }
}
// 移除原result字段，新增统计字段
item.remove('result')
item.TP = String.valueOf(NoOfTP)
item.Partner = String.valueOf(PartnerCount)

// 将修改后的JSON写入输出流
is = new ByteArrayInputStream(JsonOutput.toJson(jsonDoc).getBytes());
dataContext.storeStream(is, props);

修复说明 #

1. 处理JSON数组：通过jsonDoc[0]获取数组内的目标对象，避免直接访问数组属性报错。
2. 清理空格：split后用collect { it.trim() }去除每个元素的前后空格，确保字符串匹配准确。
3. 修正循环逻辑：使用增强for循环简化代码，避免手动控制索引的长度判断错误。
4. 更新JSON结构：移除原result字段，新增TP和Partner字段并赋值统计结果，最后将修改后的JSON序列化为输出流。

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