记录下方法以及踩过的坑，本文主要参考了https://blog.csdn.net/qq_16313365/article/details/72781469#commentBox，在此致谢！
（1）首先定义序列实体类：
@Document(collection ="sequence")
public class SeqInfo {
//getter and setter
@Id
private String id;
@Field("collection_name")
private String collName;
@Field("seq_id")
private Long seqId;
}
（2）接着定义注解AutoIncKey
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface AutoIncKey {
}
（3）定义监听类SaveEventListener
注意：这里面有个坑。按照https://blog.csdn.net/qq_16313365/article/details/72781469#commentBox中作者的方法以及评论中@yaoyaoyue的方法会出现问题：在下文insert的时候返回的结果显示字段值自增了，但是到数据库中一看却是初始值0，也就是并没有写入数据库AbstractMongoEventListener中的方法void onBeforeConvert(BeforeConvertEvent event)才成功。
@Component
public class SaveEventListener extends AbstractMongoEventListener {
private final MongoTemplatemongoTemplate;
@Autowired
public SaveEventListener(MongoTemplate mongoTemplate) {
this.mongoTemplate = mongoTemplate;
}
@Override
public void onBeforeConvert(BeforeConvertEvent<Object> event){
Object source = event.getSource();
if(source !=null){
ReflectionUtils.doWithFields(source.getClass(), new ReflectionUtils.FieldCallback() {
public void doWith(Field field)throws IllegalArgumentException, IllegalAccessException {
ReflectionUtils.makeAccessible(field);
// 如果字段添加了我们自定义的AutoIncKey注解
if (field.isAnnotationPresent(AutoIncKey.class)) {
// 设置自增ID
field.set(source, getNextId(source.getClass().getSimpleName()));
}
}
});
}
}
/**
* Description:获取下一个自增ID
*
* @param collName 集合名（这里用类名）
* @return 序列值
* @author example@gmail.com
* @Date 2019/2/21 9:58
*/
private LonggetNextId(String collName){
Query query =new Query(Criteria.where("collection_name").is(collName));
Update update =new Update();
update.inc("seqId", 1);
FindAndModifyOptions options =new FindAndModifyOptions();
options.upsert(true);
options.returnNew(true);
SeqInfo seqInfo =mongoTemplate.findAndModify(query, update, options, SeqInfo.class);
assert seqInfo !=null;
return seqInfo.getSeqId();
}
}
（4）使用方法如下。本文才用的是MongoRepository方式对MongoDB进行增删查改，只需要insert即可，不需要其他代码。
@Service
public class ApplicationService Implimplements ApplicationService {
@Autowired
private ApplicationRepository applicationRepository;
@Override
public Application create(Application application) {
//其他代码
return applicationRepository.insert(application);
}
这样就可以了。
最后说一句，简书自带的MarkDown模式真难用，插入代码块后，一部分代码还是显示为文本。请问你们的技术是怎么搞得？
网友评论