package com.example.demo.config;

import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.SQLException;
import org.apache.tomcat.jdbc.pool.DataSource;
import org.apache.tomcat.jdbc.pool.PoolProperties;


/**
 * Created by Administrator on 2019/8/5.
 */

public class DynamicDataSource extends DataSource {
    //private static Logger log = LogManager.getLogger(DynamicDataSource.class);

    /**
     * 改写本方法是为了在请求不同工程的数据时去连接不同的数据库。
     */
    @Override
    public Connection getConnection() {
        String projectCode = DBIdentifier.getProjectCode();
        System.out.println(projectCode+"projectCode");
        //1、获取数据源
        DataSource dds = DDSHolder.instance().getDDS(projectCode);
        System.out.println(dds+"dds");
        //2、如果数据源不存在则创建
        if (dds == null) {
            try {
                DataSource newDDS = initDDS(projectCode);
                DDSHolder.instance().addDDS(projectCode, newDDS);
            } catch (IllegalArgumentException | IllegalAccessException e) {
                //log.error("Init data source fail. projectCode:" + projectCode);
                return null;
            }
        }
        dds = DDSHolder.instance().getDDS(projectCode);
        try {
            return dds.getConnection();
        } catch (SQLException e) {
            e.printStackTrace();
            return null;
        }
    }

    /**
     * 以当前数据对象作为模板复制一份。
     *
     * @return dds
     * @throws IllegalAccessException
     * @throws IllegalArgumentException
     */
    private DataSource initDDS(String projectCode) throws IllegalArgumentException, IllegalAccessException {
        DataSource dds = new DataSource();
        // 2、复制PoolConfiguration的属性
        PoolProperties property = new PoolProperties();
        Field[] pfields = PoolProperties.class.getDeclaredFields();
        for (Field f : pfields) {
            f.setAccessible(true);
            Object value = f.get(this.getPoolProperties());
            try {
                f.set(property, value);
            } catch (Exception e) {
                //log.info("Set value fail. attr name:" + f.getName());
                continue;
            }
        }
        dds.setPoolProperties(property);
        // 3、设置数据库名称和IP(一般来说，端口和用户名、密码都是统一固定的)
        String urlFormat = this.getUrl();
        String username=this.getUsername();
        String password= this.getPassword();
        System.out.println("urlFormat"+urlFormat);
        System.out.println("getUsername"+username);
        System.out.println("getPassword"+password);
        String url = String.format(urlFormat, ProjectDBMgr.instance().getDBIP(projectCode),
                ProjectDBMgr.instance().getDBName(projectCode));
        System.out.println("url"+url);
        dds.setUrl(url);
        dds.setUsername(ProjectDBMgr.instance().getUsername(projectCode));
        dds.setPassword(ProjectDBMgr.instance().getPassword(projectCode));
        return dds;
    }
}

package com.example.demo.config;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Timer;
import org.apache.tomcat.jdbc.pool.DataSource;
/**
 * Created by Administrator on 2019/8/5.
 */
public class DDSHolder {
    /**
     * 管理动态数据源列表。<工程编码，数据源>
     */
    private Map<String, DDSTimer> ddsMap = new HashMap<String, DDSTimer>();
    /**
     * 通过定时任务周期性清除不使用的数据源
     */
    private static Timer clearIdleTask = new Timer();
    static {
        clearIdleTask.schedule(new ClearIdleTimerTask(), 5000, 60 * 1000);
    };
    private DDSHolder() {
    }
    /*
    * 获取单例对象
    */
    public static DDSHolder instance() {
        return DDSHolderBuilder.instance;
    }
    /**
     * 添加动态数据源。
     *
     * @param projectCode 项目编码
     * @param dds dds
     */
    public synchronized void addDDS(String projectCode, DataSource dds) {
        DDSTimer ddst = new DDSTimer(dds);
        ddsMap.put(projectCode, ddst);
    }
    /**
     * 查询动态数据源
     *
     * @param projectCode 项目编码
     * @return dds
     */
    public synchronized DataSource getDDS(String projectCode) {
        if (ddsMap.containsKey(projectCode)) {
            DDSTimer ddst = ddsMap.get(projectCode);
            ddst.refreshTime();
            return ddst.getDds();
        }
        return null;
    }
    /**
     * 清除超时无人使用的数据源。
     */
    public synchronized void clearIdleDDS() {
        Iterator<Entry<String, DDSTimer>> iter = ddsMap.entrySet().iterator();
        for (; iter.hasNext(); ) {
            Entry<String, DDSTimer> entry = iter.next();
            if (entry.getValue().checkAndClose())
            {
                iter.remove();
            }
        }
    }
    /**
     * 单例构件类
     * @author elon
     * @version 2018年2月26日
     */
    private static class DDSHolderBuilder {
        private static DDSHolder instance = new DDSHolder();
    }
}

package com.example.demo.config;    
import com.example.demo.util.DBData;    
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;    
/**
 *  * 项目数据库管理。提供根据项目编码查询数据库名称和IP的接口。
 *  * @author wu
 *  * @version 2 2019/8/5.
 *  格式化  ctrl+ALT+;
 */
public class ProjectDBMgr {
    /**
     *  * 保存项目编码与数据名称的映射关系。这里是硬编码，实际开发中这个关系数据可以保存到redis缓存中；
     *  * 新增一个项目或者删除一个项目只需要更新缓存。到时这个类的接口只需要修改为从缓存拿数据。
     *  
     */
    private Map<String, DBData> depDBMap = new HashMap<>();
    //private Map<String, String> dbNameMap = new HashMap<String, String>();
    /**
     *  * 保存项目编码与数据库IP的映射关系。
     *  
     */
    //private Map<String, String> dbIPMap = new HashMap<String, String>();

    private ProjectDBMgr() {
    //注意你的项目用到的数据库不同，如果表结果也不同的话 那你得写不同的bean  
        depDBMap.put("5642",new DBData("5642","青和居西区","10.10.10.10","3306","数据库名字","账号","密码");
        depDBMap.put("5455",new DBData("5455","八大家花西园","10.10.10.10","3306","数据库名字","账号","密码"));
        depDBMap.put("5643",new DBData("5643","青和居西区","10.10.10.10","3306","数据库名字","账号","密码"));
        depDBMap.put("5454",new DBData("5454","八大家花东园","10.10.10.10","3306","数据库名字","账号","密码"));
        .
        .
        .
        
    }

    public static ProjectDBMgr instance() {
        return ProjectDBMgrBuilder.instance;
    }

    // 实际开发中改为从缓存获取
    public String getDBName(String projectCode) {
        if (depDBMap.containsKey(projectCode)) {
            return depDBMap.get(projectCode).getDbName();
        }
        return "";
    }

    //实际开发中改为从缓存中获取
    public String getDBIP(String projectCode) {
        if (depDBMap.containsKey(projectCode)) {
            return depDBMap.get(projectCode).getIp();
        }
        return "";
    }
    //动态获取密码
    public String getPassword(String projectCode) {
        if (depDBMap.containsKey(projectCode)) {
            return depDBMap.get(projectCode).getPassword();
        }
        return "";
    }
    //动态获取username
    public String getUsername(String projectCode) {
        if (depDBMap.containsKey(projectCode)) {
            return depDBMap.get(projectCode).getUser();
        }
        return "";
    }

    private static class ProjectDBMgrBuilder {
        private static ProjectDBMgr instance = new ProjectDBMgr();
    }
}