說明：spring-sharding.xml配置的分庫分表規則：authuser表分到id為sjds${0..3}的四個庫中，表名保持不變；其他表在id為sjds_default庫中，不分庫也不分表；整合sharding-jdbc的核心就是將SqlSessionFactoryBean需要的dataSource屬性修改為 shardingDataSource，把資料源交給sharding-jdbc處理；

分庫邏輯 AuthUserDatabaseShardingAlgorithm的程式碼很簡單，原始碼如下：

/**
 * @author wangzhenfei9
 * @version 1.0.0
 * @since 2018年02月08日
 */
public class AuthUserDatabaseShardingAlgorithm implements SingleKeyDatabaseShardingAlgorithm<Long> {
    private final Logger logger = LoggerFactory.getLogger(this.getClass());
    private static final int SHARDING_NUMBER = 4;
    @Override
    public String doEqualSharding(final Collection<String> availableTargetNames, final ShardingValue<Long> shardingValue) {
        for (String each : availableTargetNames) {
            if (each.endsWith(shardingValue.getValue() % SHARDING_NUMBER + "")) {
                logger.debug("the target database name: {}", each);
                return each;
            }
        }
        throw new UnsupportedOperationException();
    }
    @Override
    public Collection<String> doInSharding(final Collection<String> availableTargetNames, final ShardingValue<Long> shardingValue) {
        Collection<String> result = new LinkedHashSet<>(availableTargetNames.size());
        Collection<Long> values = shardingValue.getValues();
        for (Long value : values) {
            for (String each : availableTargetNames) {
                if (each.endsWith(value % SHARDING_NUMBER + "")) {
                    result.add(each);
                }
            }
        }
        return result;
    }
    @Override
    public Collection<String> doBetweenSharding(final Collection<String> availableTargetNames, final ShardingValue<Long> shardingValue) {
        Collection<String> result = new LinkedHashSet<>(availableTargetNames.size());
        Range<Long> range = shardingValue.getValueRange();
        for (Long value = range.lowerEndpoint(); value <= range.upperEndpoint(); value++) {
            for (String each : availableTargetNames) {
                if (each.endsWith(value % SHARDING_NUMBER + "")) {
                    result.add(each);
                }
            }
        }
        return result;
    }
}

這段程式碼參考sharding-jdbc原始碼中 DatabaseShardingAlgorithm.java介面的實現即可，例如 ModuloDatabaseShardingAlgorithm.java；

4. 註意事項

無法識別sharding-jdbc分庫分表規則inline-expression問題，例如： logic-table="auth_user"actual-tables="sj_ds_${0..3}.auth_user"database-strategy="databaseStrategy"/>

1. 根本原因： 根本原因是spring把 ${}當做佔位符， ${0..3}這種運算式，spring會嘗試去properties檔案中找key為 0..3的屬性。但是這裡是sharding-jdbc分庫分表規則的inline運算式，需要spring忽略這種行為。否則會丟擲異常: java.lang.IllegalArgumentException: Could not resolve placeholder '0..3' in value "sjds${0..3}.auth_user"

2. 解決辦法： 配置: name="ignoreUnresolvablePlaceholders"value="true"/> 或者: ****** ignore-unresolvable="true"/>

5. Main測試

Main.java用來測試分庫分表是否OK，其原始碼如下：

/**
 * @author wangzhenfei9
 * @version 1.0.0
 * @since 2018年02月08日
 */
public class Main {
    public static void main(String[] args) {
        ApplicationContext context = new ClassPathXmlApplicationContext(
                "/META-INF/spring/spring-*.xml");
        // auth_user有進行分庫，
        AuthUserMapper authUserMapper = context.getBean(AuthUserMapper.class);
        AuthUser authUser = authUserMapper.selectByPrimaryKey(7L);
        System.out.println("-----> The auth user: "+JSON.toJSONString(authUser));
        // user_permission沒有分庫分表
        UserPermissionMapper userPermissionMapper = context.getBean(UserPermissionMapper.class);
        UserPermission userPermission = userPermissionMapper.selectPermissionByUsername("wangzhenfei", "FINANCE_WALLET");
        System.out.println("-----< The user permission: "+JSON.toJSONString(userPermission));
    }
}

AuthUserMapper.selectByPrimaryKey()和UserPermissionMapper.selectPermissionByUsername()的程式碼和沒有分庫分表的程式碼完全一樣；

6. 遺留問題

Main方法測試，或者啟動服務後的呼叫測試都沒有問題，但是透過junit測試用例訪問就會丟擲異常，作為一個待解決的遺留問題：

org.springframework.beans.factory.NoUniqueBeanDefinitionException: No qualifying bean of type 'javax.sql.DataSource' available: expected single matching bean but found 6: sj_ds_0,sj_ds_1,sj_ds_2,sj_ds_3,sj_ds_default,shardingDataSource

END

阿飛哥

贊賞

長按二維碼向我轉賬

受蘋果公司新規定影響，微信 iOS 版的贊賞功能被關閉，可透過二維碼轉賬支援公眾號。

閱讀原文


微信掃一掃
使用小程式
即將開啟""小程式
取消
開啟