From 9c059f6ef65b6453ad4353f39f38eefa6285d10a Mon Sep 17 00:00:00 2001
From: yangbo <1970638756@qq.com>
Date: Mon, 9 Nov 2020 12:02:24 +0800
Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E7=BB=84=E4=B8=AD=E7=9A=84=E6=A8=A1?=
 =?UTF-8?q?=E7=B3=8A=E6=9F=A5=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/cn/craccd/mongoHelper/utils/MongoHelper.java    | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/main/java/cn/craccd/mongoHelper/utils/MongoHelper.java b/src/main/java/cn/craccd/mongoHelper/utils/MongoHelper.java
index cc2559e..6f7c429 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;
 	}
 
 	/**
-- 
Gitee