MongoDB单查询实现指定文档返回并移除数组首元素

可以通过MongoDB的findOneAndUpdate方法一次原子性操作完成需求：既返回包含目标首元素的指定格式文档，同时移除原文档数组的首元素。该方法确保更新与查询操作在单个事务中执行，避免多查询带来的并发问题。

MongoDB Shell 实现 #

1. 返回单个property字段（推荐格式） #

db.yourCollection.findOneAndUpdate(
  { _id: "123" }, // 指定目标文档ID
  { $pop: { properties: -1 } }, // 原子移除数组首元素（-1表示从头部删除）
  {
    projection: {
      _id: 1,
      name: 1,
      age: 1,
      property: { $arrayElemAt: ["$properties", 0] } // 提取原数组首元素作为新字段
    },
    returnDocument: "before" // 返回更新前的文档，确保能拿到完整的原数组
  }
)

执行后返回结果：

{
  "_id": "123",
  "name": "Bob",
  "age": 23,
  "property": "a"
}

同时数据库中该文档的properties字段会变为["b", "c"]。

2. 返回包含首元素的数组格式 #

若需要返回properties: ["a"]格式，只需修改投影配置：

db.yourCollection.findOneAndUpdate(
  { _id: "123" },
  { $pop: { properties: -1 } },
  {
    projection: {
      _id: 1,
      name: 1,
      age: 1,
      properties: { $slice: [0, 1] } // 只保留原数组的第一个元素
    },
    returnDocument: "before"
  }
)

Java 驱动实现（4.x+版本） #

import com.mongodb.client.MongoClients;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.FindOneAndUpdateOptions;
import com.mongodb.client.model.Projections;
import com.mongodb.client.model.Updates;
import com.mongodb.client.model.ReturnDocument;
import org.bson.Document;
import org.bson.conversions.Bson;

public class MongoPopExample {
    public static void main(String[] args) {
        try (var mongoClient = MongoClients.create("mongodb://localhost:27017")) {
            MongoCollection<Document> collection = mongoClient.getDatabase("yourDB").getCollection("yourCollection");
            
            // 1. 构造查询条件：指定目标ID
            Bson filter = new Document("_id", "123");
            
            // 2. 构造更新操作：移除数组首元素
            Bson update = Updates.pop("properties", -1);
            
            // 3. 构造投影：提取首元素为property字段，保留其他必填字段
            Bson projection = Projections.fields(
                    Projections.include("_id", "name", "age"),
                    Projections.computed("property", new Document("$arrayElemAt", new Object[]{"$properties", 0}))
            );
            
            // 4. 配置选项：返回更新前的文档
            FindOneAndUpdateOptions options = new FindOneAndUpdateOptions()
                    .returnDocument(ReturnDocument.BEFORE)
                    .projection(projection);
            
            // 5. 执行操作并获取结果
            Document result = collection.findOneAndUpdate(filter, update, options);
            
            if (result != null) {
                System.out.println(result.toJson());
            }
        }
    }
}

边界情况说明 #

• 若目标文档的properties数组为空，$pop操作不会执行任何修改，返回的property字段为null。可在查询条件中添加{properties: {$ne: []}}过滤空数组文档：

  { _id: "123", properties: {$ne: []} }
  

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