From 3da20f9c9db78916d3f02f235e89bd3e87050290 Mon Sep 17 00:00:00 2001
From: guo <guo@guo>
Date: Tue, 2 Apr 2024 11:13:17 +0800
Subject: [PATCH] =?UTF-8?q?=E5=88=86=E9=A1=B5-bug=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/zhgd/config/DataSourceOneConfig.java | 5 +++++
 src/main/java/com/zhgd/config/DataSourceTwoConfig.java | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/src/main/java/com/zhgd/config/DataSourceOneConfig.java b/src/main/java/com/zhgd/config/DataSourceOneConfig.java
index 6bb6f82d7..fcb59695e 100644
--- a/src/main/java/com/zhgd/config/DataSourceOneConfig.java
+++ b/src/main/java/com/zhgd/config/DataSourceOneConfig.java
@@ -2,6 +2,7 @@ package com.zhgd.config;
 
 import com.baomidou.mybatisplus.core.MybatisConfiguration;
 import com.baomidou.mybatisplus.extension.spring.MybatisSqlSessionFactoryBean;
+import com.wflow.config.MyBatisPlusConfig;
 import org.apache.ibatis.logging.stdout.StdOutImpl;
 import org.apache.ibatis.session.SqlSessionFactory;
 import org.mybatis.spring.SqlSessionTemplate;
@@ -16,11 +17,14 @@ import org.springframework.context.annotation.Primary;
 import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
 import org.springframework.jdbc.datasource.DataSourceTransactionManager;
 
+import javax.annotation.Resource;
 import javax.sql.DataSource;
 
 @Configuration
 @MapperScan(basePackages = {"com.zhgd.*.*.*.*.mapper", "com.zhgd.*.*.*.mapper"}, sqlSessionTemplateRef = "db1SqlSessionTemplate")
 public class DataSourceOneConfig {
+    @Resource
+    private MybatisPlusConfig mybatisPlusConfig;
 
     @Bean
     @ConfigurationProperties(prefix = "spring.datasource.db1")
@@ -39,6 +43,7 @@ public class DataSourceOneConfig {
         configuration.setLogImpl(StdOutImpl.class);
         bean.setConfiguration(configuration);
         bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:com/zhgd/xmgl/**/*.xml"));
+        bean.setPlugins(mybatisPlusConfig.mybatisPlusInterceptor());
         return bean.getObject();
     }
 
diff --git a/src/main/java/com/zhgd/config/DataSourceTwoConfig.java b/src/main/java/com/zhgd/config/DataSourceTwoConfig.java
index 31a747580..5b06a0db6 100644
--- a/src/main/java/com/zhgd/config/DataSourceTwoConfig.java
+++ b/src/main/java/com/zhgd/config/DataSourceTwoConfig.java
@@ -15,11 +15,14 @@ import org.springframework.context.annotation.Primary;
 import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
 import org.springframework.jdbc.datasource.DataSourceTransactionManager;
 
+import javax.annotation.Resource;
 import javax.sql.DataSource;
 
 @Configuration
 @MapperScan(basePackages = "com.wflow.mapper", sqlSessionTemplateRef = "sqlSessionTemplate2")
 public class DataSourceTwoConfig {
+    @Resource
+    private MybatisPlusConfig mybatisPlusConfig;
 
     @Primary
     @Bean(name = "dataSource2")
@@ -39,6 +42,7 @@ public class DataSourceTwoConfig {
         configuration.setLogImpl(StdOutImpl.class);
         bean.setConfiguration(configuration);
         bean.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath*:mapper/*.xml"));
+        bean.setPlugins(mybatisPlusConfig.mybatisPlusInterceptor());
         return bean.getObject();
     }