SpringBoot+Mongodb不同用户动态切换数据源(MongoTemplate)(二)

最新推荐文章于 2024-05-28 14:08:35 发布
最新推荐文章于 2024-05-28 14:08:35 发布
阅读量3.1k 收藏 7
点赞数
分类专栏： 微服务
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_40384470/article/details/103437628
版权

目录

开发(业务)环境

数据库表名(tianfu_tenant)主要字段如下

  `id` bigint(64) UNSIGNED NOT NULL COMMENT '主键',
  `tenant_code` varchar(12) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NOT NULL COMMENT '租户编码',
  `is_private_db` tinyint(2) NULL DEFAULT NULL COMMENT '是否启用私有库',
  `private_db_addr` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '私有数据库地址,MySQL和MongoDB一致，如果启用私有库且地址是(空、.、127.0.0.1)即表示是云平台的私有库',
  `mysql_port` int(255) NULL DEFAULT NULL COMMENT 'mysql私有数据库端口',
  `mysql_user` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT 'mysql私有数据库用户名',
  `mysql_password` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT 'mysql私有数据库密码',
  `mongo_port` int(255) NULL DEFAULT NULL COMMENT 'mongo私有数据库端口',
  `mongo_user` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT 'mongo私有数据库用户名',
  `mongo_password` varchar(255) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT 'mongo私有数据库密码'

业务逻辑:我使用Spring Aop切面拦截,在每个请求进去controller之前判断当前用户是否使用私有库,进行切换数据源操作
核心依赖

		//mongo依赖
 		<dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-mongodb</artifactId>
        </dependency>
        //这个是我用到了Spring读取配置文件的注解,所以需要导一下
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-configuration-processor</artifactId>
            <optional>true</optional>
        </dependency>

业务处理核心代码

核心配置文件

#服务器端口
server:
  port: 8225
#数据源配置
spring:
  #排除DruidDataSourceAutoConfigure,druid自动配置
  autoconfigure:
    exclude: com.alibaba.druid.spring.boot.autoconfigure.DruidDataSourceAutoConfigure
  datasource:
    dynamic:
      #设置默认的数据源或者数据源组,默认值即为master
      primary: master
      datasource:
        master:
          url: ${tfenergy.datasource.dev.url}
          username: ${tfenergy.datasource.dev.username}
          password: ${tfenergy.datasource.dev.password}
          driver-class-name: ${tfenergy.datasource.dev.driverclassname}
  #配置mongodb链接信息,我这里使用uri的方式配置mongo
  data:
    mongodb:
      uri: mongodb://admin:密码@IP:端口/认证库
      database: 链接库
#添加Swagger
swagger:
  base-packages: ${swagger.base-package}

Mongo抽象类

package org.tfcloud.energy.mongo;
import com.mongodb.MongoClientURI;
import lombok.Data;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
/**
 * 抽象类-Mongo
 */
@Data
public abstract class MongoConfig {
private String uri;
	/*
	 * Method that creates MongoDbFactory Common to both of the MongoDb
	 * connections
	 */
	public MongoDbFactory mongoDbFactory() throws Exception {

		return new SimpleMongoDbFactory(new MongoClientURI(uri));
	}

	/*
	 * Factory method to create the MongoTemplate
	 */
	abstract public MongoTemplate getMongoTemplate() throws Exception;
}

默认Mongo数据源

package org.tfcloud.energy.mongo;
import com.mongodb.MongoClient;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import lombok.Data;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
/**
 *   Mongo 从配置文件当中读取的  
 */
@Component
@Configuration
@Data
public class AbstractMongoDbConfig extends MongoConfig {
	@Value("${spring.data.mongodb.uri}")
	private String uri;
	@Value("${spring.data.mongodb.database}")
	private String fileDatabase;

