Springboot下Mongo数据库写入时字段序号自增

2019-02-21  本文已影响0人  wxb2dyj

记录下方法以及踩过的坑,本文主要参考了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模式真难用,插入代码块后,一部分代码还是显示为文本。请问你们的技术是怎么搞得?

上一篇下一篇

猜你喜欢

热点阅读