面试题：MongoDB更新操作返回文档在复杂嵌套文档场景下的实现

Question

假设存在一个文档，其结构为{ 'name': 'John', 'address': { 'city': 'New York', 'zip': '10001' }, 'hobbies': ['reading', 'traveling'] }。现在要更新'hobbies'数组，添加一个新的爱好'drawing'，并返回更新后的整个文档。在Java的MongoDB驱动环境下，如何实现这个更新并返回文档的操作？

Answer 1

import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Filters;
import com.mongodb.client.model.UpdateOptions;
import com.mongodb.client.model.Updates;
import org.bson.Document;
import org.bson.conversions.Bson;

public class MongoDBUpdateExample {
    public static void main(String[] args) {
        // 连接到MongoDB
        MongoClient mongoClient = MongoClients.create("mongodb://localhost:27017");
        MongoDatabase database = mongoClient.getDatabase("yourDatabaseName");
        MongoCollection<Document> collection = database.getCollection("yourCollectionName");

        // 定义查询条件，假设以name为唯一标识
        Bson filter = Filters.eq("name", "John");

        // 定义更新操作，向hobbies数组中添加'drawing'
        Bson update = Updates.push("hobbies", "drawing");

        // 执行更新操作并返回更新后的文档
        UpdateOptions options = new UpdateOptions().returnOriginal(false);
        Document updatedDocument = collection.findOneAndUpdate(filter, update, options);

        System.out.println(updatedDocument);

        // 关闭MongoDB连接
        mongoClient.close();
    }
}

上述代码步骤如下：

1. 连接MongoDB：使用MongoClients.create方法连接到本地MongoDB服务器。
2. 获取数据库和集合：指定要操作的数据库和集合名称。
3. 定义查询条件：使用Filters.eq方法定义基于name字段的查询条件。
4. 定义更新操作：使用Updates.push方法向hobbies数组中添加新元素。
5. 执行更新并返回文档：使用findOneAndUpdate方法执行更新操作，并通过UpdateOptions设置返回更新后的文档。
6. 关闭连接：操作完成后关闭MongoDB连接。

请注意，代码中的yourDatabaseName和yourCollectionName需要替换为实际的数据库和集合名称。