	/**
	 * mongo默认数据源组装
	 * @return
	 * @throws Exception
	 */
	@SuppressWarnings("all")
	@Primary
	@Bean(name = "mongobaolong")
	@Override
	public MongoTemplate getMongoTemplate() throws Exception {
		List<ServerAddress> addressList=new ArrayList<>();
		List<MongoCredential> credentials=new LinkedList<>();
		String newUrl = uri.replace("mongodb://", "");
		String username = StringUtils.substringBefore(newUrl, ":");
		String psdUrl = newUrl.replace(username + ":", "");
		String password = StringUtils.substringBefore(psdUrl, "@");
		String ipUrl = psdUrl.replace(password + "@", "");
		String ip = StringUtils.substringBefore(ipUrl, ":");
		String portUrl = ipUrl.replace(ip + ":", "");
		Integer port = Integer.parseInt(StringUtils.substringBefore(portUrl, "/"));
		String source = portUrl.replace(port + "/", "");
		MongoCredential mongoCredential=MongoCredential
			.createScramSha1Credential(username
				,source,password.toCharArray());
		credentials.add(mongoCredential);
		ServerAddress serverAddress=new ServerAddress(ip,port);
		addressList.add(serverAddress);
		MongoClient mongoClient=new MongoClient(addressList, credentials);
		return new MongoTemplate(new SimpleMongoDbFactory(mongoClient,fileDatabase));
	}


}

私有库数据源

package org.tfcloud.energy.mongo;
import com.mongodb.MongoClient;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import lombok.Data;
import org.springframework.context.annotation.Bean;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
/**
 * 139mongo
 */
@Component
@Data
public class AbstractMongoDbConfigTwo  extends MongoConfig{
	//用户名
	private String username;
	//认证库
	private String source;
	//密码
	private String password;
	//IP
	private String ip;
	//端口
	private Integer port;
	//链接数据库
	private String fileDatabase;

	/**
	 * 私有数据源创建
	 * @return
	 * @throws Exception
	 */
	@SuppressWarnings("all")
	public  MongoTemplate getMongoTemplateTwo(String username,String source,String password,String ip,Integer port,String fileDatabase) throws Exception {
		List<ServerAddress> addressList=new ArrayList<>();
		List<MongoCredential> credentials=new LinkedList<>();
		MongoCredential mongoCredential=MongoCredential
			.createScramSha1Credential(username,source,password.toCharArray());
		credentials.add(mongoCredential);
		ServerAddress serverAddress=new ServerAddress(ip,port);
		addressList.add(serverAddress);
		MongoClient mongoClient=new MongoClient(addressList, credentials);
		return new MongoTemplate(new SimpleMongoDbFactory(mongoClient,fileDatabase));
	}

	@Override
	public MongoTemplate getMongoTemplate() throws Exception {
		return null;
	}
}

AOP切面

package org.tfcloud.energy.mongo;

import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springblade.core.secure.utils.SecureUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.system.entity.Tenant;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.Ordered;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.tfcloud.energy.mapper.DataSourceMapper;

import javax.servlet.http.HttpServletRequest;

@Aspect
@Component
@SuppressWarnings("all")
public class MongoAspect implements Ordered {
	//认证库
	private static final String SROUCE_MONGO="test";
	//链接库(该字段可以随用户设置进行动态切换)
	private static final String AUTHORIZATION_DATABASE="TF_Iot_Energy";
	@Autowired
	private AbstractMongoDbConfig abstractMongoDbConfig;
	@Autowired
	private AbstractMongoDbConfigTwo abstractMongoDbConfigTwo;
	@Autowired
	private DataSourceMapper dataSourceMapper;

