diff --git a/src/main/java/cn/craccd/mongoHelper/utils/MongoHelper.java b/src/main/java/cn/craccd/mongoHelper/utils/MongoHelper.java index cc2559e4b47aeb1c743554716fc48e3368473f39..6f7c429be9165e44a69c7a294565791452b01710 100644 --- a/src/main/java/cn/craccd/mongoHelper/utils/MongoHelper.java +++ b/src/main/java/cn/craccd/mongoHelper/utils/MongoHelper.java @@ -243,7 +243,7 @@ public class MongoHelper implements ApplicationContextAware { * * @param object 对象 */ - public void insertOrUpdate(Object object) { + public String insertOrUpdate(Object object) { Long time = System.currentTimeMillis(); String id = (String) ReflectUtil.getFieldValue(object, "id"); @@ -265,6 +265,7 @@ public class MongoHelper implements ApplicationContextAware { logSave(object); mongoTemplatePrimary.save(object); + id=(String) ReflectUtil.getFieldValue(object, "id"); } else { // 更新 Field[] fields = ReflectUtil.getFields(object.getClass()); @@ -282,7 +283,7 @@ public class MongoHelper implements ApplicationContextAware { logSave(objectOrg); mongoTemplatePrimary.save(objectOrg); } - + return id; } /** @@ -290,7 +291,7 @@ public class MongoHelper implements ApplicationContextAware { * * @param object 对象 */ - public void insert(Object object) { + public String insert(Object object) { String id = (String) ReflectUtil.getFieldValue(object, "id"); Object objectOrg = StrUtil.isNotEmpty(id) ? getMongoTemplate().findById(id, object.getClass()) : null; if (objectOrg != null) { @@ -299,6 +300,8 @@ public class MongoHelper implements ApplicationContextAware { } insertOrUpdate(object); + id = (String) ReflectUtil.getFieldValue(object, "id"); + return id; } /**