merage

kimi

2020-05-27 c007f0ca1785db093d48f4846cda82fe8e955765

 src/main/java/com/highdatas/mdm/service/impl/SysAssembleDbServiceImpl.java

@@ -1,20 +1,15 @@
package com.highdatas.mdm.service.impl;

import com.alibaba.druid.pool.DruidDataSource;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.highdatas.mdm.entity.SysAssemble;
import com.baomidou.mybatisplus.service.impl.ServiceImpl;
import com.highdatas.mdm.entity.SysAssembleDb;
import com.highdatas.mdm.entity.SysAssembleDbdriver;
import com.highdatas.mdm.mapper.SysAssembleDbMapper;
import com.highdatas.mdm.pojo.kettle.UnBigDataDataSourceInfo;
import com.highdatas.mdm.service.ISysAssembleDbService;
import com.baomidou.mybatisplus.service.impl.ServiceImpl;
import com.highdatas.mdm.service.ISysAssembleDbdriverService;
import com.highdatas.mdm.service.ISysAssembleService;
import com.highdatas.mdm.util.Constant;
import org.apache.commons.dbutils.DbUtils;
import org.apache.commons.lang3.StringUtils;
import org.pentaho.di.core.KettleClientEnvironment;
import org.pentaho.di.core.database.DatabaseMeta;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

@@ -22,6 +17,7 @@
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

/**
 * <p>
@@ -43,12 +39,18 @@
            return null;
        }

        return getConnection(assemble);

    }

    private Connection getConnection(SysAssembleDb assemble) {
        Connection connection  = null;
        try {
            SysAssembleDbdriver sysAssembleDbdriver = dbdriverService.selectOne(new EntityWrapper<SysAssembleDbdriver>().eq(Constant.TYPE, assemble.getDatasourceType()));
            String driver = "com.mysql.cj.jdbc.Driver";
            if (sysAssembleDbdriver != null) {
                String driver = sysAssembleDbdriver.getDriver();
                Class.forName(driver).newInstance();
                driver = sysAssembleDbdriver.getDriver();
//                dataSource.setDriverClassName(driver);c
            }

//            Boolean bigdata = assemble.getBigdata();
@@ -58,16 +60,47 @@
//                dataMetaOutput = new DatabaseMeta("Output", unBigDataDataSourceInfo.getDbType(), "Native", unBigDataDataSourceInfo.getDbHostName(), unBigDataDataSourceInfo.getDbName(), unBigDataDataSourceInfo.getDbPort(),
//                        unBigDataDataSourceInfo.getUsername(), unBigDataDataSourceInfo.getPassword());
//            }
            DriverManager.setLoginTimeout(5);
            String datasourcePass = assemble.getDatasourcePass();
            try {
                datasourcePass = com.datacvg.common.EncryptUtils.decrypt(datasourcePass);
            }catch (Exception e){
                e.printStackTrace();
            }

            connection = DriverManager.getConnection(assemble.getDatasourceUrl(), assemble.getDatasourceUser(), assemble.getDatasourcePass());
//            dataSource.setUrl(assemble.getDatasourceUrl());
//            dataSource.setUsername(assemble.getDatasourceUser());
//            dataSource.setPassword(datasourcePass);
//            connection = dataSource.getConnection();
            Properties properties = new Properties();
            properties.setProperty("user",assemble.getDatasourceUser());
            properties.setProperty("password",datasourcePass);
            DbUtils.loadDriver(driver);
//            Class.forName(driver);
            String datasourceUrl = assemble.getDatasourceUrl();
            if (!datasourceUrl.contains(Constant.timeZone)) {
                datasourceUrl += "&serverTimezone=Asia/Shanghai";
            }
            datasourceUrl += "&nullCatalogMeansCurrent=true";

            connection = DriverManager.getConnection(datasourceUrl,properties);
            return connection;
        }  catch (Exception e) {
                e.printStackTrace();
                return null;
            }

    }


    public boolean testDatasource(String key,String driveClass,String url,String username,String password){
        try {
            Class.forName(driveClass);
            DriverManager.getConnection(url, username, password);
            return true;
        } catch (Exception e) {
            return false;
        }
    }
    @Override
    public List<String> getFieldsByTableName(String parentId, String tableName) {
        SysAssembleDb assemble = this.selectById(parentId);
@@ -112,6 +145,37 @@
    }

    @Override
    public boolean testConnection(SysAssembleDb sysAssembleDb) {
        Connection connection = null;
        try{
            connection = getConnection(sysAssembleDb);
            if (connection != null) {
                return true;
            }
            return false;
        }
        catch (Exception e) {
            e.printStackTrace();
            return false;
        }
        finally {
            if (connection != null) {
                try {
                    connection.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
    }
    /**
     *
     * @description:  测试连接
     * @param parentId 汇集id
     * @return: 是否测试连接成功
     *
     */
    @Override
    public boolean testConnection(String parentId) {
        Connection connection = null;
        try{
@@ -136,7 +200,14 @@
        }

    }

    /**
     *
     * @description:  测试sql是否语法正常
     * @param sql sql
     * @param parentId 汇集id
     * @return: 语法是否正确
     *
     */
    @Override
    public boolean testSql(String parentId, String sql) {
        Connection connection = null;
@@ -170,7 +241,14 @@
            }
        }
    }

    /**
     *
     * @description:  通过sql 获取sql 内使用的字段
     * @param sql sql
     * @param parentId 汇集id
     * @return: 字段列表
     *
     */
    @Override
    public List<String> getFieldsBySql(String parentId, String sql) {
        Connection connection = null;