jboot的动态配置数据源_JAVA_编程开发_程序员俱乐部

中国优秀的程序员网站程序员频道CXYCLUB技术地图
热搜:
更多>>
 
您所在的位置: 程序员俱乐部 > 编程开发 > JAVA > jboot的动态配置数据源

jboot的动态配置数据源

 2019/7/11 13:09:21  sensejw  程序员俱乐部  我要评论(0)
  • 摘要:Jboot通过ShardingJDBC实现分库分表,但是分库分表函数主要是通过字段值进行计算,但如果分库方式与数据无关,并且分库需要在运行时动态新增配置就无法满足要求。下面是通过改造基本Model类,并动态配置数据源的方法:1、新增数据源缓存类注意,configName要使用线程本地类,JBoot非线程安全。publicclassDataConfigName{privatestaticDataConfigNameme;@JSONField(serialize=false
  • 标签:配置 数据
Jboot通过ShardingJDBC实现分库分表,但是分库分表函数主要是通过字段值进行计算,但如果分库方式与数据无关,并且分库需要在运行时动态新增配置就无法满足要求。
下面是通过改造基本Model类,并动态配置数据源的方法:

1、新增数据源缓存类
   注意,configName要使用线程本地类,JBoot非线程安全
class="java" name="code">
public class DataConfigName {
    private static DataConfigName me;
    @JSONField(serialize = false)
    transient private ThreadLocal<String> configName = new ThreadLocal<>();

    private DataConfigName(){}

    public static DataConfigName getInstance(){
        if(me == null){
            me = new DataConfigName();
        }
        return me;
    }

    public void setConfigName(String name){
        configName.set(name);
    }

    public String getConfigName(){
        String name = configName.get();
        if(null == name){
            name = DbKit.MAIN_CONFIG_NAME;
        }
        return name;
    }
}


2、修改Model基础类
   JBoot操作数据库主要通过Model来实现,框架提供了基础类JBootModel,需要对这个类进行扩展,覆盖所有方法,如下所示,方法中都增加了use子句,实现数据源的切换,否则会使用默认数据源:
public class GlueModel<M extends GlueModel<M>> extends JbootModel<M> {
    public GlueModel(){};
    ……
    

    public M use(String configName) {
        return super.use(configName);
    }

    M superUse(String configName) {
        return super.use(configName);
    }

    public boolean saveOrUpdate() {
        this.use(DataConfigName.getInstance().getConfigName());
        return super.saveOrUpdate();
    }

    public boolean save() {
        this.use(DataConfigName.getInstance().getConfigName());
        return super.save();
    }
    ……
}


3、替换所有Db类的操作,增加use子句,如:
public class TransactionModel extends GlueModel<TransactionModel> {

    ……

    public static TransactionModel getByTno(String tno){
        ……
        Record record = Db.use(DataConfigName.getInstance().getConfigName()).findFirst(sql);
        ……
    }
    ……
}


4、创建一个数据源动态管理类。
Jboot虽然在2.1.2版本的DataSourceConfigManager类中开放了addConfig方法,但由于Db类是属于JFinal框架的,无法影响到,因此需要调用JFinal的DbKit的addConfig方法完成数据源的新增。
下面的代码只提供了从默认数据库获取数据源配置参数的实现方式,如果需要通过本地配置文件、网络文件等方式,可参考自行添加
public class DynamicDataSourceManager {
    /**
     * 从数据库获取数据源配置参数
     * @param tableName
     * @param corpId
     * @return
     */
    public static Config loadConfigFromDB(String tableName, String corpId){
        Config config = DbKit.getConfig(corpId);
        if(null == config) {
            SqlPara sql = new SqlPara();
            sql.setSql("select * from " + tableName + " where corpid=?").addPara(corpId);
            List<Record> records = Db.find(sql);
            if (null != records && records.size()>0) {
                Record record = records.get(0);
                DataSourceConfig dataSourceConfig = new DataSourceConfig();
                dataSourceConfig.setName(record.getStr("corpid"));
                dataSourceConfig.setType(DataSourceConfig.TYPE_MYSQL);
                dataSourceConfig.setDriverClassName("com.mysql.cj.jdbc.Driver");
                dataSourceConfig.setUrl("jdbc:mysql://"
                        + record.getStr("write_host")
                        + ":" + record.getStr("write_port")
                        + "/"
                        + record.getStr("write_database")
                        + "?useUnicode=true&characterEncoding="
                        + record.getStr("charset")
                        + "&useSSL=false");
                dataSourceConfig.setUser(record.getStr("write_username"));
                dataSourceConfig.setPassword(record.getStr("write_password"));
                dataSourceConfig.setPoolName(record.getStr("write_database"));

                DataSource dataSource = new DataSourceBuilder(dataSourceConfig).build();

                config = new Config(
                        record.getStr("corpid"),
                        dataSource,
                        new MysqlDialect(),
                        false,
                        false,
                        DbKit.DEFAULT_TRANSACTION_LEVEL,
                        IContainerFactory.defaultContainerFactory,
                        new EhCache()
                );
                DbKit.addConfig(config);
            }
        }
        return config;
    }
}


5、在切片方法或入口方法中实现数据源的配置
    ……
    if(null != corpId){
        Config config = DynamicDataSourceManager.loadConfigFromDB("corp_database", corpId);
        if(null != config) {
            DataConfigName.getInstance().setConfigName(corpId);
        }
     }
    ……


通过上述修改即可完成动态分库处理,实际的Model中无需再做任何特殊处理。
发表评论
用户名: 匿名