	/**
	 * 我这里 创建的切入点就是我的controller的路径,这里你换成你自己的
	 */
	@Pointcut("execution(* org.tfcloud.energy.controller..*(..)))")
	public void mongoPointcut() {
	}
	//请求业务之前拦截处理 切换私有库
	@Before(value = "mongoPointcut()")
	public void before(JoinPoint joinPoint) throws Exception {
	//新建 Request域,当然也可以存储session  redis等  都OK 
		HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
		//获得当前租户,我这里是已经封装的工具类,这里你可以换成你的查询语句去数据库查
		String tenantCode = SecureUtil.getUser().getTenantCode();
		//得到用户对象
		Tenant tenant = dataSourceMapper.findDataSourceByTenant(tenantCode);
		/**
		 * 这里以0为启用私有mongo
		 * String username,String source,String password,String ip,Integer port,String fileDatabase
		 */
		if (Func.isNotEmpty(tenant.getIsPrivateDb())&&tenant.getIsPrivateDb() == 0){
		//在request域中注入私有库所需要的mongo链接信息(mongoTemplate)
			MongoTemplate mongoTemplateTwo = abstractMongoDbConfigTwo.getMongoTemplateTwo(tenant.getMongoUser(), SROUCE_MONGO
				, tenant.getMongoPassword(), tenant.getPrivateDbAddr(), tenant.getMongoPort(), AUTHORIZATION_DATABASE);
			request.setAttribute("mongoDataSource", mongoTemplateTwo);
		}else {
		//默认mongo数据源
			request.setAttribute("mongoDataSource", abstractMongoDbConfig.getMongoTemplate());
		}
	}
	//请求结束之后 执行
	@After(value = "mongoPointcut()")
	public void after(JoinPoint joinPoint) throws Exception {
		//每执行完一次请求 清空request域
		HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
		request.removeAttribute("mongoDataSource");
	}
//ordered   返回数值越小 越靠前执行
	@Override
	public int getOrder() {
		return 2;
	}

}

打通Mongo测试多数据源接口(供参考)

package org.tfcloud.energy.controller;
import lombok.AllArgsConstructor;
import org.springblade.core.tool.api.R;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.tfcloud.energy.dto.TFlotEnergy;
import org.tfcloud.energy.mongo.AbstractMongoDbConfig;
import org.tfcloud.energy.mongo.AbstractMongoDbConfigTwo;
import org.tfcloud.energy.mongo.MongoTemplateChildren;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
/**
 * 打通mongo数据源接口入口
 */
@RestController
@RequestMapping("dd")
@AllArgsConstructor
@SuppressWarnings("all")
public class MongoController {


	/**
	 * 打通mongo动态
	 * @return
	 * @throws Exception
	 */
	@GetMapping(value = "getMongoTestTwo")
	public R<TFlotEnergy> demos(HttpServletRequest request)throws Exception {
		Query query = new Query();
		Criteria criterName = Criteria.where("scheme_name").is("1#染色_冷水表");
		Criteria criteria = Criteria.where("_time").is("2019-12-06 13:21:35");
		query.addCriteria(criterName);
		query.addCriteria(criteria);
		MongoTemplate mongoDataSource = (MongoTemplate)request.getAttribute("mongoDataSource");
		TFlotEnergy tFlotEnergy = mongoDataSource.findOne(query, TFlotEnergy.class);
		return R.data(tFlotEnergy);
	}
}

后续会持续更新,优化

联系博主方式

如果有任何问题都可以向我提问
QQ：2509647976 微信: x331191249

