spring-data-mongodb之自增ID实现

来源：猿天地

链接：http://cxytiandi.com/blog/detail/1897

用了mongodb之后要是问我mongo和mysql的区别在哪里？第一点我就会想到的是没有自增ID,mongo里面是ObjectId。今天我们就自己来实现自增的ID。

像mysql这种数据库是内部实现了自增ID,今天我们要自己实现一个，不知道大家有没有具体的思路。

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

我们既然用了spring-data-mongodb这个框架，就要基于这个框架来实现一套逻辑，而且每次插入都要自己去手动的调用方法获取一次ID,是不是太繁琐了。

我们用的是监听的模式，在数据插入到集合之前，我们通过反射将ID设置到保存的对象中，来实现自动设置，对写代码的人来说完全透明。

首先我们定义一个用于存储每个集合的ID记录，记录每个集合的自增ID到了多少。

@Document(collection = "sequence")
public class SequenceId {   
    @Id
    private String id;    
    @Field("seq_id")    
    private long seqId;    
    @Field("coll_name")    
    private String collName;

}

接下来定义我们测试的实体类,注意自增ID的类型不要定义成Long这种包装类，mongotemplate的源码里面对主键ID的类型有限制。

@Documentpublic class Student {    
    @GeneratedValue
    @Id
    private long id;    
    private String name;
}

下面我们定义个注解来标识此字段要自动增长ID,有些场景下可能不需要自动增长，需要自动增长的时候我们加上这个注解。

@Retention(RetentionPolicy.RUNTIME)
@Target({ ElementType.FIELD })
public @interface GeneratedValue {

}

接下来就是实现自动注入ID的监听器了,需要配置让spring管理

<bean class="com.cxytiandi.mongo.autoid.SaveMongoEventListener"></bean>

public class SaveMongoEventListener extends AbstractMongoEventListener<Object> {
    @Resource
    private MongoTemplate mongoTemplate;    @Override
    public void onBeforeConvert(final Object source) {        
        if(source != null) {            
            ReflectionUtils.doWithFields(source.getClass(), 
                    new ReflectionUtils.FieldCallback() {
                public void doWith(Field field) throws IllegalArgumentException, IllegalAccessException {                    ReflectionUtils.makeAccessible(field);                    if (field.isAnnotationPresent(GeneratedValue.class)) {                        //设置自增ID
                        field.set(source, getNextId(source.getClass().getSimpleName()));
                }
               }
            });
        }
   }
 /**
  * 获取下一个自增ID
  * @author yinjihuan
  * @param collName  集合名
  * @return
  */
 private Long getNextId(String collName) {     
    Query query = new Query(Criteria.where("collName").is(collName));     
    Update update = new Update();
    update.inc("seqId", 1);     
    FindAndModifyOptions options = new FindAndModifyOptions();
    options.upsert(true);
    options.returnNew(true);     
    SequenceId seqId = mongoTemplate.findAndModify(query, update, options, SequenceId.class);     return seqId.getSeqId();
 }

}

findAndModify()是原子操作，所以不用担心并发问题