为MongoDB增加自增长主键生成的功能

为MongoDB增加自增长主键生成的功能

标签: MongoDB 自增长主键

每个MongoDB的document都有一个_id字段作为它的第一个属性，这个值通常是一个BSON对象id，因此，这个id对于集合中的每个成员都是唯一的，如果用户插入一个document没有提供一个id，数据库将自动生成一个id，并存储在_id字段。   
   The BSON ObjectId Datatype   
一个BSON ObjectID是由12个字节组成：4字节时间+3字节机器id+2字节进程id+3字节的数字
{ "_id" : ObjectId("4c691e72ed2a47b462dfa806") }


有时候我们的应用中需要自增长的数字型主键，MongoDB在这方面并没有给我们提供支持，我们需要加以改造，使其具有自增长主键生成的功能。此次的功能改造，依赖的是morphia开源项目（MongoDB在java语言上的ORM实现，http://code.google.com/p/morphia/），直接上代码吧。


首先定义一个保存各个 collection的主键增量值的系统配置collection：StoredSeqence

java 代码

/**
 * MongoDB自增长主键维护队列，类似于MSSQL，Oracle维护主键的方式
 *
 * @author yongtree
 * @date 2011-1-17 下午06:58:05
 * @version 1.0
 */
@Entity(noClassnameStored=true)
public class StoredSeqence implements Serializable {

    private static final long serialVersionUID = 1L;

    @Id
    String collName;

    Long value;


    public StoredSeqence(){

    }

    public StoredSeqence(String collName) {
        this.collName = collName;
    }

    public Long getValue() {
        return value;
    }

    public void setValue(Long value) {
        this.value = value;
    }

    public String getCollName() {
        return collName;
    }

    public void setCollName(String collName) {
        this.collName = collName;
    }




}

然后定义一个实体的基类，在基类中处理主键生成。

java 代码

/**
 * 自增长数字类型主键的Mongo实体
 *
 * @author yongtree
 * @date 2011-1-17 下午04:11:04
 * @version 1.0
 */
public abstract class LongPKMongoEO extends BaseMongoEO {

    @Id
    Long _id;

    @Transient
    protected Datastore ds;



    public void setDs(Datastore ds) {
        this.ds = ds;
    }

    @PrePersist
    void prePersist() {

        //自增性主键的处理

        if (_id == null) {
            String collName = ds.getCollection(getClass()).getName();
            Query<StoredSeqence> q = ds.find(StoredSeqence.class, "_id",
                    collName);
            StoredSeqence ss = q.get();
            if(ss==null){//不存在该实体的注册，则新创建一个
                ss = new StoredSeqence(collName);
                ss.setValue(1l);
            }else{
                ss.setValue(ss.getValue()+1);
            }
            ds.save(ss);
            _id=ss.value;
        }
    }

    public Long getId() {
        return _id;
    }

}

这样自增长主键的生成的主要功能基本上已经完成了，具体如何使用，接下来将根据实际的项目再做介绍，请继续关注我的博客。

附件：

LongPKMongoEO.java

StoredSeqence.java

旧一篇：架构设计之–Don’t Repeat Yourself  |  新一篇：敏捷开发：拥抱变化的产品开发流程