spring-data-mongodb多數(shù)據(jù)庫訪問實(shí)現(xiàn)

pf_miles 發(fā)布于2019-08-15 13:47 / 2707人閱讀

摘要：最近公司項(xiàng)目需要支持，每個(gè)數(shù)據(jù)庫地址不同，需求就是修改配置文件，添加或者刪除數(shù)據(jù)庫配置，重啟系統(tǒng)后就可以完成，不需要額外修改代碼。那怎么來完成數(shù)據(jù)庫的訪問的呢核心類是，而變量才是完成訪問的關(guān)鍵，而這個(gè)其實(shí)就是。

最近公司項(xiàng)目需要支持multi-tenancy，每個(gè)tenant數(shù)據(jù)庫地址不同，需求就是修改配置文件，添加或者刪除數(shù)據(jù)庫配置，重啟系統(tǒng)后就可以完成，不需要額外修改代碼。

網(wǎng)上搜索了一下，基本上都是將配置寫在了代碼里了，例如：http://www.devzxd.top/2017/06...，不能動(dòng)態(tài)增減數(shù)據(jù)庫配置，并且對(duì)不同的數(shù)據(jù)庫創(chuàng)建不同的Repository。繼續(xù)Google之后，在github上找到一個(gè)例子multi-tenant-spring-mongodb，看了一下代碼，還是不符合需求。沒辦法，只能看代碼，調(diào)試來看看自己能不能實(shí)現(xiàn)這個(gè)需求了。

那spring-data-mongodb怎么來完成數(shù)據(jù)庫的訪問的呢？核心類是SimpleMongoRepository，而mongoOperations變量才是完成訪問的關(guān)鍵，而這個(gè)mongoOperations其實(shí)就是mongoTemplate。那就意味著如果我們能夠動(dòng)態(tài)改變這個(gè)mongoTemplate的值就可以切換數(shù)據(jù)庫了，竊喜。

解決的方法有了，但是如何實(shí)現(xiàn)呢？庫并沒有提供類似Hibernate那樣的MultiTenantConnectionProvider和CurrentTenantIdentifierResolver來幫助我們實(shí)現(xiàn)。既然spring是通過AOP和Proxy來完成功能的調(diào)用的，我們似乎也可以這么玩。

我們要對(duì)所有的Repository方法的訪問創(chuàng)建一個(gè)PointCut，這樣我們就可以在訪問之前搞事情了。

利用反射拿到j(luò)oinPoint的target，然后調(diào)用用AopProxyUtils.getSingletonTarget(target)取到最終的SimpleMongoRepository實(shí)例。

通過反射設(shè)置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();
}

這樣我們就可以完成數(shù)據(jù)庫的切換了，可以利用spring-data-mongodb，盡可能的少些模板代碼。