mec-cloud_IntelligentManufacturing_DB/src/main/java/com/customer/utils/DataSourceUtils.java

package com.customer.utils;

import com.customer.config.GenConfig;
import com.customer.constant.DataSourceType;
import com.customer.pojo.DataSource;

public class DataSourceUtils {

    public static GenConfig changeDataSource(DataSource dataSource) {


        GenConfig genConfig = new GenConfig();
        genConfig.setDbtype(dataSource.getDatabaseType());
        genConfig.setUsername(dataSource.getUsername());
        genConfig.setPassword(dataSource.getPassword());
        if(dataSource.getDatabaseType().equals(DataSourceType.MYSQL.getDataSourceName())){
            genConfig.setDriverClassName("com.mysql.cj.jdbc.Driver");
            genConfig.setUrl("jdbc:mysql://"+dataSource.getDatabaseIp()+":"+dataSource.getPortNumber()+"/"+dataSource.getDatabaseName());
        }else if(dataSource.getDatabaseType().equals(DataSourceType.SQLSERVER.getDataSourceName())){
            genConfig.setDriverClassName("com.microsoft.sqlserver.jdbc.SQLServerDriver");
            genConfig.setUrl("jdbc:sqlserver://"+dataSource.getDatabaseIp()+":"+dataSource.getPortNumber()+";DatabaseName="+dataSource.getDatabaseName());
        }else if(dataSource.getDatabaseType().equals(DataSourceType.ORACLE.getDataSourceName())){
            genConfig.setDriverClassName("oracle.jdbc.driver.OracleDriver");
            genConfig.setUrl("jdbc:oracle:thin:@"+dataSource.getDatabaseIp()+":"+dataSource.getPortNumber()+dataSource.getDatabaseName());
        }else if(dataSource.getDatabaseType().equals(DataSourceType.DM.getDataSourceName())){
            genConfig.setDriverClassName("dm.jdbc.driver.DmDriver");
            genConfig.setUrl("jdbc:dm://"+dataSource.getDatabaseIp()+":"+dataSource.getPortNumber()+"?schema="+dataSource.getDatabaseName());
        }
        return genConfig;
    }
}