MyBatis动态创建SqlSessionFactory

深碍√TFBOYSˉ_ 2022-12-04 01:07 16阅读 0赞

## 一、前言 ##

云平台上运行了很多项目，根据项目id动态的切换数据库，mybatis-config.xml启动的时候只会加载一次，很显然无法满足。每次有新项目上线都需要改配置文件然后重启，太low了，所以需要自己根据项目id手动创建SqlSessionFactory进行数据库连接。

原理：将SqlSessionFactory放入Map集合里，判断对应项目id是否创建SqlSessionFactory，若没创建就创建新的，Map有点话就更加动态切换。

动态创建SqlSessionFactory有两种方式：

第一种：是在程序中构建这些对象来创建；

第二种：是依赖springboot来实现；这种方式打jar提交storm集群的时候始终找不到主类，在spring-boot-maven-plugin和maven-shade-plugin之间痛苦了很久果断放弃了，如果是单纯的java或web项目可以用这样方式，毕竟springboot还是很高效的。接下来说的是第一种。

## 二、**在程序中构建**SqlSessionFactory ##

MyBatis框架主要是围绕着SqlSessionFactory这个类进行的，这个的创建过程如下：

1.  定义一个Configuration对象，其中包含数据源、事务、mapper文件资源以及影响数据库行为属性设置settings
2.  通过配置对象，则可以创建一个SqlSessionFactoryBuilder对象
3.  通过 SqlSessionFactoryBuilder 获得SqlSessionFactory 的实例。
4.  SqlSessionFactory 的实例可以获得操作数据的SqlSession实例，通过这个实例对数据库进行操作

### 1、重新定义MybatisUtil ###

package com.siger.storm.utils;
    
    import com.siger.storm.mapper.*;
    import com.siger.storm.model.TableName;
    import org.apache.ibatis.datasource.pooled.PooledDataSource;
    import org.apache.ibatis.mapping.Environment;
    import org.apache.ibatis.session.Configuration;
    import org.apache.ibatis.session.SqlSession;
    import org.apache.ibatis.session.SqlSessionFactory;
    import org.apache.ibatis.session.SqlSessionFactoryBuilder;
    import org.apache.ibatis.transaction.TransactionFactory;
    import org.apache.ibatis.transaction.jdbc.JdbcTransactionFactory;
    import org.apache.log4j.LogManager;
    import org.apache.log4j.Logger;
    
    import java.util.concurrent.ConcurrentHashMap;
    
    public class MybatisUtil {
    
        public static final Logger log = LogManager.getLogger(MybatisUtil.class);
        public static com.siger.storm.constant.Configuration config = com.siger.storm.constant.Configuration.getInstance();
        public static ConcurrentHashMap sessionHashMap = new ConcurrentHashMap();
        private static SqlSessionFactory sessionFactory;
    
        public static synchronized SqlSession getSqlSession() {
            SqlSession session = null;
    
            try {
                if (sessionHashMap.get(TableName.getTableName()) != null) {
                    sessionFactory = (SqlSessionFactory) sessionHashMap.get(TableName.getTableName());
                    if (sessionFactory.openSession() != null) {
                        session = sessionFactory.openSession();
                    }
                } else {
                    sessionFactory = CreateSessionFactory(TableName.getTableName());
                    if (sessionFactory != null) {
                        sessionHashMap.put(TableName.getTableName(), sessionFactory);
                        session = sessionFactory.openSession();
                        if (sessionFactory.openSession() != null) {
                            session = sessionFactory.openSession();
                        }
                    } else {
                        log.warn("sessionFactory is null...");
                    }
                }
            } catch (Exception e) {
                log.error(e.getMessage() + "-->catch Exception");
            }
            return session;
        }
    
    
        /**
         * 根据库名动态创建会话
         *
         * @param dbName
         * @return
         */
        public static synchronized SqlSessionFactory CreateSessionFactory(String dbName) {
    
            //创建使用缓存池的数据源
    		/* 
    		 * <dataSource type="POOLED">
    				<property name="driver" value="${jdbc.driverClassName}"/>
    				<property name="url" value="${jdbc.url}"/>
    				<property name="username" value="${jdbc.username}"/>
    				<property name="password" value="${jdbc.password}"/>
    			</dataSource>
    		 */
    
            // 数据库连接
            PooledDataSource dataSource = new PooledDataSource();
            dataSource.setDriver(config.getString("cloud.mysql.jdbc.driver", ""));
            dataSource.setUsername(config.getString("cloud.mysql.user", ""));
            dataSource.setPassword(config.getString("cloud.mysql.password", ""));
            dataSource.setUrl(config.getString("cloud.mysql.url", "") + dbName +
                    "?serverTimezone=Asia/Shanghai&characterEncoding=utf8&useSSL=false&autoReconnect=true");
    
            //创建事务
            /*
             * <transactionManager type="JDBC" />
             */
            // 事务
            TransactionFactory transactionFactory = new JdbcTransactionFactory();
            Environment environment = new Environment(dbName, transactionFactory, dataSource);
    
            //加入资源
            /*
             * <mapper resource="ssm/BlogMapper.xml"/>
             */
            // 创建configuration信息, 添加映射器
            // **Mapper.xml 要放在 **Mapper.class 同一级目录下
            Configuration configuration = new Configuration(environment);
            configuration.addMapper(xxMapper.class);
    
    
            SqlSessionFactory sqlSessionFactory = new SqlSessionFactoryBuilder().build(configuration);
            log.info(dbName + " sqlSessionFactory created successfully...");
    
            return sqlSessionFactory;
        }
    
    }

2、新建TableName实体类缓存库名

package com.siger.storm.model;
    
    public class TableName {
    
        public static ThreadLocal<String>tablename = new ThreadLocal<>();
    
        public static void setTableName(String cid_pid){
            tablename.set(cid_pid);
        }
    
        public static String getTableName(){
            return tablename.get();
        }
    
        public static void removeTableName(){
            tablename.remove();
        }
    
    }

3、写个测试类测试一下

至于实体类、xxMapper.class和xMapper.xml生成就不多说了，可以用generator自动生成，log和读取配置文件常量类自定义

package com.siger.storm;
    
    import com.siger.storm.dao.impl.ProductionDaoImpl;
    import com.siger.storm.model.Production;
    import com.siger.storm.model.TableName;
    
    
    /**
     * 描述: 测试
     * author: weiy
     * create: 2020-09-04 18:12
     */
    public class test {
    
        public static ProductionDaoImpl pDaoImpl = new ProductionDaoImpl();
    
        public static void main(String[] args) {
            TableName.setTableName("12_79");
    
            final Production production = new Production();
            production.setMachineid(23);
            production.setStatus(1);
            pDaoImpl.insert(production);
        }
    
    }

## 三、注意事项 ##

1、xxMapper.class 和 xxMapper.xml必须在同一级目录下，不然无法识别，底层就是这样定义的，有兴趣朋友可以看下源码。

2、pom.xml需如下配置，不然加载不到src/main/java目录下的xxMapper.xml

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIxMTUzNjE5_size_16_color_FFFFFF_t_70][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIxMTUzNjE5_size_16_color_FFFFFF_t_70]: /images/20221123/23212e1c2ed944849aad7306fb5f4279.png