提交 ff12a33d authored 作者: taojinlong's avatar taojinlong

refactor: 格式化代码

上级 8401f4e1
...@@ -31,12 +31,14 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -31,12 +31,14 @@ public class JdbcProvider extends DefaultJdbcProvider {
@Resource @Resource
private DeDriverMapper deDriverMapper; private DeDriverMapper deDriverMapper;
@Override @Override
public boolean isUseDatasourcePool(){ public boolean isUseDatasourcePool() {
return true; return true;
} }
@Override @Override
public String getType(){ public String getType() {
return "built-in"; return "built-in";
} }
/** /**
...@@ -69,14 +71,14 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -69,14 +71,14 @@ public class JdbcProvider extends DefaultJdbcProvider {
@Override @Override
public List<TableField> getTableFileds(DatasourceRequest datasourceRequest) throws Exception { public List<TableField> getTableFileds(DatasourceRequest datasourceRequest) throws Exception {
if(datasourceRequest.getDatasource().getType().equalsIgnoreCase("mongo")){ if (datasourceRequest.getDatasource().getType().equalsIgnoreCase("mongo")) {
datasourceRequest.setQuery("select * from " + datasourceRequest.getTable()); datasourceRequest.setQuery("select * from " + datasourceRequest.getTable());
return fetchResultField(datasourceRequest); return fetchResultField(datasourceRequest);
} }
List<TableField> list = new LinkedList<>(); List<TableField> list = new LinkedList<>();
try (Connection connection = getConnectionFromPool(datasourceRequest)) { try (Connection connection = getConnectionFromPool(datasourceRequest)) {
if (datasourceRequest.getDatasource().getType().equalsIgnoreCase("oracle")) { if (datasourceRequest.getDatasource().getType().equalsIgnoreCase("oracle")) {
Method setRemarksReporting = extendedJdbcClassLoader.loadClass("oracle.jdbc.driver.OracleConnection").getMethod("setRemarksReporting",boolean.class); Method setRemarksReporting = extendedJdbcClassLoader.loadClass("oracle.jdbc.driver.OracleConnection").getMethod("setRemarksReporting", boolean.class);
setRemarksReporting.invoke(((DruidPooledConnection) connection).getConnection(), true); setRemarksReporting.invoke(((DruidPooledConnection) connection).getConnection(), true);
} }
DatabaseMetaData databaseMetaData = connection.getMetaData(); DatabaseMetaData databaseMetaData = connection.getMetaData();
...@@ -105,10 +107,10 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -105,10 +107,10 @@ public class JdbcProvider extends DefaultJdbcProvider {
} catch (SQLException e) { } catch (SQLException e) {
DataEaseException.throwException(e); DataEaseException.throwException(e);
} catch (Exception e) { } catch (Exception e) {
if(datasourceRequest.getDatasource().getType().equalsIgnoreCase("ds_doris")){ if (datasourceRequest.getDatasource().getType().equalsIgnoreCase("ds_doris")) {
datasourceRequest.setQuery("select * from " + datasourceRequest.getTable()); datasourceRequest.setQuery("select * from " + datasourceRequest.getTable());
return fetchResultField(datasourceRequest); return fetchResultField(datasourceRequest);
}else { } else {
DataEaseException.throwException(Translator.get("i18n_datasource_connect_error") + e.getMessage()); DataEaseException.throwException(Translator.get("i18n_datasource_connect_error") + e.getMessage());
} }
...@@ -210,9 +212,9 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -210,9 +212,9 @@ public class JdbcProvider extends DefaultJdbcProvider {
private List<String[]> getDataResult(ResultSet rs, DatasourceRequest datasourceRequest) throws Exception { private List<String[]> getDataResult(ResultSet rs, DatasourceRequest datasourceRequest) throws Exception {
String charset = null; String charset = null;
if(datasourceRequest != null && datasourceRequest.getDatasource().getType().equalsIgnoreCase("oracle")){ if (datasourceRequest != null && datasourceRequest.getDatasource().getType().equalsIgnoreCase("oracle")) {
JdbcConfiguration JdbcConfiguration = new Gson().fromJson(datasourceRequest.getDatasource().getConfiguration(), JdbcConfiguration.class); JdbcConfiguration JdbcConfiguration = new Gson().fromJson(datasourceRequest.getDatasource().getConfiguration(), JdbcConfiguration.class);
if(StringUtils.isNotEmpty(JdbcConfiguration.getCharset()) && !JdbcConfiguration.getCharset().equalsIgnoreCase("Default") ){ if (StringUtils.isNotEmpty(JdbcConfiguration.getCharset()) && !JdbcConfiguration.getCharset().equalsIgnoreCase("Default")) {
charset = JdbcConfiguration.getCharset(); charset = JdbcConfiguration.getCharset();
} }
} }
...@@ -233,9 +235,9 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -233,9 +235,9 @@ public class JdbcProvider extends DefaultJdbcProvider {
row[j] = rs.getBoolean(j + 1) ? "1" : "0"; row[j] = rs.getBoolean(j + 1) ? "1" : "0";
break; break;
default: default:
if(charset != null && StringUtils.isNotEmpty(rs.getString(j + 1))){ if (charset != null && StringUtils.isNotEmpty(rs.getString(j + 1))) {
row[j] = new String(rs.getString(j + 1).getBytes(charset), "UTF-8"); row[j] = new String(rs.getString(j + 1).getBytes(charset), "UTF-8");
}else { } else {
row[j] = rs.getString(j + 1); row[j] = rs.getString(j + 1);
} }
break; break;
...@@ -278,6 +280,7 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -278,6 +280,7 @@ public class JdbcProvider extends DefaultJdbcProvider {
} }
return fieldList; return fieldList;
} }
@Override @Override
public List<String[]> getData(DatasourceRequest dsr) throws Exception { public List<String[]> getData(DatasourceRequest dsr) throws Exception {
List<String[]> list = new LinkedList<>(); List<String[]> list = new LinkedList<>();
...@@ -376,26 +379,26 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -376,26 +379,26 @@ public class JdbcProvider extends DefaultJdbcProvider {
customDriver = hiveConfiguration.getCustomDriver(); customDriver = hiveConfiguration.getCustomDriver();
jdbcurl = hiveConfiguration.getJdbc(); jdbcurl = hiveConfiguration.getJdbc();
if(StringUtils.isNotEmpty(hiveConfiguration.getAuthMethod()) && hiveConfiguration.getAuthMethod().equalsIgnoreCase("kerberos")){ if (StringUtils.isNotEmpty(hiveConfiguration.getAuthMethod()) && hiveConfiguration.getAuthMethod().equalsIgnoreCase("kerberos")) {
System.setProperty("java.security.krb5.conf", "/opt/dataease/conf/krb5.conf"); System.setProperty("java.security.krb5.conf", "/opt/dataease/conf/krb5.conf");
ExtendedJdbcClassLoader classLoader; ExtendedJdbcClassLoader classLoader;
if(isDefaultClassLoader(customDriver)){ if (isDefaultClassLoader(customDriver)) {
classLoader = extendedJdbcClassLoader; classLoader = extendedJdbcClassLoader;
}else { } else {
deDriver = deDriverMapper.selectByPrimaryKey(customDriver); deDriver = deDriverMapper.selectByPrimaryKey(customDriver);
classLoader = getCustomJdbcClassLoader(deDriver); classLoader = getCustomJdbcClassLoader(deDriver);
} }
Class<?> ConfigurationClass = classLoader.loadClass("org.apache.hadoop.conf.Configuration"); Class<?> ConfigurationClass = classLoader.loadClass("org.apache.hadoop.conf.Configuration");
Method set = ConfigurationClass.getMethod("set",String.class, String.class) ; Method set = ConfigurationClass.getMethod("set", String.class, String.class);
Object obj = ConfigurationClass.newInstance(); Object obj = ConfigurationClass.newInstance();
set.invoke(obj, "hadoop.security.authentication", "Kerberos"); set.invoke(obj, "hadoop.security.authentication", "Kerberos");
Class<?> UserGroupInformationClass = classLoader.loadClass("org.apache.hadoop.security.UserGroupInformation"); Class<?> UserGroupInformationClass = classLoader.loadClass("org.apache.hadoop.security.UserGroupInformation");
Method setConfiguration = UserGroupInformationClass.getMethod("setConfiguration",ConfigurationClass) ; Method setConfiguration = UserGroupInformationClass.getMethod("setConfiguration", ConfigurationClass);
Method loginUserFromKeytab = UserGroupInformationClass.getMethod("loginUserFromKeytab",String.class, String.class) ; Method loginUserFromKeytab = UserGroupInformationClass.getMethod("loginUserFromKeytab", String.class, String.class);
setConfiguration.invoke(null, obj); setConfiguration.invoke(null, obj);
loginUserFromKeytab.invoke(null, hiveConfiguration.getUsername(), "/opt/dataease/conf/" + hiveConfiguration.getPassword()); loginUserFromKeytab.invoke(null, hiveConfiguration.getUsername(), "/opt/dataease/conf/" + hiveConfiguration.getPassword());
}else { } else {
username = hiveConfiguration.getUsername(); username = hiveConfiguration.getUsername();
password = hiveConfiguration.getPassword(); password = hiveConfiguration.getPassword();
} }
...@@ -428,12 +431,12 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -428,12 +431,12 @@ public class JdbcProvider extends DefaultJdbcProvider {
} }
Connection conn; Connection conn;
String driverClassName ; String driverClassName;
ExtendedJdbcClassLoader jdbcClassLoader; ExtendedJdbcClassLoader jdbcClassLoader;
if(isDefaultClassLoader(customDriver)){ if (isDefaultClassLoader(customDriver)) {
driverClassName = defaultDriver; driverClassName = defaultDriver;
jdbcClassLoader = extendedJdbcClassLoader; jdbcClassLoader = extendedJdbcClassLoader;
}else { } else {
driverClassName = deDriver.getDriverClass(); driverClassName = deDriver.getDriverClass();
jdbcClassLoader = getCustomJdbcClassLoader(deDriver); jdbcClassLoader = getCustomJdbcClassLoader(deDriver);
} }
...@@ -442,11 +445,11 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -442,11 +445,11 @@ public class JdbcProvider extends DefaultJdbcProvider {
ClassLoader classLoader = Thread.currentThread().getContextClassLoader(); ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
try { try {
Thread.currentThread().setContextClassLoader(jdbcClassLoader); Thread.currentThread().setContextClassLoader(jdbcClassLoader);
conn= driverClass.connect(jdbcurl, props); conn = driverClass.connect(jdbcurl, props);
}catch (Exception e){ } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
throw e; throw e;
}finally { } finally {
Thread.currentThread().setContextClassLoader(classLoader); Thread.currentThread().setContextClassLoader(classLoader);
} }
return conn; return conn;
...@@ -454,7 +457,7 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -454,7 +457,7 @@ public class JdbcProvider extends DefaultJdbcProvider {
@Override @Override
public JdbcConfiguration setCredential(DatasourceRequest datasourceRequest, DruidDataSource dataSource) throws Exception{ public JdbcConfiguration setCredential(DatasourceRequest datasourceRequest, DruidDataSource dataSource) throws Exception {
DatasourceTypes datasourceType = DatasourceTypes.valueOf(datasourceRequest.getDatasource().getType()); DatasourceTypes datasourceType = DatasourceTypes.valueOf(datasourceRequest.getDatasource().getType());
JdbcConfiguration jdbcConfiguration = new JdbcConfiguration(); JdbcConfiguration jdbcConfiguration = new JdbcConfiguration();
switch (datasourceType) { switch (datasourceType) {
...@@ -537,9 +540,9 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -537,9 +540,9 @@ public class JdbcProvider extends DefaultJdbcProvider {
dataSource.setUsername(jdbcConfiguration.getUsername()); dataSource.setUsername(jdbcConfiguration.getUsername());
ExtendedJdbcClassLoader classLoader; ExtendedJdbcClassLoader classLoader;
if(isDefaultClassLoader(jdbcConfiguration.getCustomDriver())){ if (isDefaultClassLoader(jdbcConfiguration.getCustomDriver())) {
classLoader = extendedJdbcClassLoader; classLoader = extendedJdbcClassLoader;
}else { } else {
DeDriver deDriver = deDriverMapper.selectByPrimaryKey(jdbcConfiguration.getCustomDriver()); DeDriver deDriver = deDriverMapper.selectByPrimaryKey(jdbcConfiguration.getCustomDriver());
classLoader = getCustomJdbcClassLoader(deDriver); classLoader = getCustomJdbcClassLoader(deDriver);
} }
...@@ -667,7 +670,7 @@ public class JdbcProvider extends DefaultJdbcProvider { ...@@ -667,7 +670,7 @@ public class JdbcProvider extends DefaultJdbcProvider {
case sqlServer: case sqlServer:
return "select name from sys.schemas;"; return "select name from sys.schemas;";
case db2: case db2:
return "select SCHEMANAME from syscat.SCHEMATA WHERE \"DEFINER\" ='USER'".replace("USER", db2Configuration.getUsername().toUpperCase()) ; return "select SCHEMANAME from syscat.SCHEMATA WHERE \"DEFINER\" ='USER'".replace("USER", db2Configuration.getUsername().toUpperCase());
case pg: case pg:
return "SELECT nspname FROM pg_namespace;"; return "SELECT nspname FROM pg_namespace;";
case redshift: case redshift:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论