基于druid和spring的动态数据库以及读写分离-白红宇

基于druid和spring的动态数据库以及读写分离

阅读量：6231 次

发布时间：2019-06-21

本文共 3649 字，大约阅读时间需要 12 分钟。

spring与druid可以实现动态数据源，夸库查询，读写分离等功能。现在说一下配置：

1、需要配置多个spring数据源

spring-data.xml

2、需要写一个DynamicDataSource类继承AbstractRoutingDataSource，并实现determineCurrentLookupKey方法

public class DynamicDataSource extends AbstractRoutingDataSource {  	          /**      *       * override determineCurrentLookupKey      *       * Title: determineCurrentLookupKey      * 
      *       * Description: 自动查找datasource      * 
      *       * @return      */      @Override      protected Object determineCurrentLookupKey() {          return DBContextHolder.getDSType();      }    }

3、参考spring事务管理，使用线程变量来切换数据源

public class DBContextHolder {	/**	 * 线程threadlocal	 */	private static ThreadLocal
     
       contextHolder = new ThreadLocal<>();	private static Logger logger = LoggerFactory			.getLogger(DBContextHolder.class);	public static String getDSType() {		try {		} catch (Exception e) {			e.printStackTrace();			logger.error("get DBTYPE faild with error:[" + e.getMessage() + "]");		}		String db = contextHolder.get();		 if (db == null) {		 db =UrlConnect.getKey(ConfigHelper.getToWriteKey());// 默认是读写库		 }		return db;	}	/**	 * 	 * 设置本线程的dbtype	 * 	 * @param str	 * @see [相关类/方法](可选)	 * @since [产品/模块版本](可选)	 */	public static boolean setDSType(String str) {		try {			clearDBType();			if (str != null&&!str.equals("")) {				contextHolder.set(str);				logger.info("change thread[" + str + "] success!");				return true;			} else {				logger.info("change thread[" + str + "] faild!");				return false;			}		} catch (Exception e) {			e.printStackTrace();			logger.error("change thread[" + str + "] faild!");			return false;		}	}	/**	 * clearDBType	 * 	 * @Title: clearDBType	 * @Description: 清理连接类型	 */	public static void clearDSType() {		contextHolder.remove();	}}

4、在dao中切换数据源

@Repositorypublic class BaseDAO extends SqlSessionDaoSupport {       @Resource	private SqlSessionTemplate sqlSessionTemplate;@Resource	public void setSqlSessionTemplate(SqlSessionTemplate sqlSessionTemplate) {		super.setSqlSessionTemplate(sqlSessionTemplate);	}public  
     
       PageList
      
        selectPublicListPage(String countSqlID,String sqlID,			PageList
       
         page, Object obj) {		DBContextHolder.setDbType("dataSourceRead");		//查询总数		Integer total = this.getSqlSession().selectOne(countSqlID, obj);		RowBounds rowBounds=new RowBounds(page.getFirstResult(),page.getPageSize());		// 查询列表信息		List
        
          list = this.getSqlSession().selectList(				sqlID, obj,rowBounds);		page.setTotalRecord(total!=null?total:0);		page.setDataSource(list);		page.setTotalPage((total + page.getPageSize() - 1)				/ page.getPageSize());		return page;	}	public int insert(String sqlID, Object paramObj) {		DBContextHolder.setDbType("dataSourceWrite");		return this.getSqlSession().insert(sqlID, paramObj);	}}