feat:多数据源模块

master
hbr 2024-06-05 20:44:52 +08:00
parent d89b7ab7ce
commit 931c5a4634
21 changed files with 206 additions and 135 deletions

View File

@ -0,0 +1,71 @@
package com.zhiLian.datasource;
import com.zhiLian.datasource.config.factory.DruidDataSourceFactory;
import com.zhiLian.datasource.config.role.DynamicDataSource;
import com.zhiLian.datasource.domain.DataSourceInfo;
import com.zhiLian.datasource.domain.EntInfo;
import lombok.extern.log4j.Log4j2;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @author DongZl
* @description:
* @Date 2023-8-1 11:05
*/
@Log4j2
@Configuration
public class ManyDataSource {
private List<EntInfo> dataSourceInfoList(){
List<EntInfo> databaseNameList = new ArrayList<>(){{
add(EntInfo.builder()
.entCode("test_1")
.ip("192.168.120.128")
.port(3306)
.build());
add(EntInfo.builder()
.entCode("test_2")
.ip("122.51.111.225")
.port(6666)
.build());
add(EntInfo.builder()
.entCode("test_3")
.ip("122.51.111.225")
.port(3333)
.build());
}};
return databaseNameList;
}
@Bean
@Primary
public DynamicDataSource dynamicDataSource(DruidDataSourceFactory druidDataSourceFactory) {
//查企业
Map<Object, Object> dataSourceMap = new HashMap<>();
dataSourceInfoList()
.stream()
.map(entInfo-> DataSourceInfo.hostAndPortBuild(entInfo.getEntCode(),entInfo.getIp(),entInfo.getPort()))
.forEach(dataSourceInfo -> {
dataSourceMap.put(dataSourceInfo.getKey(), druidDataSourceFactory.create(dataSourceInfo));
});
//设置动态数据源
DynamicDataSource dynamicDataSource = new DynamicDataSource();
// dynamicDataSource.setDefaultTargetDataSource(masterDataSource());
dynamicDataSource.setTargetDataSources(dataSourceMap);
//将数据源信息备份在defineTargetDataSources中
dynamicDataSource.setDefineTargetDataSources(dataSourceMap);
return dynamicDataSource;
}
}

View File

