9 changed files with 162 additions and 104 deletions
@ -0,0 +1,124 @@ |
|||||||
|
package com.fr.design.data.datapane.preview; |
||||||
|
|
||||||
|
import com.fr.data.auth.AuthenticationType; |
||||||
|
import com.fr.data.auth.kerberos.KerberosAuthentication; |
||||||
|
import com.fr.data.auth.kerberos.KerberosUtils; |
||||||
|
import com.fr.data.impl.Connection; |
||||||
|
import com.fr.data.impl.JDBCDatabaseConnection; |
||||||
|
import com.fr.data.pool.DBCPConnectionPoolAttr; |
||||||
|
import com.fr.data.security.ssh.BaseSsh; |
||||||
|
import com.fr.data.security.ssh.SshType; |
||||||
|
import com.fr.data.security.ssh.impl.KeyVerifySsh; |
||||||
|
import com.fr.data.security.ssl.BaseSsl; |
||||||
|
import com.fr.data.security.ssl.SslType; |
||||||
|
import com.fr.data.security.ssl.impl.NormalSsl; |
||||||
|
import com.fr.decision.privilege.TransmissionTool; |
||||||
|
import com.fr.decision.webservice.bean.datasource.ConnectionInfoBean; |
||||||
|
import com.fr.decision.webservice.bean.datasource.JDBCConnectionBean; |
||||||
|
import com.fr.decision.webservice.utils.DecisionServiceConstants; |
||||||
|
import com.fr.decision.webservice.v10.datasource.connection.processor.impl.ConnectionProcessorFactory; |
||||||
|
import com.fr.decision.webservice.v10.datasource.connection.processor.impl.JDBCConnectionProcessor; |
||||||
|
import com.fr.security.encryption.transmission.TransmissionEncryptors; |
||||||
|
import com.fr.stable.StringUtils; |
||||||
|
import com.fr.third.fasterxml.jackson.databind.ObjectMapper; |
||||||
|
|
||||||
|
/** |
||||||
|
* 数据连接传输工具类 |
||||||
|
* |
||||||
|
* @author Destiny.Lin |
||||||
|
* @since 11.0 |
||||||
|
* Created on 2024/7/25 |
||||||
|
*/ |
||||||
|
public class ConnectionInfoBeanHelper { |
||||||
|
private static ObjectMapper objectMapper = new ObjectMapper(); |
||||||
|
|
||||||
|
/** |
||||||
|
* 创建数据连接Bean，可自定义是否携带密码 |
||||||
|
*/ |
||||||
|
public static ConnectionInfoBean createConnectionInfoBean(String name, Connection connection, boolean withPassword) throws Exception { |
||||||
|
if (JDBCConnectionProcessor.KEY.acceptConnections().contains(connection.getClass())) { |
||||||
|
ConnectionInfoBean bean = new ConnectionInfoBean(); |
||||||
|
bean.setConnectionData(objectMapper.writeValueAsString(convertToJDBCConnectionBean(connection, withPassword))); |
||||||
|
bean.setConnectionType(JDBCConnectionProcessor.CONNECTION_TYPE); |
||||||
|
bean.setConnectionType(JDBCConnectionProcessor.KEY.getConnectionType(connection)); |
||||||
|
bean.setConnectionName(name); |
||||||
|
bean.setCreator(connection.getCreator()); |
||||||
|
return bean; |
||||||
|
} else { |
||||||
|
return ConnectionProcessorFactory.createConnectionInfoBean(name, connection); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 创建数据连接Bean，默认携带密码 |
||||||
|
*/ |
||||||
|
public static ConnectionInfoBean createConnectionInfoBean(String name, Connection connection) throws Exception { |
||||||
|
return createConnectionInfoBean(name, connection, true); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 创建数据连接bean |
||||||
|
*/ |
||||||
|
public static ConnectionInfoBean createConnectionInfoBean(Connection connection) throws Exception { |
||||||
|
return createConnectionInfoBean(connection.getConnectionName(), connection, true); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/** |
||||||
|
* jdbc的特殊处理 |
||||||
|
*/ |
||||||
|
private static JDBCConnectionBean convertToJDBCConnectionBean(Connection connection, boolean withPassword) { |
||||||
|
JDBCDatabaseConnection jdbcConnection = (JDBCDatabaseConnection) connection; |
||||||
|
JDBCConnectionBean jdbcConnectionBean = new JDBCConnectionBean(); |
||||||
|
DBCPConnectionPoolAttr poolAttr = new DBCPConnectionPoolAttr(); |
||||||
|
try { |
||||||
|
poolAttr = (DBCPConnectionPoolAttr) jdbcConnection.getDbcpAttr().clone(); |
||||||
|
} catch (CloneNotSupportedException ignore) { |
||||||
|
} |
||||||
|
poolAttr.setValidationQuery(TransmissionTool.encrypt(poolAttr.getValidationQuery())); |
||||||
|
KerberosAuthentication kerberosAuthentication = null; |
||||||
|
if (jdbcConnection.getAuthentication().type() == AuthenticationType.KERBEROS) { |
||||||
|
//获取数据连接信息的时候，做一下兼容处理（兼容旧的kerberos形式）
|
||||||
|
KerberosUtils.compatibilityProcess(((KerberosAuthentication) jdbcConnection.getAuthentication())); |
||||||
|
kerberosAuthentication = ((KerberosAuthentication) jdbcConnection.getAuthentication()); |
||||||
|
} |
||||||
|
//因为设计器创建连接时database属性为空，所以这里直接取数据库保存的fetchSize值，默认值为-1
|
||||||
|
return jdbcConnectionBean |
||||||
|
.newCharsetName(jdbcConnection.getNewCharsetName()) |
||||||
|
.originalCharsetName(jdbcConnection.getOriginalCharsetName()) |
||||||
|
.database(jdbcConnection.getDatabase()) |
||||||
|
.user(jdbcConnection.getUser()) |
||||||
|
.driver(jdbcConnection.getDriver()) |
||||||
|
// 关键中的关键，由设计器发出的请求如果要携带密码，不能传明文，必须必须必须传加密后的密码
|
||||||
|
.password(withPassword ? TransmissionEncryptors.getInstance().encrypt(jdbcConnection.getPassword()) : DecisionServiceConstants.DEFAULT_PASSWORD) |
||||||
|
.schema(jdbcConnection.getSchema()) |
||||||
|
.url(jdbcConnection.getURL()) |
||||||
|
.creator(jdbcConnection.getCreator()) |
||||||
|
.source(jdbcConnection.getDriverSource()) |
||||||
|
.connectionPoolAttr(poolAttr.create()) |
||||||
|
.authType(kerberosAuthentication != null ? "kerberos" : StringUtils.EMPTY) |
||||||
|
.principal(kerberosAuthentication != null ? kerberosAuthentication.getPrincipal() : StringUtils.EMPTY) |
||||||
|
.keyPath(kerberosAuthentication != null ? kerberosAuthentication.getKeyPath() : StringUtils.EMPTY) |
||||||
|
.krb5Path(kerberosAuthentication != null ? KerberosUtils.getKrb5Path(kerberosAuthentication.getKeyPath(), kerberosAuthentication.getPrincipal()) : StringUtils.EMPTY) |
||||||
|
.useJaas(jdbcConnection.getAuthentication().type() == AuthenticationType.KERBEROS && ((KerberosAuthentication) jdbcConnection.getAuthentication()).getUseJaas()) |
||||||
|
.fetchSize(jdbcConnection.getFetchSize()).identity(connection.getIdentity()) |
||||||
|
.sshType(jdbcConnection.getSsh().getSshType().toString()) |
||||||
|
.sshIp(((BaseSsh) jdbcConnection.getSsh()).getIp()) |
||||||
|
.usingSsh(jdbcConnection.getSsh().isUsingSsh()) |
||||||
|
.sshUser(((BaseSsh) jdbcConnection.getSsh()).getUser()) |
||||||
|
.sshPort(((BaseSsh) jdbcConnection.getSsh()).getPort()) |
||||||
|
.redirectPort(jdbcConnection.getSsh().getRedirectPort()) |
||||||
|
.redirectIp((jdbcConnection.getSsh()).getRedirectIp()) |
||||||
|
.sshTimeOut(((BaseSsh) jdbcConnection.getSsh()).getTimeOut()) |
||||||
|
.sshSecret(withPassword ? ((BaseSsh) jdbcConnection.getSsh()).getSecret() : DecisionServiceConstants.DEFAULT_PASSWORD) |
||||||
|
.sshPrivateKeyPath(jdbcConnection.getSsh().getSshType() == SshType.KEY ? ((KeyVerifySsh) jdbcConnection.getSsh()).getPrivateKeyPath() : StringUtils.EMPTY) |
||||||
|
.usingSsl((jdbcConnection.getSsl()).isUsingSsl()) |
||||||
|
.sslType(jdbcConnection.getSsl().getSslType().toString()) |
||||||
|
.sslClientCertificate(((BaseSsl) jdbcConnection.getSsl()).getClientCertificate()) |
||||||
|
.sslClientPrivateKey(((BaseSsl) jdbcConnection.getSsl()).getClientPrivateKey()) |
||||||
|
.caCertificate(((BaseSsl) jdbcConnection.getSsl()).getCaCertificate()) |
||||||
|
.properties(jdbcConnection.getProperties()) |
||||||
|
.verifyCa(jdbcConnection.getSsl().getSslType() == SslType.NORMAL && ((NormalSsl) jdbcConnection.getSsl()).isVerifyCa()); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue