Springboot MongoDB实现自增序列的项目实践

@Document(collection = "sequences")
@Data
public class IdSequence {
    @Id
    private String id;
    /**
     * 自增的序列
     */
    private long nextId;
}

@Getter
@AllArgsConstructor
public enum IdType {
    /**
     * 图书
     */
    BOOK("book");
    private final String type;
}

@Component
public class IdHelper {
    private final MongoTemplate mongoTemplate;
    public IdHelper(MongoTemplate mongoTemplate) {
        this.mongoTemplate = mongoTemplate;
    }
    /**
     * 查询指定类型的自增序列
     *
     * @param idType 类型
     * @return 自增序列
     */
    public long nextId(IdType idType) {
        //构建查询对象
        Query query = Query.query(Criteria.where(LambdaUtil.getFieldName(IdSequence::getId)).is(idType.getType()));
        //构建自增条件
        Update update = new Update().inc(LambdaUtil.getFieldName(IdSequence::getNextId), 1L);
        //构建选项
        FindAndModifyOptions options = FindAndModifyOptions.options().upsert(true).returnNew(true);
        //查询自增序列
        IdSequence idSequence = mongoTemplate.findAndModify(query, update, options, IdSequence.class);
        //返回自增序列
        return Objects.requireNonNull(idSequence).getNextId();
    }
}

@RestController
@RequestMapping(value = "/id")
public class IdSequenceController {
    @Resource
    private IdHelper idHelper;
    @GetMapping(value = "/nextId")
    public long nextId() {
        return idHelper.nextId(IdType.BOOK);
    }
}