ShardingContextHolder.java 487 Bytes
Newer Older
eddie.woo's avatar
eddie.woo committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
package pwc.taxtech.atms.common.datasource;

public class ShardingContextHolder {
    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    /**
     * 设置数据库名
     *
     * @param key key
     */
    public static void setDataSourceKey(String key) {
        contextHolder.set(key);
    }

    /**
     * 获取数据库名
     *
     * @return key
     */
    public static String getDataSourceKey() {
        return contextHolder.get();
    }
}