From c01d3c611e28cd1351e292fa92524e8261de9b32 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期五, 16 二月 2024 22:25:16 +0800
Subject: [PATCH] #

---
 /dev/null                                                                       |   47 -----------
 zy-asrs-openapi/src/main/java/com/zy/asrs/openapi/config/MybatisPlusConfig.java |   28 +++++++
 zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/common/config/MybatisPlusConfig.java  |   82 ++++++++++++++++++++
 zy-asrs-wms/src/main/java/com/zy/asrs/wms/config/MybatisPlusConfig.java         |   28 +++++++
 4 files changed, 138 insertions(+), 47 deletions(-)

diff --git a/zy-asrs-common/src/main/java/com/zy/asrs/common/config/MybatisPlusConfig.java b/zy-asrs-common/src/main/java/com/zy/asrs/common/config/MybatisPlusConfig.java
deleted file mode 100644
index 05814ee..0000000
--- a/zy-asrs-common/src/main/java/com/zy/asrs/common/config/MybatisPlusConfig.java
+++ /dev/null
@@ -1,47 +0,0 @@
-package com.zy.asrs.common.config;
-
-import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
-import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
-import org.springframework.context.annotation.Bean;
-import org.springframework.context.annotation.Configuration;
-
-/**
- * MybatisPlus閰嶇疆
- *
- * @author vincent
- * @since 2018-02-22 11:29:28
- */
-@Configuration
-public class MybatisPlusConfig {
-
-    @Bean
-    public MybatisPlusInterceptor mybatisPlusInterceptor() {
-        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
-
-//        // 澶氱鎴锋彃浠堕厤缃�
-//        TenantLineHandler tenantLineHandler = new TenantLineHandler() {
-//            @Override
-//            public Expression getTenantId() {
-//                return getLoginUserTenantId();
-//            }
-//
-//            @Override
-//            public boolean ignoreTable(String tableName) {
-//                return Arrays.asList(
-//                        "sys_tenant",
-//                        "sys_dictionary",
-//                        "sys_dictionary_data"
-//                ).contains(tableName);
-//            }
-//        };
-//        TenantLineInnerInterceptor tenantLineInnerInterceptor = new TenantLineInnerInterceptor(tenantLineHandler);
-//        interceptor.addInnerInterceptor(tenantLineInnerInterceptor);
-
-        // 鍒嗛〉鎻掍欢閰嶇疆
-        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
-        interceptor.addInnerInterceptor(paginationInnerInterceptor);
-
-        return interceptor;
-    }
-
-}
diff --git a/zy-asrs-openapi/src/main/java/com/zy/asrs/openapi/config/MybatisPlusConfig.java b/zy-asrs-openapi/src/main/java/com/zy/asrs/openapi/config/MybatisPlusConfig.java
new file mode 100644
index 0000000..9543297
--- /dev/null
+++ b/zy-asrs-openapi/src/main/java/com/zy/asrs/openapi/config/MybatisPlusConfig.java
@@ -0,0 +1,28 @@
+package com.zy.asrs.openapi.config;
+
+import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * MybatisPlus閰嶇疆
+ *
+ * @author vincent
+ * @since 2018-02-22 11:29:28
+ */
+@Configuration
+public class MybatisPlusConfig {
+
+    @Bean
+    public MybatisPlusInterceptor mybatisPlusInterceptor() {
+        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
+
+        // 鍒嗛〉鎻掍欢閰嶇疆
+        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
+        interceptor.addInnerInterceptor(paginationInnerInterceptor);
+
+        return interceptor;
+    }
+
+}
diff --git a/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/common/config/MybatisPlusConfig.java b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/common/config/MybatisPlusConfig.java
new file mode 100644
index 0000000..cbe345a
--- /dev/null
+++ b/zy-asrs-wcs/src/main/java/com/zy/asrs/wcs/common/config/MybatisPlusConfig.java
@@ -0,0 +1,82 @@
+package com.zy.asrs.wcs.common.config;
+
+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.wcs.sys.entity.User;
+import net.sf.jsqlparser.expression.Expression;
+import net.sf.jsqlparser.expression.LongValue;
+import net.sf.jsqlparser.expression.NullValue;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+
+import java.util.Arrays;
+
+/**
+ * MybatisPlus閰嶇疆
+ *
+ * @author vincent
+ * @since 2018-02-22 11:29:28
+ */
+@Configuration
+public class MybatisPlusConfig {
+
+    @Bean
+    public MybatisPlusInterceptor mybatisPlusInterceptor() {
+        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
+
+        TenantLineHandler tenantLineHandler = new TenantLineHandler() {
+
+            @Override
+            public Expression getTenantId() {
+                return getLoginUserTenantId();
+            }
+
+            @Override
+            public String getTenantIdColumn() {
+                return "host_id";
+            }
+
+            @Override
+            public boolean ignoreTable(String tableName) {
+                return Arrays.asList(
+                        "sys_host",
+                        "sys_user_role",
+                        "sys_role_menu"
+                ).contains(tableName);
+            }
+
+        };
+        TenantLineInnerInterceptor tenantLineInnerInterceptor = new TenantLineInnerInterceptor(tenantLineHandler);
+        interceptor.addInnerInterceptor(tenantLineInnerInterceptor);
+
+        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
+        interceptor.addInnerInterceptor(paginationInnerInterceptor);
+
+        return interceptor;
+    }
+
+    /**
+     * 鑾峰彇褰撳墠鐧诲綍鐢ㄦ埛鐨勭鎴穒d
+     *
+     * @return Integer
+     */
+    public Expression getLoginUserTenantId() {
+        try {
+            Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+            if (authentication != null) {
+                Object object = authentication.getPrincipal();
+                if (object instanceof User) {
+                    return new LongValue(((User) object).getHostId());
+                }
+            }
+        } catch (Exception e) {
+            System.out.println(e.getMessage());
+        }
+        return new NullValue();
+    }
+
+}
diff --git a/zy-asrs-wms/src/main/java/com/zy/asrs/wms/config/MybatisPlusConfig.java b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/config/MybatisPlusConfig.java
new file mode 100644
index 0000000..ed31864
--- /dev/null
+++ b/zy-asrs-wms/src/main/java/com/zy/asrs/wms/config/MybatisPlusConfig.java
@@ -0,0 +1,28 @@
+package com.zy.asrs.wms.config;
+
+import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
+import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * MybatisPlus閰嶇疆
+ *
+ * @author vincent
+ * @since 2018-02-22 11:29:28
+ */
+@Configuration
+public class MybatisPlusConfig {
+
+    @Bean
+    public MybatisPlusInterceptor mybatisPlusInterceptor() {
+        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
+
+        // 鍒嗛〉鎻掍欢閰嶇疆
+        PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor();
+        interceptor.addInnerInterceptor(paginationInnerInterceptor);
+
+        return interceptor;
+    }
+
+}

--
Gitblit v1.9.1