十里八方
关注
  • 0
    点赞
  • 踩
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
专栏目录
SpringBoot使用MongoTemplate动态切换MongoDB
JavaAndJava的专栏
11-22 7518
SpringBoot使用MongoTemplate动态切换Mongo     当前有一个项目包含一套代码和1个mongo的数据库，但要给2个部门用户使用，而且要做数据隔离，现有做法是部署2套代码和2个mongo数据库，2个域名，这样就能各自访问各自的系统，做到隔离。但为节省资源和减少运维工作，计划将2个系统合并为一个系统，一套代码部署，2个数据库。那么如何使用最少的成本完成系统升级呢。     计...
springboot mongodb 动态数据源 分库 分表
fengyi_336494的博客
03-16 4477
背景 业务侧分库分表 根据账户切换不同数据库，多个账户在同一个库。例如0001-0005的客户在一个库。 根据账户分表，账户名_表名 其实使用mongodb不用业务侧做这种分库分表，历史原因，还是实现了。 分库 springboot mongodb 的 MongoTemplate 默认是一个database。我们需要动态切换。所以需要改造。 通过跟踪MongoTemplate构造我们发现，有一个关键的类MongoDatabaseFactory database工厂。 这个接口定义了一个方法getMon
Invalid bound statement (not found): org.springblade.core.mp.mapper.BladeMapper.insert
最新发布
yszbrzdd的博客
05-28 180
虽然方法不是从这篇博文中找到的，但是这个博文挺优秀的可以参考下其他的因素导致的扫描不到包。启动类上没加：@MapperScan("com.dh.**.mapper")mapper扫描包没修改。
springboot+mongoDB 切换数据库
qq_33396462的博客
04-03 550
该方法是，springboot配置默认数据库后，待查数据又存在其他数据库时，可采用该方法进行数据库的切换。最重要的是通过 默认的mongo客户端 调用。
springboot根据前端传参动态切换数据源，查询数据库所有的表
qq_42752478的博客
03-27 2757
springboot根据前端传参动态切换数据源，查询数据库所有的表
mongo动态数据源spring-starter
时光熬走了谁love的博客
10-16 864
mongodb动态切换实例和数据库 mongodb多数据源支持
mongo多数据源教程 mongoTemplate 多数据源
wonder_dog的博客
06-14 3229
参考链接 参考与csdn以为大佬的文章 找到链接再附上 第一步 添加mongo依赖 版本号统一在父pom管理 <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data...
SpringBoot+MongoDB+Vue前后分离
06-16
SpringBoot2.x+MongoDB4.x+Vue均选用能进行生产环境部署的稳定版本【开发模式】前后端分离开发，即当前端又当后端，分角色开发【课程受益】一次上手，即学即会，提供全套源代码，直接可以运行【技术介绍】MongoDB...
SpringBoot+MongoDB+Echarts图表数据可视化
07-21
综上所述，"SpringBoot+MongoDB+Echarts图表数据可视化"项目利用SpringBoot作为后端处理逻辑，MongoDB作为数据存储，Echarts作为前端数据展示工具，共同实现了高效、直观的数据可视化解决方案。这个组合在现代Web...
springboot +mongodb多数据源
01-18
spring.data.primary.mongodb.uri= mongodb://192.168.10.136:27017/eqt #spring.data.secondary.mongodb.uri= mongodb:...切换mongodb数据源 可以同一台机器，或者不同机器 实现动态插入 。 该例是根据companyId来处理
springboot+mongodb 实现按日期分组分页查询功能
08-27
AggregationResults<BasicDBObject> eatOutputType = mongoTemplate.aggregate(eatAggregation, "inner_cash_change", BasicDBObject.class); ``` 在上面的代码中，我们使用 `Aggregation.newAggregation` 方法来...
springboot+mongodb 水下网络拓扑发现系统
01-26
在构建“springboot+mongodb 水下网络拓扑发现系统”时，我们结合了Spring Boot框架和MongoDB数据库，以实现一个高效且灵活的数据存储和处理解决方案。Spring Boot简化了Spring应用程序的开发过程，而MongoDB则为非...
springboot使用MongoTemplate绑定多数据源（并且插入时去掉_class）
dgywj的专栏
01-14 1177
开门见山，直接上码 1、pom.xml <!-- 访问mongoDb依赖 --> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-data-mongodb</artifactId> </dependency>
SpringBoot+Mybatis-plus不同用户动态切换数据源(一)
偏偏行
11-04 8933
步骤如下 的撒 大撒大撒 的撒的 杀杀杀
使用mybatisplus实现用户级的动态数据源切换
qq_33440246的博客
01-12 3640
前言 昨天在B站看了基于SpringBoot和MyBatis-Plus多数据源分析的视频，评论区有小伙伴想实现页面级的数据源切换，经过初步的分析就有了下面的想法。 通过分析源码可知道，MyBatis-Plus提供了dynamic-datasource-spring-boot-starter 以支持多数据源的需求，在使用方式上，运用注解可灵活便捷地实现指定数据源进行表操作。能否实现用户级的多数据源切换，用户在前端页面做出选择，后端匹配对应的数据源? 很简单，下面来详细谈谈。 原理分析 dynamic-data
十、MongoDB 切换数据库
qq_25843323的博客
02-16 910
十、MongoDB 切换数据库
springboot整和mogodb（MongoRepository）简单操作（增删改查，模糊、条件和分页查询）和SpringData 方法定义规范
weixin_53472653的博客
02-11 2642
Spring Data提供了对mongodb数据访问的支持，我们只需要继承MongoRepository类 前面的基础流程应用依赖，创建实体类参考下这篇文章参考文本 1.创建接口继承MongoRepository package com.example.mongodb.repository; import com.example.mongodb.entity.Student; import org.springframework.data.mongodb.repository.MongoRepos
SpringBoot高级篇MongoDB之修改基本使用姿势
weixin_33856370的博客
03-02 290
原文: 190218-SpringBoot高级篇MongoDB之修改基本使用姿势 本篇依然是MongoDB curd中的一篇，主要介绍document的更新，主要内容如下 常见类型成员的修改 数组类型成员的增删改 document类型成员的增删改 I. 基本使用 首先是准备好基本环境，可以参考博文 181213-SpringBoot高级篇MongoDB之基本环境搭建与使用 190113...
MongoDB之SpringBoot操作篇
热门推荐
小宋的博客
06-13 2万+
本篇博客主讲如何使用SpringBoot操作MongoDB。 SpringBoot操作MongoDB实现增删改查 （1）pom.xml引入依赖 <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring...
Springboot mongodb 多数据源动态切换
05-25
在Springboot中使用MongoDB多数据源并动态切换，可以通过以下步骤实现： 1. 添加MongoDB的依赖 在pom.xml文件中加入对spring-boot-starter-data-mongodb和mongodb-driver的依赖。 2. 配置多数据源 在application.yml或application.properties中配置多个数据源，如下所示： ``` spring: data: mongodb: uri: mongodb://localhost:27017/test1 database: test1 seconddata: mongodb: uri: mongodb://localhost:27017/test2 database: test2 ``` 其中第一个数据源是默认的数据源，第二个数据源是自定义的数据源。 3. 配置数据源切换组件 在Springboot中使用MongoDB多数据源需要使用MongoDbFactory和MongoTemplate两个组件。我们可以通过自定义MongoTemplate的方式来实现数据源的切换。具体实现可以参考以下代码： ```java @Configuration public class MultipleMongoConfig { @Bean(name = "firstMongoTemplate") @Primary public MongoTemplate firstMongoTemplate() throws Exception { return new MongoTemplate(firstFactory()); } @Bean(name = "secondMongoTemplate") public MongoTemplate secondMongoTemplate() throws Exception { return new MongoTemplate(secondFactory()); } @Bean @Primary public MongoDbFactory firstFactory() throws Exception { return new SimpleMongoDbFactory(new MongoClientURI( env.getProperty("spring.data.mongodb.uri"))); } @Bean public MongoDbFactory secondFactory() throws Exception { return new SimpleMongoDbFactory(new MongoClientURI( env.getProperty("seconddata.mongodb.uri"))); } } ``` 4. 动态切换数据源 在需要切换数据源的地方，可以通过注入MongoTemplate来实现。具体实现可以参考以下代码： ```java @Service public class UserServiceImpl implements UserService { @Autowired @Qualifier("firstMongoTemplate") private MongoTemplate firstMongoTemplate; @Autowired @Qualifier("secondMongoTemplate") private MongoTemplate secondMongoTemplate; public void save(User user) { if (user.isUseSecondDataSource()) { secondMongoTemplate.save(user); } else { firstMongoTemplate.save(user); } } public User findById(String id, boolean useSecondDataSource) { if (useSecondDataSource) { return secondMongoTemplate.findById(id, User.class); } else { return firstMongoTemplate.findById(id, User.class); } } } ``` 在上面的示例中，我们可以通过isUseSecondDataSource()方法来判断是否需要使用第二个数据源。如果需要使用第二个数据源，则使用secondMongoTemplate；否则使用firstMongoTemplate。 以上就是在Springboot中使用MongoDB多数据源并动态切换的全部步骤。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值