@ -1,7 +1,8 @@
package com.zhiLian.datasource.config;
import com.zhiLian.common.security.utils.SecurityUtils;
import com.zhiLian.datasource.DynamicDataSourceHolder;
import com.zhiLian.datasource.config.holder.DynamicDataSourceHolder;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
@ -17,7 +18,7 @@ import org.springframework.stereotype.Component;
@Component
public class DataSourceAsp {
@Pointcut("execution(public * com.zhiLian.datasource..*Controller.*(..))")
@Pointcut("execution(public * com.zhiLian.vehicle.controller.*Controller.*(..))")
public void pointcut () {
}
@ -30,8 +31,6 @@ public class DataSourceAsp {
Long storeId = SecurityUtils.getLoginUser().getUserid();
DynamicDataSourceHolder.setDynamicDataSourceKey("test_"+storeId);
}
/**
*
*

View File

@ -1,71 +0,0 @@
package com.zhiLian.datasource.config;
import com.alibaba.druid.pool.DruidDataSource;
import com.zhiLian.datasource.DataSourceInfo;
import lombok.extern.log4j.Log4j2;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @author DongZl
* @description:
* @Date 2023-8-1 11:05
*/
@Log4j2
@Configuration
public class DruidConfig {
private List<DataSourceInfo> getDataSourceInfoList(){
List<String> databaseNameList = new ArrayList<>(){{
add("3306");
// add("3307");
}};
return databaseNameList.stream().map(DataSourceInfo::databaseNameBuild).toList();
}
/**
* @Description:
* @Author Dongzl
*/
public DruidDataSource createDataSourceConnection(DataSourceInfo dataSourceInfo) {
DruidDataSource druidDataSource = new DruidDataSource();
druidDataSource.setUrl(dataSourceInfo.getUrl());
druidDataSource.setUsername(dataSourceInfo.getUserName());
druidDataSource.setPassword(dataSourceInfo.getPassword());
druidDataSource.setBreakAfterAcquireFailure(true);
druidDataSource.setConnectionErrorRetryAttempts(0);
try {
druidDataSource.getConnection(2000);
log.info("{} -> 数据源连接成功", dataSourceInfo.getKey());
return druidDataSource;
} catch (SQLException throwables) {
log.error("数据源 {} 连接失败,用户名:{},密码 {}",dataSourceInfo.getUrl(),dataSourceInfo.getUserName(),dataSourceInfo.getPassword());
return null;
}
}
@Bean
@Primary
public DynamicDataSource dynamicDataSource() {
Map<Object, Object> dataSourceMap = new HashMap<>();
getDataSourceInfoList().forEach(dataSourceInfo -> {
dataSourceMap.put(dataSourceInfo.getKey(), createDataSourceConnection(dataSourceInfo));
});
//设置动态数据源
DynamicDataSource dynamicDataSource = new DynamicDataSource();
// dynamicDataSource.setDefaultTargetDataSource(masterDataSource());
dynamicDataSource.setTargetDataSources(dataSourceMap);
//将数据源信息备份在defineTargetDataSources中
dynamicDataSource.setDefineTargetDataSources(dataSourceMap);
return dynamicDataSource;
}
}

View File

@ -0,0 +1,44 @@
package com.zhiLian.datasource.config.factory;
import com.alibaba.druid.pool.DruidDataSource;
import com.zhiLian.datasource.domain.DataSourceInfo;
import lombok.extern.log4j.Log4j2;
import org.springframework.stereotype.Component;
import java.sql.SQLException;
/**
* BingRui.Hou
*
* @Description
* @ClassName DruidDataSourceFactory
* @Date 2024/06/04 14:47
*/
@Log4j2
@Component
public class DruidDataSourceFactory {
/**
* @Description:
* @Author Dongzl
*/
public DruidDataSource create(DataSourceInfo dataSourceInfo) {
DruidDataSource druidDataSource = new DruidDataSource();
druidDataSource.setUrl(dataSourceInfo.getUrl());
druidDataSource.setUsername(dataSourceInfo.getUserName());
druidDataSource.setPassword(dataSourceInfo.getPassword());
druidDataSource.setBreakAfterAcquireFailure(true);
druidDataSource.setConnectionErrorRetryAttempts(0);
try {
druidDataSource.getConnection(2000);
log.info("{} -> 数据源连接成功", dataSourceInfo.getKey());
return druidDataSource;
} catch (SQLException throwables) {
log.error("数据源 {} 连接失败,用户名:{},密码 {}",dataSourceInfo.getUrl(),dataSourceInfo.getUserName(),dataSourceInfo.getPassword());
return null;
}
}
}

View File

@ -1,4 +1,5 @@
package com.zhiLian.datasource;
package com.zhiLian.datasource.config.holder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.Assert;

View File

@ -1,8 +1,10 @@
package com.zhiLian.datasource.config;
package com.zhiLian.datasource.config.role;
import com.zhiLian.datasource.DynamicDataSourceHolder;
import com.zhiLian.datasource.config.holder.DynamicDataSourceHolder;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.NoArgsConstructor;
import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource;
@ -17,8 +19,11 @@ import java.util.Map;
@Data
@AllArgsConstructor
@NoArgsConstructor
@EqualsAndHashCode(callSuper = true)
public class DynamicDataSource extends AbstractRoutingDataSource {
//备份所有数据源信息,
/**
*
*/
private Map<Object, Object> defineTargetDataSources;
/**

View File

@ -1,18 +0,0 @@
package com.zhiLian.datasource.contents;
/**
* @Description:
* @Author Dongzl
* @Date 2022/8/18 16:33
*
*/
public class CommonConstant {
/**
*
*/
public static final String MASTER = "master";
/**
*
*/
public static final String SLAVE = "slave";
}

View File

@ -7,7 +7,7 @@ package com.zhiLian.datasource.contents;
*/
public class DatasourceContent {
public final static String DATASOURCE_URL = "jdbc:mysql://122.51.111.225:{}/zhiLian-cloud?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8";
public final static String DATASOURCE_URL = "jdbc:mysql://{}:{}/zhiLian-cloud?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8";
public final static String USER_NAME = "root";

View File

@ -1,4 +1,4 @@
package com.zhiLian.datasource;
package com.zhiLian.datasource.domain;
import com.zhiLian.common.core.utils.StringUtils;
@ -6,8 +6,10 @@ import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import static com.zhiLian.datasource.contents.DatasourceContent.*;
/**
* @author DongZl
* @description:
@ -19,19 +21,30 @@ import static com.zhiLian.datasource.contents.DatasourceContent.*;
@AllArgsConstructor
public class DataSourceInfo {
/**
*
*/
private String key;
/**
*
*/
private String url;
/**
*
*/
private String userName;
/**
*
*/
private String password;
public static DataSourceInfo databaseNameBuild(String databaseName){
public static DataSourceInfo hostAndPortBuild(String key,String host, Integer port){
return DataSourceInfo.builder()
.key(databaseName)
.url(StringUtils.format(DATASOURCE_URL, databaseName))
.key(key)
.url(StringUtils.format(DATASOURCE_URL, host,port))
.password(PASSWORD)
.userName(USER_NAME)
.build();

View File

@ -0,0 +1,29 @@
package com.zhiLian.datasource.domain;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
/**
* BingRui.Hou
*
* @Description
* @ClassName EntInfo
* @Date 2024/06/04 14:34
*/
@Data
@AllArgsConstructor
@NoArgsConstructor
@Builder
public class EntInfo {
private String entCode;
//ip
private String ip;
//端口
private Integer port;
}

View File

@ -1,9 +1,11 @@
package com.zhiLian.datasource.util;
import com.alibaba.druid.pool.DruidDataSource;
import com.zhiLian.datasource.DataSourceInfo;
import com.zhiLian.datasource.config.DruidConfig;
import com.zhiLian.datasource.config.DynamicDataSource;
import com.zhiLian.datasource.ManyDataSource;
import com.zhiLian.datasource.config.factory.DruidDataSourceFactory;
import com.zhiLian.datasource.config.role.DynamicDataSource;
import com.zhiLian.datasource.domain.DataSourceInfo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
@ -22,13 +24,16 @@ import java.util.Map;
public class DataSourceService {
@Autowired
private DruidConfig druidConfig;
private ManyDataSource druidConfig;
@Autowired
private DruidDataSourceFactory druidDataSourceFactory;
@Resource
private DynamicDataSource dynamicDataSource;
public void addDataSource(DataSourceInfo dataSourceInfo){
addDefineDynamicDataSource(druidConfig.createDataSourceConnection(dataSourceInfo), dataSourceInfo.getKey());
addDefineDynamicDataSource(druidDataSourceFactory.create(dataSourceInfo), dataSourceInfo.getKey());
}
/**

View File

@ -30,7 +30,7 @@ import java.util.List;
public class IotDBSessionConfig {
private static Session session;
private static final String LOCAL_HOST = "111.229.102.61";
private static final String LOCAL_HOST = "192.168.120.128";
@Bean
public Session getSession() throws IoTDBConnectionException, StatementExecutionException {
if (session == null) {

View File

@ -6,7 +6,7 @@ spring:
kafka:
producer:
# Kafka服务器
bootstrap-servers: 122.51.111.225:9092
bootstrap-servers: 192.168.120.128:9092
# 开启事务,必须在开启了事务的方法中发送,否则报错
transaction-id-prefix: kafkaTx-
# 发生错误后消息重发的次数开启事务必须设置大于0。
@ -27,7 +27,7 @@ spring:
consumer:
# Kafka服务器
bootstrap-servers: 122.51.111.225:9092
bootstrap-servers: 192.168.120.128:9092
group-id: firstGroup
# 自动提交的时间间隔 在spring boot 2.X 版本中这里采用的是值的类型为Duration 需要符合特定的格式如1S,1M,2H,5D
#auto-commit-interval: 2s
@ -90,10 +90,10 @@ spring:
nacos:
discovery:
# 服务注册地址
server-addr: 122.51.111.225:8848
server-addr: 192.168.120.128:8848
config:
# 配置中心地址
server-addr: 122.51.111.225:8848
server-addr: 192.168.120.128:8848
# namespace: a9b66e92-e507-47ba-9674-6f939f793aca
# 配置文件格式
file-extension: yml

View File

@ -1,17 +1,15 @@
package com.zhiLian.vehicle.datasource;
import com.alibaba.druid.pool.DruidDataSource;
import com.zhiLian.vehicle.datasource.config.factory.DruidDataSourceFactory;
import com.zhiLian.vehicle.datasource.config.role.DynamicDataSource;
import com.zhiLian.vehicle.datasource.domain.DataSourceInfo;
import com.zhiLian.vehicle.datasource.domain.EntInfo;
import com.zhiLian.vehicle.datasource.config.role.DynamicDataSource;
import lombok.extern.log4j.Log4j2;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
@ -31,9 +29,19 @@ public class ManyDataSource {
List<EntInfo> databaseNameList = new ArrayList<>(){{
add(EntInfo.builder()
.entCode("test_1")
.ip("192.168.120.128")
.port(3306)
.build());
add(EntInfo.builder()
.entCode("test_2")
.ip("122.51.111.225")
.port(6666)
.build());
add(EntInfo.builder()
.entCode("test_3")
.ip("122.51.111.225")
.port(3333)
.build());
}};
return databaseNameList;
}

View File

@ -2,18 +2,11 @@ package com.zhiLian.vehicle.datasource.config.factory;
import com.alibaba.druid.pool.DruidDataSource;
import com.zhiLian.vehicle.datasource.config.role.DynamicDataSource;
import com.zhiLian.vehicle.datasource.domain.DataSourceInfo;
import lombok.extern.log4j.Log4j2;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Primary;
import org.springframework.stereotype.Component;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* BingRui.Hou

View File

@ -1,5 +1,6 @@
package com.zhiLian.vehicle.datasource.config.role;
import com.zhiLian.vehicle.datasource.config.holder.DynamicDataSourceHolder;
import lombok.AllArgsConstructor;
import lombok.Data;

View File

@ -5,7 +5,6 @@ import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.SuperBuilder;
/**
* BingRui.Hou

View File

@ -3,8 +3,8 @@ package com.zhiLian.vehicle.datasource.util;
import com.alibaba.druid.pool.DruidDataSource;
import com.zhiLian.vehicle.datasource.ManyDataSource;
import com.zhiLian.vehicle.datasource.config.factory.DruidDataSourceFactory;
import com.zhiLian.vehicle.datasource.domain.DataSourceInfo;
import com.zhiLian.vehicle.datasource.config.role.DynamicDataSource;
import com.zhiLian.vehicle.datasource.domain.DataSourceInfo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

View File

@ -1,16 +1,11 @@
package com.zhiLian.vehicle.service.impl;
import java.util.List;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zhiLian.common.core.utils.DateUtils;
import com.zhiLian.vehicle.domain.Fence;
import com.zhiLian.vehicle.domain.Vehicle;
import com.zhiLian.vehicle.mapper.FenceMapper;
import com.zhiLian.vehicle.mapper.VehicleMapper;
import com.zhiLian.vehicle.service.IFenceService;
import com.zhiLian.vehicle.service.IVehicleService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

View File

@ -3,11 +3,8 @@ package com.zhiLian.vehicle.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.zhiLian.common.core.utils.DateUtils;
import com.zhiLian.vehicle.domain.Fence;
import com.zhiLian.vehicle.domain.Group;
import com.zhiLian.vehicle.mapper.FenceMapper;
import com.zhiLian.vehicle.mapper.GroupMapper;
import com.zhiLian.vehicle.service.IFenceService;
import com.zhiLian.vehicle.service.IGroupService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;