From 791b8c88d67767c9847c7c052faca3e65ec36016 Mon Sep 17 00:00:00 2001
From: zhaowenxuan <chacca165@163.com>
Date: 星期五, 07 二月 2025 13:59:27 +0800
Subject: [PATCH] 配置动态切库类以及拦截器切库
---
ltkj-hosp/src/main/java/com/ltkj/db/DataSourceConfig.java | 73 +++++++++++++++---------
ltkj-hosp/src/main/resources/mapper/hosp/DictHospMapper.xml | 8 ++
ltkj-hosp/src/main/java/com/ltkj/db/HospDynamicDataSource.java | 9 ++-
ltkj-hosp/src/main/java/com/ltkj/hosp/domain/DictHosp.java | 1
ltkj-framework/src/main/java/com/ltkj/framework/interceptor/DBChangeInterceptor.java | 38 ++++++++++++
ltkj-framework/src/main/java/com/ltkj/framework/config/WebConfig.java | 9 +++
6 files changed, 103 insertions(+), 35 deletions(-)
diff --git a/ltkj-framework/src/main/java/com/ltkj/framework/config/WebConfig.java b/ltkj-framework/src/main/java/com/ltkj/framework/config/WebConfig.java
index 1d9e1fd..cefe709 100644
--- a/ltkj-framework/src/main/java/com/ltkj/framework/config/WebConfig.java
+++ b/ltkj-framework/src/main/java/com/ltkj/framework/config/WebConfig.java
@@ -1,5 +1,7 @@
package com.ltkj.framework.config;
+import com.ltkj.framework.interceptor.DBChangeInterceptor;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
@@ -11,6 +13,10 @@
//瀹氫箟鎷︽埅鍣�
@Configuration
public class WebConfig implements WebMvcConfigurer {
+
+ @Autowired
+ private DBChangeInterceptor dbChangeInterceptor;
+
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new TokenInterceptor())
@@ -24,5 +30,8 @@
"/lis/**","/api/His/**","/pacs/**",
"/callBack/**"
});
+
+ registry.addInterceptor(dbChangeInterceptor)
+ .addPathPatterns("/**");
}
}
diff --git a/ltkj-framework/src/main/java/com/ltkj/framework/interceptor/DBChangeInterceptor.java b/ltkj-framework/src/main/java/com/ltkj/framework/interceptor/DBChangeInterceptor.java
index 0a04013..006b157 100644
--- a/ltkj-framework/src/main/java/com/ltkj/framework/interceptor/DBChangeInterceptor.java
+++ b/ltkj-framework/src/main/java/com/ltkj/framework/interceptor/DBChangeInterceptor.java
@@ -1,30 +1,64 @@
package com.ltkj.framework.interceptor;
+import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.google.gson.Gson;
+import com.ltkj.db.DataSourceConfig;
import com.ltkj.db.DataSourceContextHolder;
+import com.ltkj.hosp.domain.DictHosp;
+import com.ltkj.hosp.service.IDictHospService;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.annotation.Order;
+import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
/**
* @Company: 瑗垮畨璺嘲绉戞妧鏈夐檺鍏徃
* @Author: zhaowenxuan
* @Date: 2025/2/7 11:07
*/
+@Component
@Order(2)
public class DBChangeInterceptor implements HandlerInterceptor {
- @Override
- public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
+ @Autowired
+ private IDictHospService dictHospService;
+ @Autowired
+ private DataSourceConfig dataSourceConfig;
+
+ @Override
+ public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws IOException {
+ String hospId = request.getHeader("hospId");
+ if (StrUtil.isBlank(hospId)){
+ response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
+ response.setContentType("application/json;charset=UTF-8");
+ response.getWriter().write("{\"message\":\"401锛氭壘涓嶅埌闄㈠尯缂栧彿\"}");
+ return false;
+ }
+ DataSourceContextHolder.setDataSourceKey("default");
+ LambdaQueryWrapper<DictHosp> wrapper = new LambdaQueryWrapper<>();
+ wrapper.eq(DictHosp::getCode,hospId);
+ DictHosp hosp = dictHospService.getOne(wrapper);
+ if (hosp == null || hosp.getDatabase() == null || hosp.getDatabase().trim().isEmpty()) {
+ response.setStatus(HttpServletResponse.SC_UNAUTHORIZED);
+ response.setContentType("application/json;charset=UTF-8");
+ response.getWriter().write("{\"message\":\"401锛氭壘涓嶅埌闄㈠尯鏁版嵁\"}");
+ return false;
+ }
+ dataSourceConfig.addDataSource(hosp.getDatabase());
+ DataSourceContextHolder.setDataSourceKey(hosp.getDatabase());
return true;
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
DataSourceContextHolder.setDataSourceKey("default");
+ DataSourceContextHolder.clear();
HandlerInterceptor.super.afterCompletion(request, response, handler, ex);
}
}
diff --git a/ltkj-hosp/src/main/java/com/ltkj/db/DataSourceConfig.java b/ltkj-hosp/src/main/java/com/ltkj/db/DataSourceConfig.java
index 9d24863..6656298 100644
--- a/ltkj-hosp/src/main/java/com/ltkj/db/DataSourceConfig.java
+++ b/ltkj-hosp/src/main/java/com/ltkj/db/DataSourceConfig.java
@@ -1,47 +1,49 @@
package com.ltkj.db;
import com.alibaba.druid.pool.DruidDataSource;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
import javax.sql.DataSource;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
+import java.util.Properties;
-//@Configuration
+@Configuration
public class DataSourceConfig {
-
- // 浣跨敤浠ｇ爜
-// try {
-// long id = Long.parseLong(request.getAttribute("id").toString());
-// DataSourceContextHolder.setDataSourceKey(String.valueOf(id));
-// dataSourceConfig.addDataSource(String.valueOf(id)); // 纭繚鏁版嵁婧愬凡娉ㄥ唽
-// if (log.getSize()>50) log.setSize(50);
-// if (log.getLang().equals("zh-cn")) log.setLang("zh");
-// return serverLogService.getLog(log,id);
-// }finally {
-// DataSourceContextHolder.setDataSourceKey("default"); // 鍒囨崲鍥炰富搴�
-// }
private static final String DEFAULT_DATA_SOURCE_KEY = "default"; // 涓诲簱鐨勬爣璇�
private final Map<String, DataSource> dataSourceCache = new HashMap<>(); // 鏁版嵁婧愮紦瀛�
+ @Value("${config.properties}")
+ private String url;
+
+ @Value("${config.path}")
+ private String path;
+
// 浠� application.yml 涓鍙栦富搴撶殑閰嶇疆
- @Value("${spring.datasource.url}")
+// @Value("${spring.datasource.url}")
private String primaryUrl;
- @Value("${spring.datasource.username}")
+ private String primaryPort;
+
+ // @Value("${spring.datasource.username}")
private String primaryUsername;
- @Value("${spring.datasource.password}")
+ // @Value("${spring.datasource.password}")
private String primaryPassword;
- @Value("${dbUrl}")
+ // @Value("${dbUrl}")
private String dbUrl;
- @Bean
- public DataSource dynamicDataSource() {
- DynamicDataSource dynamicDataSource = new DynamicDataSource();
+ @Bean(name = "hospDynamicDataSources")
+ public DataSource hospDynamicDataSources() {
+ HospDynamicDataSource dynamicDataSource = new HospDynamicDataSource();
// 鍒濆鍖栭粯璁ゆ暟鎹簮涓轰富搴�
dynamicDataSource.addTargetDataSource(DEFAULT_DATA_SOURCE_KEY, createDataSource(primaryUrl, primaryUsername, primaryPassword));
@@ -60,15 +62,30 @@
}
// 鏍规嵁鐢ㄦ埛 ID 鍔ㄦ�佽幏鍙栨暟鎹簮
- public void addDataSource(String userId) {
- String dbName = "scum_admin_user_" + userId;
+ public void addDataSource(String dbName) {
+ FileInputStream fis = null;
+ Properties props = new Properties();
+ try {
+ fis = new FileInputStream(url);
+ props.load(fis);
+ fis.close();
+ dbUrl = props.getProperty("ip");
+ primaryPassword = props.getProperty("password");
+ primaryPort = props.getProperty("prot");
+ primaryUsername = props.getProperty("username");
+ } catch (IOException e) {
+ throw new RuntimeException("璇诲彇閰嶇疆鏂囦欢澶辫触", e);
+ }
+
// 妫�鏌ョ紦瀛樹腑鏄惁宸茬粡瀛樺湪璇ユ暟鎹簮
- if (!dataSourceCache.containsKey(userId)) {
- String url = "jdbc:mysql://"+dbUrl+":3306/"+dbName;
- DataSource dataSource = createDataSource(url, primaryUsername, primaryPassword);
- dataSourceCache.put(userId, dataSource);
- DynamicDataSource dynamicDataSource = (DynamicDataSource) dynamicDataSource();
- dynamicDataSource.addTargetDataSource(userId, dataSource);
+ if (!dataSourceCache.containsKey(dbName)) {
+ synchronized (this) {
+ String url = "jdbc:mysql://" + dbUrl + ":" + primaryPort + "/" + dbName;
+ DataSource dataSource = createDataSource(url, primaryUsername, primaryPassword);
+ dataSourceCache.put(dbName, dataSource);
+ HospDynamicDataSource dynamicDataSource = (HospDynamicDataSource) hospDynamicDataSources();
+ dynamicDataSource.addTargetDataSource(dbName, dataSource);
+ }
}
}
}
diff --git a/ltkj-hosp/src/main/java/com/ltkj/db/DynamicDataSource.java b/ltkj-hosp/src/main/java/com/ltkj/db/HospDynamicDataSource.java
similarity index 67%
rename from ltkj-hosp/src/main/java/com/ltkj/db/DynamicDataSource.java
rename to ltkj-hosp/src/main/java/com/ltkj/db/HospDynamicDataSource.java
index 8d6bb07..1cd8baa 100644
--- a/ltkj-hosp/src/main/java/com/ltkj/db/DynamicDataSource.java
+++ b/ltkj-hosp/src/main/java/com/ltkj/db/HospDynamicDataSource.java
@@ -1,14 +1,17 @@
package com.ltkj.db;
+import org.springframework.context.annotation.Bean;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
+import org.springframework.stereotype.Component;
import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
-public class DynamicDataSource extends AbstractRoutingDataSource {
+public class HospDynamicDataSource extends AbstractRoutingDataSource {
- private final Map<Object, Object> targetDataSources = new HashMap<>(); // 瀛樺偍鎵�鏈夋暟鎹簮
+ private final Map<Object, Object> targetDataSources = new ConcurrentHashMap<>(); // 瀛樺偍鎵�鏈夋暟鎹簮
@Override
protected Object determineCurrentLookupKey() {
@@ -18,7 +21,7 @@
// 娣诲姞鐩爣鏁版嵁婧�
public void addTargetDataSource(String key, DataSource dataSource) {
targetDataSources.put(key, dataSource);
- super.setTargetDataSources(targetDataSources); // 鏇存柊鐩爣鏁版嵁婧�
+ super.setTargetDataSources(new ConcurrentHashMap<>(targetDataSources)); // 鏇存柊鐩爣鏁版嵁婧�
super.afterPropertiesSet(); // 閲嶆柊鍒濆鍖栨暟鎹簮
}
diff --git a/ltkj-hosp/src/main/java/com/ltkj/hosp/domain/DictHosp.java b/ltkj-hosp/src/main/java/com/ltkj/hosp/domain/DictHosp.java
index c1de204..cda83e6 100644
--- a/ltkj-hosp/src/main/java/com/ltkj/hosp/domain/DictHosp.java
+++ b/ltkj-hosp/src/main/java/com/ltkj/hosp/domain/DictHosp.java
@@ -206,6 +206,7 @@
.append("updateByName", getUpdateByName())
.append("imgbase64", getImgbase64())
.append("deleted", getDeleted())
+ .append("database", getDatabase())
.toString();
}
}
diff --git a/ltkj-hosp/src/main/resources/mapper/hosp/DictHospMapper.xml b/ltkj-hosp/src/main/resources/mapper/hosp/DictHospMapper.xml
index 6ad3365..ebf675b 100644
--- a/ltkj-hosp/src/main/resources/mapper/hosp/DictHospMapper.xml
+++ b/ltkj-hosp/src/main/resources/mapper/hosp/DictHospMapper.xml
@@ -31,6 +31,7 @@
<result property="updateByName" column="update_by_name"/>
<result property="imgbase64" column="imgBase64"/>
<result property="deleted" column="deleted"/>
+ <result property="database" column="database"/>
</resultMap>
<sql id="selectDictHospVo">
@@ -59,7 +60,7 @@
create_by_name,
update_by_name,
imgBase64,
- deleted
+ deleted,database
from dict_hosp
</sql>
@@ -116,6 +117,7 @@
<if test="updateByName != null and updateByName != ''">update_by_name,</if>
<if test="imgbase64 != null">imgBase64,</if>
<if test="deleted != null">deleted,</if>
+ <if test="database != null">database,</if>
</trim>
<trim prefix="values (" suffix=")" suffixOverrides=",">
<if test="hospAreaId != null">#{hospAreaId},</if>
@@ -144,6 +146,7 @@
<if test="updateByName != null and updateByName != ''">#{updateByName},</if>
<if test="imgbase64 != null">#{imgbase64},</if>
<if test="deleted != null">#{deleted},</if>
+ <if test="database != null">#{database},</if>
</trim>
</insert>
@@ -175,6 +178,7 @@
<if test="updateByName != null and updateByName != ''">update_by_name = #{updateByName},</if>
<if test="imgbase64 != null">imgBase64 = #{imgbase64},</if>
<if test="deleted != null">deleted = #{deleted},</if>
+ <if test="database != null">database = #{database},</if>
</trim>
where hosp_area_id = #{hospAreaId}
</update>
@@ -191,4 +195,4 @@
#{hospAreaId}
</foreach>
</delete>
-</mapper>
\ No newline at end of file
+</mapper>
--
Gitblit v1.8.0