动态数据源 mysql

前言:

     最近开发时遇到这样一个场景: 在一段定时任务中需要将A数据库中查询的结果处理后,然后插入B数据库,这是就考虑到用动态数据源,在这里记录一下配置过程,仅供参考:

一: 在spring.xml添加配置

二: 添加相应的工具类

public interface DataSourceEntry {

    // 默认数据源
    public final static String DEFAULT_SOURCE = "dataSource1";

    /**
     * 还原数据源
     *
     * @param join
     */
    public void restore(JoinPoint join);

    /**
     * 设置数据源
     *
     * @param source
     */
    public void set(String source);

    /**
     * 获取数据源
     *
     * @return String
     */
    public String get();

    /**
     * 清空数据源
     */
    public void clear();
}

/**
 * 数据源切换实现类类
 *
 * @create 2013.04.03
 */
public class DataSourceEntryImpl implements DataSourceEntry {

    private final static ThreadLocal local = new ThreadLocal();

    public void clear() {
        local.remove();
    }

    public String get() {
        return local.get();
    }

    public void restore(JoinPoint join) {
        local.set(DEFAULT_SOURCE);
    }

    public void set(String source) {
        local.set(source);
    }

}

public class DbContextHolder {
    @SuppressWarnings("rawtypes")

    private static final ThreadLocal contextHolder = new ThreadLocal();

    @SuppressWarnings("unchecked")
    public static void setDbType(String dbType) {
        contextHolder.set(dbType);
    }

    public static String getDbType() {
        return (String) contextHolder.get();
    }

    public static void clearDbType() {
        contextHolder.remove();
    }
}

public class DynamicDataSource extends AbstractRoutingDataSource {


    private DataSourceEntry dataSourceEntry;
    @Override
    protected Object determineCurrentLookupKey() {
        return DbContextHolder.getDbType();
    }

    @Resource
    public void setDataSourceEntry(DataSourceEntry dataSourceEntry) {
        this.dataSourceEntry = dataSourceEntry;
    }
}

三: 具体的是业务方法

public void writeLocalData(String startTime,String endTime) {
		try {
			System.out.println(sdf1.format(new Date()));
			//日报进行记录
			List carList1 = dataServise.oneData(startTime,endTime);
			//切换到新的数据库
			DbContextHolder.setDbType("dataSource2");
			//向新的数据库插入数据
			dataServise.addLocalData(carList1);
			System.out.println(sdf1.format(new Date()));
		} catch (Exception e) {
			e.printStackTrace();
		}
	}