首页 > 解决方案 > MongoDB - 如何使用另一个集合中的字段将字段添加到一个集合中的嵌套对象?

问题描述

我有两个集合 A 和 B。

集合 A 有一个嵌套对象“nested”,其中有一个字段“id”。集合 B 也有一个字段“id”和另一个“类型”。

我的问题是:

如何将集合 B 的“类型”字段添加到集合 A 的嵌套对象中,其中 id 匹配?

标签: mongodbnosqlaggregation

解决方案


最初,我正在寻找一个神奇的 mongo 单行查询来完成这项工作,但我想我会利用这个想法!

最后,我使用了以下 Java 解决方案:

 private void updateNestedInCollectionAWithTypeFromCollectionBWhereIdsMatch(List<CollectionBPojo> collectionB, MongoCollection<Document> collectionA) {

    StreamUtils.createStreamFromIterator(collectionA.find().iterator()).forEach(d -> {
        final List<Document> nestedList = (List<Document>)d.get("nested");
        for (int i = 0; i < nestedList.size(); i++) {
            final Document nested = nestedList.get(i);
            if(!nested.containsKey("type")) {
                final String id = nested.getString("id");
                final Optional<CollectionBPojo> collectionBPojo = collectionB.stream().filter(g -> g.getId().equals(id)).findFirst();
                if (collectionBPojo.isPresent()) {
                    final String type = collectionBPojo.get().getType();                 
                    final Document query = new Document("id", d.get("id"));
                    final Document update = new Document();
                    update.append("$set", new BasicDBObject("nested" + "." + i + ".type", type));
                    updates.add(new UpdateOneModel<>(query, update));
                }
            }
        }
    });

    if(!updates.isEmpty()) {
        final BulkWriteResult result = pipelineRunsCollection.bulkWrite(updates);         
    } else {
        System.out.print.ln("Nothing to update on db: " +  db.getName());
    }
}

}


推荐阅读