From 3d3038e74a3b1003a44eef72c46d835e45a7c7b8 Mon Sep 17 00:00:00 2001
From: "583641232@qq.com" <583641232@qq.com>
Date: Wed, 22 Jan 2025 16:38:44 +0800
Subject: [PATCH] =?UTF-8?q?:sparkles:=20=E6=95=B0=E6=8D=AE=E6=96=B0?=
 =?UTF-8?q?=E5=A2=9E=E6=97=B6=E8=AE=B0=E5=BD=95=E4=B8=9A=E5=8A=A1id,?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E4=BF=AE=E6=94=B9=E6=97=B6=E8=AE=B0=E5=BD=95?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=89=8D=E5=90=8E=E5=8F=98=E5=8C=96=E5=80=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client/pom.xml                                |  6 ---
 .../clientdemo/aspectj/ActionLogAspect.java   | 37 +++++++++++++++++++
 .../demo/controller/DemoUserController.java   |  5 ++-
 3 files changed, 40 insertions(+), 8 deletions(-)

diff --git a/client/pom.xml b/client/pom.xml
index f321776..976bf0c 100644
--- a/client/pom.xml
+++ b/client/pom.xml
@@ -17,12 +17,6 @@
             <artifactId>alog-common</artifactId>
             <version>0.1.1</version>
         </dependency>
-        <!-- SpringBoot 拦截器 -->
-        <dependency>
-            <groupId>org.springframework.boot</groupId>
-            <artifactId>spring-boot-starter-aop</artifactId>
-        </dependency>
-
         <!-- <dependency>
              <groupId>com.inscloudtech</groupId>
              <artifactId>config-zk</artifactId>
diff --git a/example/src/main/java/com/inscloudtech/alog/clientdemo/aspectj/ActionLogAspect.java b/example/src/main/java/com/inscloudtech/alog/clientdemo/aspectj/ActionLogAspect.java
index 6ca679d..8fbd299 100644
--- a/example/src/main/java/com/inscloudtech/alog/clientdemo/aspectj/ActionLogAspect.java
+++ b/example/src/main/java/com/inscloudtech/alog/clientdemo/aspectj/ActionLogAspect.java
@@ -55,6 +55,13 @@ public class ActionLogAspect {
         } catch (Exception e) {
             throw e;
         }
+
+
+        //当操作行为是新增数据时，需要上报数据业务id
+        if(actionLog.businessType().equals(BusinessType.INSERT)){
+            setBusinessId(args,actionLog,actionLogMessage);
+        }
+
         // 记录结束时间
         long endTime = System.currentTimeMillis();
         // 计算响应时间
@@ -139,6 +146,36 @@ public class ActionLogAspect {
     }
 
 
+    private void setBusinessId(Object[] args,ActionLog actionLog,ActionLogMessage actionLogMessage){
+        Object arg = args[0];
+        Class<?> entityClass = actionLog.entityClass();
+        //获取tableId字段
+        String tableIdField = actionLog.idFieldName();
+//        String tableIdField = "";
+//        Field[] fields = entityClass.getDeclaredFields();
+//        for(Field field : fields) {
+//            field.setAccessible(true);
+//            if (field.isAnnotationPresent(TableId.class)) {
+//                tableIdField = field.getName();
+//                break;
+//            }
+//        }
+
+
+        try {
+            // 将 arg 转换为 entityClass 类型的对象
+            Object afterEntity = entityClass.cast(arg);
+            // 获取 tableIdField 对应的 Field 对象
+            Field idField = entityClass.getDeclaredField(tableIdField);
+            idField.setAccessible(true);
+            // 获取 id 的值
+            Long businessId = (Long)idField.get(afterEntity);
+            actionLogMessage.setBusinessId(businessId);
+        } catch (IllegalAccessException | NoSuchFieldException e) {
+            e.printStackTrace();
+        }
+
+    }
 
 
 
diff --git a/example/src/main/java/com/inscloudtech/alog/clientdemo/demo/controller/DemoUserController.java b/example/src/main/java/com/inscloudtech/alog/clientdemo/demo/controller/DemoUserController.java
index adc4382..11d9cd9 100644
--- a/example/src/main/java/com/inscloudtech/alog/clientdemo/demo/controller/DemoUserController.java
+++ b/example/src/main/java/com/inscloudtech/alog/clientdemo/demo/controller/DemoUserController.java
@@ -54,10 +54,11 @@ public class DemoUserController {
      * @param
      * @return Response
      */
-    @ActionLog(businessName = "用户管理", businessType = BusinessType.INSERT)
+    @ActionLog(businessName = "用户管理", businessType = BusinessType.INSERT,idFieldName = "userId",entityClass = DemoUser.class)
     @PostMapping
     public Response save(@RequestBody DemoUser demoUser) {
-        return new Response(demoUserService.save(demoUser));
+        demoUserService.save(demoUser);
+        return new Response(0,"新增成功！");
     }