您现在的位置是：网站首页> 编程资料编程资料

mongodb字段值自增长实现代码_MongoDB_

2023-05-27 444人已围观

简介 mongodb字段值自增长实现代码_MongoDB_

MongoDB 没有像 SQL 一样有自动增长的功能， MongoDB 的 _id 是系统自动生成的12字节唯一标识。但在某些情况下，我们可能需要实现 ObjectId 自动增长功能。由于 MongoDB 没有实现这个功能，我们可以通过编程的方式来实现，以下我们将在 counters 集合中实现_id字段自动增长。

1.创建计数器集合

期望_id字段从1,2,3,4到n，启动一个自动递增的整数序列，如：

 { "_id":1, "title": "标题", "content": "内容1", "type": "类型" }

为此，创建 counters 集合，序列字段值可以实现自动长：

 db.createCollection("counters")

初始化集合，以objId作为主键，sequence_value 字段是序列通过自动增长后的一个值：

 db.counters.insert({_id:"objId",sequence_value:0})

2.查询序列号

查询返回更新后的序列号

 db.counters.findAndModify({ query: {_id: "objId" }, update: {$inc:{sequence_value:1}}, new: true }).sequence_value;

操作符解释：

$inc可以对文档的某个值为数字型（只能为满足要求的数字）的键进行增减的操作；

 db.collection.findAndModify({ query: , //定义关于哪些记录需要修改的选择标准 sort: , //确定选择标准检索多个文档时应修改的文档 new: , //表示将显示修改后的文档 fields: , //指定要返回的字段集 upsert:  //如果选择标准无法检索文档，则创建一个新文档 remove:  //为true，query指定的文档将从数据库中删除 )}

3.测试

创建测试集合sms：

 db.createCollection("sms")

在sms集合中新增文档，实现_id自增长：

 db.sms.insert({ _id: db.counters.findAndModify({query:{_id: "objId" },update: {$inc:{sequence_value:1}},"new":true}).sequence_value, title: "标题1", content: "短信1", type: "1" })

查询sms集合：

 db.sms.find({}).sort({_id:1})

4.java实现

java实现以上功能，数据库驱动版本不同运行效果有差异，仅供参考：

 private MongoDatabase conn; static{ this.conn = getDatabase(databaseName); } /** * 连接数据库 * @param databaseName 数据库名称 * @return 数据库连接对象 */ private static MongoDatabase getDatabase(databaseName){ MongoDatabase mongoDatabase = null; try{ // 连接到 mongodb 服务 MongoClient mongoClient = new MongoClient( "localhost" , 27017 ); // 连接到数据库 MongoDatabase mongoDatabase = mongoClient.getDatabase(databaseName); System.out.println("Connect to database successfully"); }catch(Exception e){ System.err.println( e.getClass().getName() + ": " + e.getMessage() ); } return mongoDatabase; } /** * 获取最新序列号 * @return 序列号 */ private static int getNextSequenceValue(){ DBCollection collection = conn.getCollection("counters"); DBObject query = new BasicDBObject("_id", new BasicDBObject("$eq", "objId")); DBObject newDocument =new BasicDBObject(); newDocument.put("$inc", new BasicDBObject().append("sequence_value", 1)); newDocument.put("new": true); DBObject ret = collection.findAndModify(query, newDocument); if (ret == null){ return 0; }else{ return (Integer)ret.get("sequence_value"); } } /** * 新增集合文档 */ public static void addSms(){ int id = getNextSequenceValue(); if(id != 0){ DBCollection collection = conn.getCollection("sms"); List documents = new ArrayList(); for(int i = 0; i < 20; i++){ int id = getNextSequenceValue(); Document document = new Document("_id", id). append("title", "标题" + i). append("content", "短信" + i). append("type", 1); documents.add(document); } collection.insertMany(documents); System.out.println("文档插入成功"); } } /** * 查询集合 */ public static void findSms(){ DBCollection collection = conn.getCollection("sms"); FindIterable findIterable = collection.find(); MongoCursor mongoCursor = findIterable.iterator(); while(mongoCursor.hasNext()){ System.out.println(mongoCursor.next()); } }