|
|
@@ -1,68 +1,63 @@
|
|
|
-//package com.mooctest.listener;
|
|
|
-//
|
|
|
-//import java.lang.annotation.Annotation;
|
|
|
-//import java.lang.reflect.Field;
|
|
|
-//
|
|
|
-//import com.mooctest.annotation.AutoValue;
|
|
|
-//import com.mooctest.model.SequenceId;
|
|
|
-//import org.springframework.beans.factory.annotation.Autowired;
|
|
|
-//import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
|
|
|
-//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.mapping.event.BeforeSaveEvent;
|
|
|
-//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 javax.annotation.Resource;
|
|
|
-//
|
|
|
-//@Component
|
|
|
-//public class SaveEventListener extends AbstractMongoEventListener<Object> {
|
|
|
-// @Resource(name = "mongoTemplate1")
|
|
|
-// private MongoTemplate mongo;
|
|
|
-//
|
|
|
-// @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);
|
|
|
-// // 如果字段添加了我们自定义的AutoValue注解
|
|
|
-// if (field.isAnnotationPresent(AutoValue.class) && field.get(source) instanceof Number
|
|
|
-// && field.getLong(source) == 0) {
|
|
|
-// // field.get(source) instanceof Number &&
|
|
|
-// // field.getLong(source)==0
|
|
|
-// // 判断注解的字段是否为number类型且值是否等于0.如果大于0说明有ID不需要生成ID
|
|
|
-// // 设置自增ID
|
|
|
-//
|
|
|
-// field.set(source, getNextId(event.getCollectionName()));
|
|
|
-//
|
|
|
-// }
|
|
|
-// }
|
|
|
-// });
|
|
|
-// }
|
|
|
-// }
|
|
|
-//
|
|
|
-// /**
|
|
|
-// * 获取下一个自增ID
|
|
|
-// *
|
|
|
-// * @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 seq = mongo.findAndModify(query, update, options, SequenceId.class);
|
|
|
-// return seq.getSeqId();
|
|
|
-// }
|
|
|
-//}
|
|
|
+package com.mooctest.listener;
|
|
|
+
|
|
|
+import com.mooctest.annotation.AutoValue;
|
|
|
+import com.mooctest.model.SequenceId;
|
|
|
+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 javax.annotation.Resource;
|
|
|
+import java.lang.reflect.Field;
|
|
|
+
|
|
|
+@Component
|
|
|
+public class SaveEventListener extends AbstractMongoEventListener<Object> {
|
|
|
+ @Resource(name = "mongoTemplate1")
|
|
|
+ private MongoTemplate mongo;
|
|
|
+
|
|
|
+ @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);
|
|
|
+ // 如果字段添加了我们自定义的AutoValue注解
|
|
|
+ if (field.isAnnotationPresent(AutoValue.class) && field.get(source) instanceof Number
|
|
|
+ && field.getLong(source) == 0) {
|
|
|
+ // field.get(source) instanceof Number &&
|
|
|
+ // field.getLong(source)==0
|
|
|
+ // 判断注解的字段是否为number类型且值是否等于0.如果大于0说明有ID不需要生成ID
|
|
|
+ // 设置自增ID
|
|
|
+
|
|
|
+ field.set(source, getNextId(event.getCollectionName()));
|
|
|
+
|
|
|
+ }
|
|
|
+ }
|
|
|
+ });
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取下一个自增ID
|
|
|
+ *
|
|
|
+ * @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 seq = mongo.findAndModify(query, update, options, SequenceId.class);
|
|
|
+ return seq.getSeqId();
|
|
|
+ }
|
|
|
+}
|
