Spring boot MongoDB多数据源,MongoRepository实现
https://blog.csdn.net/qq_35124119/article/details/83210772
背景
最近项目中有需求,需要多个mongo库,分割数据。网上有很多文章可参考,其原理是:在Spring 容器中实例化多个MongoTemplate,代码示例:
@Configuration
@EnableMongoRepositories(basePackages = {"com.sunliangliang.service.basic"}, mongoTemplateRef = "basicMongoTemplate")
@ConfigurationProperties(prefix = "basic.mongodb")
public class BasicMongoConfigure extends AbstractMongoConfigure {
@Override
@Bean(name = "basicMongoTemplate")
public MongoTemplate getMongoTemplate() throws Exception {
return new MongoTemplate(mongoDbFactory());
}
}
使用方式:
1.使用时通过@Autowired 和@Qualifier注入MongoTemplate实例以操作不同的mongo数据库(PS:可使用@Resource注解,引入MongoTemplate 并为其实例命名为所需的已配置的bean名称,而引入相对应的MongoTemplate实例)
2.通过MongoTemplate示例配置的basePackages指定继承了MongoRepository的model接口,在操作数据库时,使用所配置的MongoTemplate实例,以操作不同的数据库。
这是最常见的Springboot Mongo多数据配置方法,但是项目中主项目因引入的子项目里已有配置,导致在主项目中MongoTemplate配置无效,MongoRepository无法按配置的basePackages路径使用对应的MongoTemplate操作mongo库。
分析
前述的实现mongo多数据库,核心是操作不同的MongoTemplate。除了直接使用MongoTemplate,我们通常会写一个接口继承MongoRepository接口,并且不需要实现该接口就可操作数据库。实际上在使用继承的MongoRepository接口的业务接口访问mongo时,使用的是SpringData的SimpleMongoRepository类的实例来操作数据库,而SimpleMongoRepository的一个非常重要的Filed是MongoOperations接口,而MongoTemplate正是MongoOperations的实现。SimpleMongoRepository也正是使用MongoOperations操作mongo数据库的。
思路
在SimpleMongoRepository操作数据库时,动态的修改其MongoOperations的值,即MongoTemplate。所以Spring AOP出场了。
解决办法
1.为项目引入Spring AOP
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
2.编写AOP代码
package com.mongo;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.aop.framework.AopProxyUtils;
import org.springframework.aop.framework.ReflectiveMethodInvocation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.WebApplicationContext;
import java.lang.reflect.Field;
/**
* 通过AOP操作,动态更改更改mongo的repository层mongoTemplate<br/>
* 以实现mongo分库
*
* @author RangoLan
* @desciption
* @date Created in 2018/10/17 15:20
*/
@Aspect
@Component
public class RepositoryAop {
@Autowired
WebApplicationContext context;
@Around("execution(* com.mongo.basic..*.*(..))")
public Object setMongoOperations(ProceedingJoinPoint joinPoint) throws Throwable {
setMongoTemplate4Repository(joinPoint, (MongoTemplate) context.getBean(AdminConfiguration.MONGO_ADMIN));
return joinPoint.proceed();
}
private void setMongoTemplate4Repository(ProceedingJoinPoint joinPoint, MongoTemplate template) throws NoSuchFieldException, IllegalAccessException {
// 通过反射获取到target
Field methodInvocationField = joinPoint.getClass().getDeclaredField("methodInvocation");
methodInvocationField.setAccessible(true);
ReflectiveMethodInvocation o = (ReflectiveMethodInvocation) methodInvocationField.get(joinPoint);
Field targetField = o.getClass().getDeclaredField("target");
targetField.setAccessible(true);
Object target = targetField.get(o);
// 获得SimpleMongoRepository,并往里面填入指定mongoTemplate
Object singletonTarget = AopProxyUtils.getSingletonTarget(target);
Field mongoOperationsField = singletonTarget.getClass().getDeclaredField("mongoOperations");
mongoOperationsField.setAccessible(true);
mongoOperationsField.set(singletonTarget, template);
}
}
单元测试OK。
注意: 由于SimpleMongoRepository在Spring容器中为单例,aop切点代码执行后,SimpleMongoRepository的MongoOperations已更改,其他未配置切点的Repository在操作mongo时,数据库会混乱。所以应该在AOP中明确配置各Repository所使用的MongoTemplate。
后记
AOP方式实现,可实现Repository层的mongodb分库,但是相较于常规做法,还是比较麻烦一点,建议构建项目时结构应尽量合理些。相较于配置MongoTemplate bean时指定basePackages,AOP可做到方法级控制,当然这是AOP的优点,但是对于数据库来说,一个表或者collection的所有操作不该是在同一个库么?
————————————————
原文链接:https://blog.csdn.net/qq_35124119/article/details/83210772