第2章 文档 CRUD 与 Java Driver 实战 ​

MongoClient client = MongoClients.create("mongodb://localhost:27017");   // 全局一个，别每次 new
MongoDatabase db = client.getDatabase("yjoffer");
MongoCollection<Document> users = db.getCollection("t_user");

// Java Driver：同一批数据，Document.append 链式拼装
List<Document> seeds = List.of(
    new Document("username", "tom").append("age", 25).append("city", "武汉")
        .append("tags", List.of("java", "mysql")).append("status", 1),
    new Document("username", "amy").append("age", 30).append("city", "北京")
        .append("tags", List.of("python")).append("status", 1),
    new Document("username", "bo").append("age", 22).append("city", "武汉")
        .append("status", 0),                                    // 没有 tags 字段
    new Document("username", "cathy").append("age", 35).append("city", "上海")
        .append("tags", List.of("java", "spring", "mongodb")).append("status", 1)
);
users.insertMany(seeds);

// 单条：拿回生成的 _id
InsertOneResult r = users.insertOne(new Document("username", "dave").append("age", 28));
ObjectId newId = (ObjectId) r.getInsertedId().asObjectId().getValue();

// 自定义 _id（合法：任何唯一、可哈希的值都行，数字/字符串均可，但分布式场景推荐 ObjectId 或雪花ID字符串）
users.insertOne(new Document("_id", 1001).append("username", "admin"));

// 批量 + 无序（对照「批量导入尽量多成功」场景）
users.insertMany(moreDocs, new InsertManyOptions().ordered(false));

// 对照 SQL: WHERE status=1 AND (city='武汉' OR city='北京') AND age>=18 ORDER BY age DESC
Bson filter = Filters.and(
        Filters.eq("status", 1),
        Filters.or(Filters.eq("city", "武汉"), Filters.eq("city", "北京")),
        Filters.gte("age", 18)
);

for (Document u : users.find(filter).sort(Sorts.descending("age"))) {
    System.out.println(u.getString("username") + " / " + u.get("age"));
}

// 对照 SQL: SELECT username, age FROM t_user   （默认 _id 总会返回，除非显式排除）
Bson proj = Projections.include("username", "age");
users.find().projection(proj);                        // mongosh: db.t_user.find({}, {username:1, age:1})

// 只排除某大字段：对照 SELECT 除了 content 都要
Projections.exclude("content");

// 计算列、改名、数组截取等进阶：
Projections.fields(
    Projections.excludeId(),
    Projections.rename("username", "name"),
    Projections.slice("tags", 2),                    // 数组只回前2个
    Projections.elemMatch("orders", Filters.eq("orders.status", 1)) // 数组内条件过滤（对照 JSON 数组里"挑一个元素"）
);

// SQL: SELECT * FROM t_user ORDER BY age DESC, id ASC LIMIT 10 OFFSET 20
users.find()
     .sort(Sorts.orderBy(Sorts.descending("age"), Sorts.ascending("_id")))
     .skip(20)
     .limit(10)
     .into(new ArrayList<>());

long total = users.countDocuments(Filters.eq("status", 1));   // 对照 SELECT COUNT(*) WHERE status=1
// 估算级：集合大、只要大概数时用（走元数据，快但不保证精确）
long estimate = db.getCollection("t_user").estimatedDocumentCount();

// 对照 SQL: UPDATE t_user SET age=26, city='深圳' WHERE username='tom'
users.updateOne(
    Filters.eq("username", "tom"),
    new Document("$set", new Document("age", 26).append("city", "深圳"))
);

// 对照 UPDATE ... SET stock = stock - 1 WHERE ...（原子扣减，MySQL 同款防超卖语义）
users.updateOne(Filters.eq("username", "tom"), new Document("$inc", new Document("age", 1)));

// 数组操作——MySQL JSON 列做不到这么顺手的部分：
users.updateOne(Filters.eq("username", "tom"), new Document("$push", new Document("tags", "juc")));    // 追加元素
users.updateOne(Filters.eq("username", "tom"), new Document("$addToSet", new Document("tags", "jvm"))); // 去重追加
users.updateOne(Filters.eq("username", "tom"), new Document("$pull", new Document("tags", "mysql")));   // 按值删除元素

// 删除字段（对照 SET col = NULL 但更彻底：字段真的没了）
users.updateOne(Filters.eq("username", "tom"), new Document("$unset", new Document("city", "")));

// 全档替换（危险：漏字段=丢数据，生产慎用；等价 DELETE+INSERT）
users.replaceOne(Filters.eq("username", "bo"), new Document("_id", boId).append("username", "bo"));

// 批量更新（对照 UPDATE 无 WHERE 主键时命中多行）
users.updateMany(Filters.eq("status", 0), new Document("$set", new Document("status", 2)));

// upsert：有则更、无则插 —— 对照 INSERT ... ON DUPLICATE KEY UPDATE
users.updateOne(
    Filters.eq("username", "eve"),
    new Document("$setOnInsert", new Document("age", 18)).append("$set", new Document("status", 1)),
    new UpdateOptions().upsert(true)
);

users.deleteOne(Filters.eq("username", "eve"));     // 对照 DELETE ... WHERE username='eve' LIMIT 1
users.deleteMany(Filters.lt("age", 18));            // 对照 DELETE ... WHERE age<18

// 清空集合数据但保留索引与集合定义（对照 TRUNCATE TABLE）
users.deleteMany(new Document());

// 删整个集合（对照 DROP TABLE；连索引一起没了）
users.drop();

// Java Driver：每个动作包成 XxxModel，一次性提交
users.bulkWrite(List.of(
        new InsertOneModel<>(new Document("username", "frank").append("age", 27).append("status", 1)),
        new UpdateOneModel<>(Filters.eq("username", "amy"), new Document("$set", new Document("city", "杭州"))),
        new DeleteOneModel<>(Filters.eq("username", "dave"))
), new BulkWriteOptions().ordered(true));

public class User {
    @Id                       // org.bson.codecs.pojo.annotations.BsonId：映射到 _id
    private ObjectId id;
    private String username;
    private Integer age;
    private List<String> tags;
    // 无参构造 + getter/setter 必需（POJO Codec 基于反射实例化）
}

// 给集合挂上 POJO 编解码 Registry（PojoCodecProvider 内置于驱动 bson 包，无额外依赖）
CodecRegistry pojoRegistry = CodecRegistries.fromRegistries(
        db.getCodecRegistry(),
        CodecRegistries.fromProviders(PojoCodecProvider.builder().automatic(true).build()));

MongoCollection<User> userColl = db.getCollection("t_user", User.class)
        .withCodecRegistry(pojoRegistry);

User u = userColl.find(Filters.eq("username", "tom")).first();   // 直接拿到强类型对象
userColl.insertOne(newUser);                                     // 实体直接入库

FindIterable<Document> rs = users.find(new Document("$and", List.of(
                Filters.eq("status", 1),
                Filters.gte("age", 20),
                Filters.lte("age", 30),
                Filters.eq("tags", "java")))                    // 数组包含，等价 LIKE '%java%' 的意图
        .projection(Projections.include("username", "age"))
        .sort(Sorts.descending("age"))
        .skip(0).limit(10);
rs.forEach((Block<? super Document>) d -> System.out.println(d.toJson()));