摘要：最近公司項目需要支持，每個數據庫地址不同，需求就是修改配置文件，添加或者刪除數據庫配置，重啟系統后就可以完成，不需要額外修改代碼。那怎么來完成數據庫的訪問的呢核心類是，而變量才是完成訪問的關鍵，而這個其實就是。
最近公司項目需要支持multi-tenancy，每個tenant數據庫地址不同，需求就是修改配置文件，添加或者刪除數據庫配置，重啟系統后就可以完成，不需要額外修改代碼。
網上搜索了一下，基本上都是將配置寫在了代碼里了，例如：http://www.devzxd.top/2017/06...，不能動態增減數據庫配置，并且對不同的數據庫創建不同的Repository。繼續Google之后，在github上找到一個例子multi-tenant-spring-mongodb，看了一下代碼，還是不符合需求。沒辦法，只能看代碼，調試來看看自己能不能實現這個需求了。
那spring-data-mongodb怎么來完成數據庫的訪問的呢？核心類是SimpleMongoRepository，而mongoOperations變量才是完成訪問的關鍵，而這個mongoOperations其實就是mongoTemplate。那就意味著如果我們能夠動態改變這個mongoTemplate的值就可以切換數據庫了，竊喜。
解決的方法有了，但是如何實現呢？庫并沒有提供類似Hibernate那樣的MultiTenantConnectionProvider和CurrentTenantIdentifierResolver來幫助我們實現。既然spring是通過AOP和Proxy來完成功能的調用的，我們似乎也可以這么玩。
我們要對所有的Repository方法的訪問創建一個PointCut，這樣我們就可以在訪問之前搞事情了。
利用反射拿到joinPoint的target，然后調用用AopProxyUtils.getSingletonTarget(target)取到最終的SimpleMongoRepository實例。
通過反射設置mongoOperations的值。
代碼如下：
@Repository
public interface WidgetDataRepository extends MongoRepository {
}
// 注意：線程不安全！！！
@Around("execution(* com.example.demo.dao.mongo.MongoWidgetDataRepo.*(..))")
public Object doSwitch(ProceedingJoinPoint joinPoint) throws Throwable {
// 拿到我們需要的tenant
String tenant = (String) RequestContextHolder.currentRequestAttributes().getAttribute("tenant", SCOPE_REQUEST);
tenant = tenant == null ? "test" : tenant;
// 通過反射獲取到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，并往里面填入值
Object singletonTarget = AopProxyUtils.getSingletonTarget(target);
Field mongoOperationsField = singletonTarget.getClass().getDeclaredField("mongoOperations");
mongoOperationsField.setAccessible(true);
mongoOperationsField.set(singletonTarget, ac.getBean("mongoTemplate" + tenant));
return joinPoint.proceed();
}
這樣我們就可以完成數據庫的切換了，可以利用spring-data-mongodb，盡可能的少些模板代碼。
文章版權歸作者所有，未經允許請勿轉載,若此文章存在違規行為，您可以聯系管理員刪除。
轉載請注明本文地址：http://specialneedsforspecialkids.com/yun/19174.html
