From 9a0968f87c5d7630def1bf477e12356c0fe7dd8c Mon Sep 17 00:00:00 2001
From: jlzhou <12020042@qq.com>
Date: Mon, 26 Aug 2024 16:17:30 +0800
Subject: [PATCH] =?UTF-8?q?update=20=E5=B0=8F=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/main/java/com/ruoyi/common/utils/MongoUtil.java  | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/utils/MongoUtil.java b/ruoyi-common/src/main/java/com/ruoyi/common/utils/MongoUtil.java
index 67e788e..347c882 100644
--- a/ruoyi-common/src/main/java/com/ruoyi/common/utils/MongoUtil.java
+++ b/ruoyi-common/src/main/java/com/ruoyi/common/utils/MongoUtil.java
@@ -503,8 +503,9 @@ public class MongoUtil {
    * @param entity
    * @param <T>
    */
-  public static <T> void fillId(T entity) {
+  public static <T> T fillId(T entity) {
     fillId(entity, FillIdType.auto);
+    return entity;
   }
 
 
@@ -515,12 +516,12 @@ public class MongoUtil {
    * @param type
    * @param <T>
    */
-  public static <T> void fillId(T entity, FillIdType type) {
+  public static <T> T fillId(T entity, FillIdType type) {
     Assert.notNull(entity, "entity不能为空");
 
     Field[] fs = ReflectUtil.getFields(entity.getClass(), field -> field.isAnnotationPresent(Id.class));
     if (fs.length != 1) {
-      return;
+      return entity;
     }
     Field f = fs[0];
     if (f.getType() == Integer.class) {
@@ -544,7 +545,7 @@ public class MongoUtil {
         ReflectUtil.setFieldValue(entity, f, UUID.fastUUID().toString(true));
       }
     }
-
+    return entity;
 
   }