admin/zhyc-sheep
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(sys | login): 多租户模式支持

Browse Source 
+ 两个业务通过用户ID分配数据源
This commit is contained in:
HashMap
2026-01-27 13:15:48 +08:00
parent 9c678c84bf
commit 3bff858643
3 changed files with 41 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
zhyc-admin/src/main/java/com/zhyc/Routing/Filter/DataSourceRoutingFilter.java
View File

@@ -3,6 +3,7 @@ package com.zhyc.Routing.Filter;
import com.zhyc.framework.config.routing.DataSourceKeys;
import com.zhyc.framework.datasource.DynamicDataSourceContextHolder;
import lombok.extern.slf4j.Slf4j;
import org.bytedeco.opencv.presets.opencv_core;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;
@@ -22,13 +23,18 @@ public class DataSourceRoutingFilter extends OncePerRequestFilter {
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
String uri = request.getRequestURI();
String username = request.getRemoteUser();
log.debug("username:{}",username);
try {
if (uri.startsWith("/app") || uri.startsWith("/base") || uri.startsWith("/biosafety") || uri.startsWith("/common")
|| uri.startsWith("/dairyProducts") || uri.startsWith("/enums") || uri.startsWith("/feed") || uri.startsWith("/frozen") || uri.startsWith("/produce")
|| uri.startsWith("sale") || uri.startsWith("/stock") || uri.startsWith("/work") || uri.startsWith("/sheepfold_management")) {
log.debug("业务请求 : BUSINESS : {}", uri);
DynamicDataSourceContextHolder.setDataSourceType(DataSourceKeys.FARM_1);
if (username.equals("admin")) {
DynamicDataSourceContextHolder.setDataSourceType(DataSourceKeys.FARM_1);
}else if (username.equals("ry")) {
DynamicDataSourceContextHolder.setDataSourceType(DataSourceKeys.FARM_2);
}
} else {
log.debug("系统请求 ： SYSTEM : {}", uri);
DynamicDataSourceContextHolder.setDataSourceType(DataSourceKeys.SYSTEM);

4
zhyc-admin/src/main/resources/application-druid.yml
View File

@@ -18,6 +18,10 @@ spring:
url: jdbc:mysql://118.182.97.76:3306/zhyc_sheep01?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8
username: zhyc
password: yszh123
farm02:
url: jdbc:mysql://118.182.97.76:3306/zhyc_sheep02?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8
username: zhyc
password: yszh123
# 初始连接数
initialSize: 5
# 最小连接池数量

60
zhyc-framework/src/main/java/com/zhyc/framework/config/DruidConfig.java
View File

@@ -27,19 +27,18 @@ import com.zhyc.framework.datasource.DynamicDataSource;
/**
* druid 配置多数据源
*
*
* @author ruoyi
*/
@Configuration
public class DruidConfig
{
public class DruidConfig {
@Bean
@ConfigurationProperties("spring.datasource.druid.master")
public DataSource masterDataSource(DruidProperties druidProperties)
{
public DataSource masterDataSource(DruidProperties druidProperties) {
DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
return druidProperties.dataSource(dataSource);
}
@Bean
@ConfigurationProperties("spring.datasource.druid.farm01")
public DataSource farm1DataSource(DruidProperties druidProperties) {
@@ -47,13 +46,17 @@ public class DruidConfig
return druidProperties.dataSource(dataSource);
}
@Bean
@ConfigurationProperties("spring.datasource.druid.farm02")
public DataSource farm2DataSource(DruidProperties druidProperties) {
DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
return druidProperties.dataSource(dataSource);
}
@Bean
@ConfigurationProperties("spring.datasource.druid.slave")
@ConditionalOnProperty(prefix = "spring.datasource.druid.slave", name = "enabled", havingValue = "true")
public DataSource slaveDataSource(DruidProperties druidProperties)
{
public DataSource slaveDataSource(DruidProperties druidProperties) {
DruidDataSource dataSource = DruidDataSourceBuilder.create().build();
return druidProperties.dataSource(dataSource);
}
@@ -62,43 +65,40 @@ public class DruidConfig
@Primary
public DataSource dynamicDataSource(
@Qualifier("masterDataSource") DataSource systemDataSource,
@Qualifier("farm1DataSource") DataSource farm1DataSource) {
@Qualifier("farm1DataSource") DataSource farm1DataSource,
@Qualifier("farm2DataSource") DataSource farm2DataSource) {
// 存储数据源路由
Map<Object, Object> targetDataSources = new HashMap<>();
targetDataSources.put("system", systemDataSource);
targetDataSources.put("farm01", farm1DataSource);
targetDataSources.put("farm02", farm2DataSource);
return new DynamicDataSource(systemDataSource, targetDataSources);
}
/**
* 设置数据源
*
*
* @param targetDataSources 备选数据源集合
* @param sourceName 数据源名称
* @param beanName bean名称
* @param sourceName 数据源名称
* @param beanName bean名称
*/
public void setDataSource(Map<Object, Object> targetDataSources, String sourceName, String beanName)
{
try
{
public void setDataSource(Map<Object, Object> targetDataSources, String sourceName, String beanName) {
try {
DataSource dataSource = SpringUtils.getBean(beanName);
targetDataSources.put(sourceName, dataSource);
}
catch (Exception e)
{
} catch (Exception e) {
}
}
/**
* 去除监控页面底部的广告
*/
@SuppressWarnings({ "rawtypes", "unchecked" })
@SuppressWarnings({"rawtypes", "unchecked"})
@Bean
@ConditionalOnProperty(name = "spring.datasource.druid.statViewServlet.enabled", havingValue = "true")
public FilterRegistrationBean removeDruidFilterRegistrationBean(DruidStatProperties properties)
{
public FilterRegistrationBean removeDruidFilterRegistrationBean(DruidStatProperties properties) {
// 获取web监控页面的参数
DruidStatProperties.StatViewServlet config = properties.getStatViewServlet();
// 提取common.js的配置路径
@@ -106,16 +106,14 @@ public class DruidConfig
String commonJsPattern = pattern.replaceAll("\\*", "js/common.js");
final String filePath = "support/http/resources/js/common.js";
// 创建filter进行过滤
Filter filter = new Filter()
{
Filter filter = new Filter() {
@Override
public void init(javax.servlet.FilterConfig filterConfig) throws ServletException
{
public void init(javax.servlet.FilterConfig filterConfig) throws ServletException {
}
@Override
public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
throws IOException, ServletException
{
throws IOException, ServletException {
chain.doFilter(request, response);
// 重置缓冲区，响应头不会被重置
response.resetBuffer();
@@ -126,9 +124,9 @@ public class DruidConfig
text = text.replaceAll("powered.*?shrek.wang</a>", "");
response.getWriter().write(text);
}
@Override
public void destroy()
{
public void destroy() {
}
};
FilterRegistrationBean registrationBean = new FilterRegistrationBean();