From ac4341ea6b66ae02427d39d35f41d42d78b2eb2e Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期五, 14 二月 2025 10:08:32 +0800
Subject: [PATCH] #

---
 zy-asrs-wms/src/main/java/com/zy/asrs/wms/common/config/MybatisPlusConfig.java |   28 ++++++++++++++++++++++++++--
 1 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/common/config/MybatisPlusConfig.java b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/common/config/MybatisPlusConfig.java
index d379c67..a586e2d 100644
--- a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/common/config/MybatisPlusConfig.java
+++ b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/common/config/MybatisPlusConfig.java
@@ -1,10 +1,14 @@
 package com.zy.asrs.wms.common.config;
 
+import com.baomidou.mybatisplus.autoconfigure.ConfigurationCustomizer;
+import com.baomidou.mybatisplus.extension.MybatisMapWrapperFactory;
 import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
 import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
 import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
 import com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor;
+import com.zy.asrs.wms.common.interceptor.DataChangeInterceptor;
 import com.zy.asrs.wms.system.entity.User;
+import com.zy.asrs.wms.system.entity.UserLogin;
 import net.sf.jsqlparser.expression.Expression;
 import net.sf.jsqlparser.expression.LongValue;
 import net.sf.jsqlparser.expression.NullValue;
@@ -12,7 +16,6 @@
 import org.springframework.context.annotation.Configuration;
 import org.springframework.security.core.Authentication;
 import org.springframework.security.core.context.SecurityContextHolder;
-
 import java.util.Arrays;
 
 /**
@@ -45,7 +48,14 @@
                 return Arrays.asList(
                         "sys_host",
                         "sys_user_role",
-                        "sys_role_menu"
+                        "sys_role_menu",
+                        "sys_language",
+                        "man_loc_sts",
+                        "man_task_sts",
+                        "man_task_type",
+                        "man_order_settle",
+                        "view_man_loc_detl_field",
+                        "sys_cache_statistics"
                 ).contains(tableName);
             }
 
@@ -55,6 +65,7 @@
 
         PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
         interceptor.addInnerInterceptor(paginationInnerInterceptor);
+
 
         return interceptor;
     }
@@ -72,6 +83,9 @@
                 if (object instanceof User) {
                     return new LongValue(((User) object).getHostId());
                 }
+                if (object instanceof UserLogin) {
+                    return new LongValue(((UserLogin) object).getHostId());
+                }
             }
         } catch (Exception e) {
             System.out.println(e.getMessage());
@@ -79,4 +93,14 @@
         return new NullValue();
     }
 
+    @Bean
+    public ConfigurationCustomizer mybatisConfigurationCustomizer() {
+        return configuration -> configuration.setObjectWrapperFactory(new MybatisMapWrapperFactory());
+    }
+
+    @Bean
+    public DataChangeInterceptor sqlStatementInterceptor() {
+        return new DataChangeInterceptor();
+    }
+
 }

--
Gitblit v1.9.1