MongoDB实现主键自增

由于MySQL的使用习惯，导致在使用MongoDB的过程中没有自增长的ID十分不爽，而自动生成的序列则又没有什么可读性，恰巧在看书的过程中发现有讲到，所以记录下。

 

第一步：自定义一个自增长标识，Java的注解就是一个非常优秀的选择，代码如下：

package com.lsm1998.userservice.mongo;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @作者：刘时明
 * @时间：2019/3/17-11:33
 * @说明：自增长标识
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface GeneratedValue
{

}

 

第二步：创建一个MongoDB文档实体，用来记录每个集合中最近的ID，代码如下：

package com.lsm1998.userservice.mongo;

import org.springframework.data.annotation.Id;
import org.springframework.data.mongodb.core.mapping.Document;
import org.springframework.data.mongodb.core.mapping.Field;

/**
 * @作者：刘时明
 * @时间：2019/3/17-11:43
 * @说明：保存每个集合最近一次的ID
 */
@Document(collection = "sequence")
public class SequenceID
{
    @Id
    private String id;
    @Field("seq_id")
    private long seqId;
    @Field("coll_name")
    private String collName;

    public String getId()
    {
        return id;
    }

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

    public long getSeqId()
    {
        return seqId;
    }

    public void setSeqId(long seqId)
    {
        this.seqId = seqId;
    }

    public String getCollName()
    {
        return collName;
    }

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

 

第三步：实现一个自定义的MongoDB保存监听器，代码如下：

package com.lsm1998.userservice.mongo;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.FindAndModifyOptions;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.mapping.event.AbstractMongoEventListener;
import org.springframework.data.mongodb.core.mapping.event.BeforeConvertEvent;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Component;
import org.springframework.util.ReflectionUtils;

/**
 * @作者：刘时明
 * @时间：2019/3/17-11:35
 * @说明：保存监听器
 */
@Component
public class SaveMongoEventListener extends AbstractMongoEventListener