|
一.
创建自定义注解：
package com.orangecds.designmaterial.api.entity;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* @author xiyun.zhao
* @title: AutoIncKey
* @projectName design-material
* @description: 标识注解：标识主键ID需要自动增长
* @date 2021/8/317:45
*/
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface AutoIncKey {
}
二.
创建自增主键实体类
package com.orangecds.designmaterial.api.entity;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import org.springframework.data.mongodb.core.index.Indexed;
import org.springframework.data.mongodb.core.mapping.Document;
/**
* @program: design-material
* @ClassName SeqInfo
* @description: 创建自增长序列
* @author: xiyun.zhao
* @create: 2021-08-03 17:41
**/
@ApiModel("序列的类")
@Data
@Document(collection = "sequence")
public class SeqInfo {
@ApiModelProperty("逻辑主键Id")
@Indexed
private String id;
@ApiModelProperty("序列值")
private Long seqId;
}
三.
创建自定义注解监听的类
package com.orangecds.designmaterial.seque.listener;
import com.orangecds.designmaterial.api.entity.AutoIncKey;
import com.orangecds.designmaterial.api.entity.SeqInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
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;
import java.lang.reflect.Field;
/**
* @description: 实现监听类
* @author xiyun.zhao
* @param:
* @return:
* @date: 2021/8/4 10:16
*/
@Component
public class SaveEventListener extends AbstractMongoEventListener<Object>{
private static final Logger logger= LoggerFactory.getLogger(SaveEventListener.class);
修改git 的用户名称：git config --global --replace-all user.name "xiyun.zhao"
@Autowired
private MongoTemplate mongo;
@Override
public void onBeforeConvert(BeforeConvertEvent<Object> event) {
final Object source = event.getSource();
if (source != null) {
ReflectionUtils.doWithFields(source.getClass(), new ReflectionUtils.FieldCallback() {
@Override
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()));
}
}
});
}
}
private Long getNextId(String id) {
Query query = new Query(Criteria.where("id").is(id));
Update update = new Update();
update.inc("seqId", 1);
FindAndModifyOptions options = new FindAndModifyOptions();
options.upsert(true);
options.returnNew(true);
SeqInfo seqInfo= mongo.findAndModify(query, update, options, SeqInfo.class);
return seqInfo.getSeqId();
}
}
|
