4.3MongoDB实现自增id

MongoDB在数据库设计上和其他关系数据库有什么不同，估计第一个想到的就是自增ID的实现。Oracle可以通过sequence来实现，mysql和Sqlserver自带自增id字段。

MongoDB怎么实现自增id呢？MongoDB官网上也提供了一种实现的方法，就是自定义一个获取自增ID的方法，然后每次插入的时候就去获取下一个ID,再插入到集合中。

下面来讲一下基于java的具体实现，原理就是先获取序列的值，然后设置给对应的实体，代码如下：

定义序列MongoSequence.java

public class MongoSequence {
    @Id
    private String id;
    private int seq;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public int getSeq() {
        return seq;
    }

    public void setSeq(int seq) {
        this.seq = seq;
    }
}

这是简单的java对象，就是模拟序列的存储

序列生成工具类MongoAutoidUtil.java

@Component
public class MongoAutoidUtil {
    @Autowired
    MongoTemplate mongo;

    public int getNextSequence(String collectionName) {
        MongoSequence seq = mongo.findAndModify(
                query(where("_id").is(collectionName)),
                new Update().inc("seq", 1),
                options().upsert(true).returnNew(true),
                MongoSequence.class);

        return seq.getSeq();
    }
}

代码很简单，就是根据传入的序列名称获取下一个序列值。
通过findAndModify这个方法来保证序列唯一，因为这是一个原子操作。
这段代码的原理就是，新建一个mongoSequence的Collection（如果没有才新增），然后根据传入的collectionName，根据_id即主键去找，如果有值，则+1，并返回修改后的值，如果没有则新增一条记录，并返回该值。

如果改为关系数据库，则原理就是这样，新建一个表mongoSequence，里面两个字段_id,seq，当根据_id来取值时，如果没有记录，则新增一条记录，并返回1，如果有记录，则把seq+1，并返回修改后的记录。通过findAndModify来实现lock锁定这一行的目的。

测试新增

@Test
    public void add() {
        for (int i = 0; i < 10; i++) {  //增加一条记录
            Article article = new Article();
            article.setId(mongoAutoidUtil.getNextSequence("seq_article"));
            article.setTitle("MongoTemplate的基本使用");
            article.setAuthor("kcy");
            article.setUrl("http://jianshu.com/");
            article.setTags(Arrays.asList("java", "mongodb", "spring"));
            article.setVisitCount(0L);
            article.setAddTime(new Date());
            mongoTemplate.save(article);
        }

        Iterable<Article> articles = articleRepository.findAll();
        articles.forEach(article2 -> {
            System.out.println(article2.toString());
        });
    }

在设置article的id时通过article.setId(mongoAutoidUtil.getNextSequence("seq_article"));来获取序列的值，并赋给这个id。

自增id的实现和oracle的序列原理基本一致。

源码下载

[本工程详细源码]
(https://github.com/chykong/java_component/tree/master/chapter4_3_mongodb_autoid)