JMeter中用Groovy查询MongoDB：按_id查询及数组计数遇阻

嘿，我懂你作为新手找不到适用示例的头疼！刚好我对JMeter操作MongoDB这块比较熟，来帮你一步步解决问题。

一、先搞定按_id查询失败的问题 #

你遇到的核心坑大概率是MongoDB的_id是ObjectId类型，不是普通字符串——直接传JSON格式的字符串肯定匹配不上，这就是查询失败的原因。试试下面的操作：

1. 必须导入ObjectId类：
   在JSR223 Sampler里先导入转换类，才能把字符串转成MongoDB能识别的_id格式。

2. 构造正确的查询条件：
   别用JSON字符串当查询条件，改用BasicDBObject构造，把目标_id字符串转换成ObjectId类型。

二、统计数组元素数量的方法 #

拿到查询到的文档后，直接获取数组字段，调用size()方法就能得到元素数量，还能把结果存到JMeter变量里方便后续使用。

完整可复用示例代码 #

把下面的代码放到JSR223 Sampler里，记得替换成你的数据库、集合、目标_id和数组字段名：

// 导入必要的依赖类
import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.BasicDBObject;
import org.bson.types.ObjectId;

// 这里假设你已经成功创建了MongoClient连接（如果没写连接代码，参考下面注释）
// MongoClient mongoClient = new MongoClient("你的MongoDB地址", 端口号); // 比如localhost:27017

// 获取目标数据库和集合
MongoDatabase targetDB = mongoClient.getDatabase("你的数据库名");
MongoCollection<BasicDBObject> targetCollection = targetDB.getCollection("你的集合名", BasicDBObject.class);

// 替换成你要查询的_id字符串
String targetObjectId = "60d21b4667d0d8992e610c85";
// 构造查询条件：把字符串转成MongoDB认可的ObjectId
BasicDBObject queryCondition = new BasicDBObject("_id", new ObjectId(targetObjectId));

// 执行查询，获取单个匹配的文档
BasicDBObject resultDocument = targetCollection.find(queryCondition).first();

if (resultDocument != null) {
    // 打印查询结果到JMeter日志（调试用）
    log.info("查询到的目标文档：" + resultDocument.toJson());
    
    // 统计数组元素数量：替换成你的数组字段名
    def arrayField = resultDocument.get("你的数组字段名");
    if (arrayField != null && arrayField instanceof List) {
        int elementCount = arrayField.size();
        log.info("数组元素总数：" + elementCount);
        // 将结果存入JMeter变量，后续组件可直接引用${array_element_count}
        vars.put("array_element_count", String.valueOf(elementCount));
    } else {
        log.warn("指定的字段不存在或者不是数组类型");
    }
} else {
    log.warn("没有找到匹配该_id的文档");
}

// 如果是在这个Sampler里创建的MongoClient，记得用完关闭
// mongoClient.close();

几个关键注意事项 #

• 驱动包要到位：把mongo-java-driver（版本建议选3.x，和JMeter3.3兼容更好）的jar包放到JMeter安装目录下的lib文件夹，然后重启JMeter，否则会报类找不到的错误。
• 选对语言：JSR223 Sampler的语言一定要选Groovy，Groovy对Java类的支持更完善，性能也比BeanShell好很多。
• 避免重复创建连接：如果多个Sampler都要操作MongoDB，建议把连接逻辑放到Test Plan的Setup Thread Group里，创建一次连接后存到JMeter属性里，其他Sampler直接复用，避免频繁创建销毁连接影响性能。

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