Skip to content

D
dataease
提交 ff12a33d authored 作者: taojinlong's avatar taojinlong
浏览文件
操作

refactor: 格式化代码

上级 8401f4e1
隐藏空白字符变更
内嵌 并排
正在显示 包含 33 行增加 和 30 行删除
backend/src/main/java/io/dataease/provider/datasource/JdbcProvider.java
浏览文件 @ ff12a33d
......@@ -31,13 +31,15 @@ public class JdbcProvider extends DefaultJdbcProvider {
@Resource
private DeDriverMapper deDriverMapper;
@Override
public boolean isUseDatasourcePool(){
public boolean isUseDatasourcePool() {
return true;
}
@Override
public String getType(){
return "built-in";
public String getType() {
return "built-in";
}
/**
* 增加缓存机制 key 由 'provider_sql_' dsr.datasource.id dsr.table dsr.query共4部分组成，命中则使用缓存直接返回不再执行sql逻辑
......@@ -69,14 +71,14 @@ public class JdbcProvider extends DefaultJdbcProvider {
@Override
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());
return fetchResultField(datasourceRequest);
}
List<TableField> list = new LinkedList<>();
try (Connection connection = getConnectionFromPool(datasourceRequest)) {
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);
}
DatabaseMetaData databaseMetaData = connection.getMetaData();
......@@ -105,10 +107,10 @@ public class JdbcProvider extends DefaultJdbcProvider {
} catch (SQLException e) {
DataEaseException.throwException(e);
} catch (Exception e) {
if(datasourceRequest.getDatasource().getType().equalsIgnoreCase("ds_doris")){
if (datasourceRequest.getDatasource().getType().equalsIgnoreCase("ds_doris")) {
datasourceRequest.setQuery("select * from " + datasourceRequest.getTable());
return fetchResultField(datasourceRequest);
}else {
} else {
DataEaseException.throwException(Translator.get("i18n_datasource_connect_error") + e.getMessage());
}
......@@ -210,9 +212,9 @@ public class JdbcProvider extends DefaultJdbcProvider {
private List<String[]> getDataResult(ResultSet rs, DatasourceRequest datasourceRequest) throws Exception {
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);
if(StringUtils.isNotEmpty(JdbcConfiguration.getCharset()) && !JdbcConfiguration.getCharset().equalsIgnoreCase("Default") ){
if (StringUtils.isNotEmpty(JdbcConfiguration.getCharset()) && !JdbcConfiguration.getCharset().equalsIgnoreCase("Default")) {
charset = JdbcConfiguration.getCharset();
}
}
......@@ -233,9 +235,9 @@ public class JdbcProvider extends DefaultJdbcProvider {
row[j] = rs.getBoolean(j + 1) ? "1" : "0";
break;
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");
}else {
} else {
row[j] = rs.getString(j + 1);
}
break;
......@@ -278,6 +280,7 @@ public class JdbcProvider extends DefaultJdbcProvider {
}
return fieldList;
}
@Override
public List<String[]> getData(DatasourceRequest dsr) throws Exception {
List<String[]> list = new LinkedList<>();
......@@ -376,26 +379,26 @@ public class JdbcProvider extends DefaultJdbcProvider {
customDriver = hiveConfiguration.getCustomDriver();
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");
ExtendedJdbcClassLoader classLoader;
if(isDefaultClassLoader(customDriver)){
if (isDefaultClassLoader(customDriver)) {
classLoader = extendedJdbcClassLoader;
}else {
} else {
deDriver = deDriverMapper.selectByPrimaryKey(customDriver);
classLoader = getCustomJdbcClassLoader(deDriver);
}
Class<?> ConfigurationClass = classLoader.loadClass("org.apache.hadoop.conf.Configuration");
Method set = ConfigurationClass.getMethod("set",String.class, String.class) ;
Class<?> ConfigurationClass = classLoader.loadClass("org.apache.hadoop.conf.Configuration");
Method set = ConfigurationClass.getMethod("set", String.class, String.class);
Object obj = ConfigurationClass.newInstance();
set.invoke(obj, "hadoop.security.authentication", "Kerberos");
Class<?> UserGroupInformationClass = classLoader.loadClass("org.apache.hadoop.security.UserGroupInformation");
Method setConfiguration = UserGroupInformationClass.getMethod("setConfiguration",ConfigurationClass) ;
Method loginUserFromKeytab = UserGroupInformationClass.getMethod("loginUserFromKeytab",String.class, String.class) ;
Class<?> UserGroupInformationClass = classLoader.loadClass("org.apache.hadoop.security.UserGroupInformation");
Method setConfiguration = UserGroupInformationClass.getMethod("setConfiguration", ConfigurationClass);
Method loginUserFromKeytab = UserGroupInformationClass.getMethod("loginUserFromKeytab", String.class, String.class);
setConfiguration.invoke(null, obj);
loginUserFromKeytab.invoke(null, hiveConfiguration.getUsername(), "/opt/dataease/conf/" + hiveConfiguration.getPassword());
}else {
} else {
username = hiveConfiguration.getUsername();
password = hiveConfiguration.getPassword();
}
......@@ -428,12 +431,12 @@ public class JdbcProvider extends DefaultJdbcProvider {
}
Connection conn;
String driverClassName ;
String driverClassName;
ExtendedJdbcClassLoader jdbcClassLoader;
if(isDefaultClassLoader(customDriver)){
if (isDefaultClassLoader(customDriver)) {
driverClassName = defaultDriver;
jdbcClassLoader = extendedJdbcClassLoader;
}else {
} else {
driverClassName = deDriver.getDriverClass();
jdbcClassLoader = getCustomJdbcClassLoader(deDriver);
}
......@@ -442,11 +445,11 @@ public class JdbcProvider extends DefaultJdbcProvider {
ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
try {
Thread.currentThread().setContextClassLoader(jdbcClassLoader);
conn= driverClass.connect(jdbcurl, props);
}catch (Exception e){
conn = driverClass.connect(jdbcurl, props);
} catch (Exception e) {
e.printStackTrace();
throw e;
}finally {
} finally {
Thread.currentThread().setContextClassLoader(classLoader);
}
return conn;
......@@ -454,7 +457,7 @@ public class JdbcProvider extends DefaultJdbcProvider {
@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());
JdbcConfiguration jdbcConfiguration = new JdbcConfiguration();
switch (datasourceType) {
......@@ -537,9 +540,9 @@ public class JdbcProvider extends DefaultJdbcProvider {
dataSource.setUsername(jdbcConfiguration.getUsername());
ExtendedJdbcClassLoader classLoader;
if(isDefaultClassLoader(jdbcConfiguration.getCustomDriver())){
if (isDefaultClassLoader(jdbcConfiguration.getCustomDriver())) {
classLoader = extendedJdbcClassLoader;
}else {
} else {
DeDriver deDriver = deDriverMapper.selectByPrimaryKey(jdbcConfiguration.getCustomDriver());
classLoader = getCustomJdbcClassLoader(deDriver);
}
......@@ -667,7 +670,7 @@ public class JdbcProvider extends DefaultJdbcProvider {
case sqlServer:
return "select name from sys.schemas;";
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:
return "SELECT nspname FROM pg_namespace;";
case redshift:
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论