diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/DataSourcemanyApplication.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/DataSourcemanyApplication.java index 1ef3b13..8f0b674 100644 --- a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/DataSourcemanyApplication.java +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/DataSourcemanyApplication.java @@ -7,7 +7,6 @@ import com.muyu.common.swagger.annotation.EnableCustomSwagger2; import org.springframework.boot.SpringApplication; import org.springframework.boot.autoconfigure.SpringBootApplication; import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration; -import org.springframework.boot.web.servlet.DynamicRegistrationBean; /** * 多数据源启动类 diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/ManyDataSource.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/ManyDataSource.java new file mode 100644 index 0000000..a430425 --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/ManyDataSource.java @@ -0,0 +1,83 @@ +package com.muyu.many.config; + +import com.alibaba.druid.pool.DruidDataSource; +import com.muyu.common.core.utils.SpringUtils; +import com.muyu.many.config.domai.model.DataSourceInfo; +import com.muyu.many.config.domai.model.EntInfo; +import com.muyu.many.config.factory.DruidDataSourceFactory; +import com.muyu.many.config.role.DynamicDataSource; +import lombok.extern.log4j.Log4j2; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Primary; +import org.springframework.stereotype.Component; + +import javax.annotation.PostConstruct; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +/** + * 多数据源 + * + * @ClassName ManyDataSource + * @Author AnNan.Wang + * @Date 2024/6/4 14:10 + */ + +@Log4j2 +@Component +public class ManyDataSource { + + + @PostConstruct + public void init() { + new Thread(() ->{ + try { + Thread.sleep(5000); + } catch (InterruptedException ignored) {} + DruidDataSourceFactory druidDataSourceFactory = SpringUtils.getBean(DruidDataSourceFactory.class); + DynamicDataSource dynamicDataSource = SpringUtils.getBean(DynamicDataSource.class);; + EntInfo entInfo = EntInfo.builder() + .entCode("ent_4588") + .ip("123.56.102.11") + .port(3308) + .build(); + DataSourceInfo dataSourceInfo = DataSourceInfo.hostAndPortBuild(entInfo.getEntCode(), entInfo.getIp(), entInfo.getPort()); + DruidDataSource druidDataSource = druidDataSourceFactory.create(dataSourceInfo); + dynamicDataSource.put(dataSourceInfo.getKey(), druidDataSource); + }).start(); + + } + + private List dataSourceInfoList() { + List list = new ArrayList<>(); + list.add( + EntInfo.builder() + .entCode("ent_4587") + .ip("123.56.102.11") + .port(3310) + .build() + ); + return list; + } + @Bean + @Primary + public DynamicDataSource dynamicDataSource(DruidDataSourceFactory druidDataSourceFactory) { + + Map 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; + } +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/contents/DatasourceContent.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/contents/DatasourceContent.java new file mode 100644 index 0000000..d3d69c8 --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/contents/DatasourceContent.java @@ -0,0 +1,15 @@ +package com.muyu.many.config.contents; + +/** + * @author DongZl + * @description: 数据源常量 + * @Date 2023-8-1 上午 11:02 + */ +public class DatasourceContent { + + public final static String DATASOURCE_URL = "jdbc:mysql://{}:{}/car?useUnicode=true&characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=true&serverTimezone=GMT%2B8"; + + public final static String USER_NAME = "root"; + + public final static String PASSWORD = "root"; +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/domai/model/DataSourceInfo.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/domai/model/DataSourceInfo.java new file mode 100644 index 0000000..262c4af --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/domai/model/DataSourceInfo.java @@ -0,0 +1,48 @@ +package com.muyu.many.config.domai.model; + +import com.muyu.common.core.utils.StringUtils; +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; +import lombok.experimental.SuperBuilder; + +import static com.muyu.many.config.contents.DatasourceContent.*; + +/** + * @ClassName DataSourceInfo + * @Author AnNan.Wang + * @Date 2024/6/4 14:20 + */ + +@Data +@SuperBuilder +@NoArgsConstructor +@AllArgsConstructor +public class DataSourceInfo { + /** + * 键 + */ + private String key; + /** + * 地址 + */ + private String url; + /** + * 用户名 + */ + private String userName; + /** + * 密码 + */ + private String password; + + + public static DataSourceInfo hostAndPortBuild(String key,String host,Integer port){ + return DataSourceInfo.builder() + .key(key) + .url(StringUtils.format(DATASOURCE_URL, host,port)) + .password(PASSWORD) + .userName(USER_NAME) + .build(); + } +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/domai/model/EntInfo.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/domai/model/EntInfo.java new file mode 100644 index 0000000..cf3f5a1 --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/domai/model/EntInfo.java @@ -0,0 +1,24 @@ +package com.muyu.many.config.domai.model; + +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.NoArgsConstructor; +import lombok.experimental.SuperBuilder; + +/** + * 企业信息 + * + * @ClassName EntInfo + * @Author AnNan.Wang + * @Date 2024/6/4 15:02 + */ + +@Data +@SuperBuilder +@NoArgsConstructor +@AllArgsConstructor +public class EntInfo { + private String entCode; + private String ip; + private Integer port; +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/factory/DruidDataSourceFactory.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/factory/DruidDataSourceFactory.java new file mode 100644 index 0000000..c0abb7b --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/factory/DruidDataSourceFactory.java @@ -0,0 +1,41 @@ +package com.muyu.many.config.factory; + +import com.alibaba.druid.pool.DruidDataSource; +import com.muyu.many.config.domai.model.DataSourceInfo; +import lombok.extern.log4j.Log4j2; +import org.springframework.stereotype.Component; + +import java.sql.SQLException; + +/** + * Druid工厂 + * + * @ClassName DruidDataSourceFactory + * @Author AnNan.Wang + * @Date 2024/6/4 14:14 + */ + +@Log4j2 +@Component +public class DruidDataSourceFactory { + /** + * @Description: 根据传递的数据源信息测试数据库连接 + * @Author AnNan.Wang + */ + 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; + } + } +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/holder/DynamicDataSourceHolder.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/holder/DynamicDataSourceHolder.java new file mode 100644 index 0000000..46e2475 --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/holder/DynamicDataSourceHolder.java @@ -0,0 +1,42 @@ +package com.muyu.many.config.holder; + +import lombok.extern.slf4j.Slf4j; +import org.springframework.util.Assert; + +/** + * 数据源切换处理 + * + * @author Dongzl + */ +@Slf4j +public class DynamicDataSourceHolder { + /** + * 保存动态数据源名称 + */ + private static final ThreadLocal DYNAMIC_DATASOURCE_KEY = new ThreadLocal<>(); + + /** + * 设置/切换数据源，决定当前线程使用哪个数据源 + */ + public static void setDynamicDataSourceKey(String key){ + log.info("数据源切换为：{}",key); + DYNAMIC_DATASOURCE_KEY.set(key); + } + + /** + * 获取动态数据源名称，默认使用mater数据源 + */ + public static String getDynamicDataSourceKey(){ + String key = DYNAMIC_DATASOURCE_KEY.get(); + Assert.notNull(key, "请携带数据标识"); + return key; + } + + /** + * 移除当前数据源 + */ + public static void removeDynamicDataSourceKey(){ + log.info("移除数据源：{}",DYNAMIC_DATASOURCE_KEY.get()); + DYNAMIC_DATASOURCE_KEY.remove(); + } +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/role/DynamicDataSource.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/role/DynamicDataSource.java new file mode 100644 index 0000000..6bdbcfe --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/config/role/DynamicDataSource.java @@ -0,0 +1,44 @@ +package com.muyu.many.config.role; + +import com.alibaba.druid.pool.DruidDataSource; +import com.muyu.many.config.domai.model.DataSourceInfo; +import com.muyu.many.config.holder.DynamicDataSourceHolder; +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.EqualsAndHashCode; +import lombok.NoArgsConstructor; +import org.springframework.jdbc.datasource.lookup.AbstractRoutingDataSource; + +import java.util.Map; + +/** + * 动态数据源 + * 调用AddDefineDataSource组件的addDefineDynamicDataSource（）方法，获取原来targetdatasources的map，并将新的数据源信息添加到map中，并替换targetdatasources中的map + * 切换数据源时可以使用@DataSource(value = "数据源名称")，或者DynamicDataSourceContextHolder.setContextKey("数据源名称") + * @author Dongzl + */ +@EqualsAndHashCode(callSuper = true) +@Data +@AllArgsConstructor +@NoArgsConstructor +public class DynamicDataSource extends AbstractRoutingDataSource { + //备份所有数据源信息，备份的是个 指针 !!! + private Map defineTargetDataSources; + + + /** + * 添加数据源 + * @param key + * @param value + */ + public void put(String key, DruidDataSource value) { + defineTargetDataSources.put(key, value); + } + /** + * 决定当前线程使用哪个数据源 + */ + @Override + protected Object determineCurrentLookupKey() { + return DynamicDataSourceHolder.getDynamicDataSourceKey(); + } +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/controller/EnterpriseController.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/controller/EnterpriseController.java new file mode 100644 index 0000000..4c18e42 --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/controller/EnterpriseController.java @@ -0,0 +1,35 @@ +package com.muyu.many.server.controller; + +import com.muyu.common.core.domain.Result; +import com.muyu.many.config.holder.DynamicDataSourceHolder; +import com.muyu.many.server.domain.Enterprise; +import com.muyu.many.server.service.EnterpriseService; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PathVariable; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +import java.util.List; + +/** + * 企业表 Controller层 + * + * @ClassName EnterpriseController + * @Author AnNan.Wang + * @Date 2024/6/4 20:53 + */ + + +@RestController +@RequestMapping("/enterprise") +public class EnterpriseController { + @Autowired + private EnterpriseService enterpriseService; + + @GetMapping("/all/{code}") + public Result> all(@PathVariable("code") String code){ + DynamicDataSourceHolder.setDynamicDataSourceKey(code); + return enterpriseService.enterpriseList(code); + } +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/domain/Enterprise.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/domain/Enterprise.java new file mode 100644 index 0000000..c87a284 --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/domain/Enterprise.java @@ -0,0 +1,96 @@ +package com.muyu.many.server.domain; + +import com.baomidou.mybatisplus.annotation.IdType; +import com.baomidou.mybatisplus.annotation.TableId; +import com.baomidou.mybatisplus.annotation.TableName; +import com.muyu.common.core.web.domain.BaseEntity; +import lombok.AllArgsConstructor; +import lombok.Data; +import lombok.EqualsAndHashCode; +import lombok.NoArgsConstructor; +import lombok.experimental.SuperBuilder; + +import java.util.Date; + +/** + * 企业表 + * + * @ClassName Enterprise + * @Author AnNan.Wang + * @Date 2024/6/4 20:47 + */ + + +@EqualsAndHashCode(callSuper = true) +@Data +@SuperBuilder +@NoArgsConstructor +@AllArgsConstructor +@TableName(value = "enterprise") +public class Enterprise extends BaseEntity { + /** + * 主键 + */ + @TableId(value = "id",type = IdType.AUTO) + private Long id; + /** + * 企业名称 + */ + private String enterpriseName; + /** + * 法定代表人 + */ + private String legalPerson; + /** + * 经营执照凭证号码 + */ + private String businessLicenseNumber; + /** + * 企业成立时间 + */ + private Date establishmentDate; + /** + * 经营范围 + */ + private String businessScope; + /** + * 注册地址 + */ + private String address; + /** + * 企业联系方式 + */ + private String contactPhone; + /** + * 公司邮箱 + */ + private String email; + /** + * 企业当前状态 + */ + private Integer status; + /** + * 企业入驻平台时期 + */ + private Date registrationDate; + /** + * 企业认证id + */ + private Long certificationId; + /** + * 认证时间 + */ + private Date authenticationDate; + /** + * 服务级别 + */ + private Integer serviceLevel; + /** + * 开通服务id + */ + private Long open_serverId; + /** + * 增值服务id + */ + private Long add_serverId; +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/mapper/EnterpriseMapper.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/mapper/EnterpriseMapper.java new file mode 100644 index 0000000..f8d2683 --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/mapper/EnterpriseMapper.java @@ -0,0 +1,14 @@ +package com.muyu.many.server.mapper; + +import com.baomidou.mybatisplus.core.mapper.BaseMapper; +import com.muyu.many.server.domain.Enterprise; + +/** + * Enterprise 企业表的Mapper接口层 + * + * @author AnNan.Wang + * @ClassName: EnterpriseMapper + * @createTime: 2024/6/4 20:58 + */ +public interface EnterpriseMapper extends BaseMapper { +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/service/EnterpriseService.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/service/EnterpriseService.java new file mode 100644 index 0000000..e5fc20e --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/service/EnterpriseService.java @@ -0,0 +1,18 @@ +package com.muyu.many.server.service; + +import com.baomidou.mybatisplus.extension.service.IService; +import com.muyu.common.core.domain.Result; +import com.muyu.many.server.domain.Enterprise; + +import java.util.List; + +/** + * Enterprise 企业表Service层接口 + * + * @author AnNan.Wang + * @ClassName: EnterpriseService + * @createTime: 2024/6/4 20:54 + */ +public interface EnterpriseService extends IService { + Result> enterpriseList(String code); +} diff --git a/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/service/impl/EnterpriseServiceImpl.java b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/service/impl/EnterpriseServiceImpl.java new file mode 100644 index 0000000..eec6c3a --- /dev/null +++ b/muyu-modules/muyu-common-many-dataSourceMany/src/main/java/com/muyu/many/server/service/impl/EnterpriseServiceImpl.java @@ -0,0 +1,35 @@ +package com.muyu.many.server.service.impl; + +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.muyu.common.core.domain.Result; +import com.muyu.many.server.domain.Enterprise; +import com.muyu.many.server.mapper.EnterpriseMapper; +import com.muyu.many.server.service.EnterpriseService; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.stereotype.Service; + +import java.util.List; + +/** + * Enterprise 企业表的EnterpriseService业务实现层 + * + * @ClassName EnterpriseServiceImpl + * @Author AnNan.Wang + * @Date 2024/6/4 20:57 + */ + +@Service +public class EnterpriseServiceImpl extends ServiceImpl + implements EnterpriseService{ + @Autowired + private EnterpriseMapper enterpriseMapper; + + + @Override + public Result> enterpriseList(String code) { + return Result.success( + enterpriseMapper.selectList(null) + ); + } +}