Skip to content

Commit 4cbcd6b

Browse files
committed
update jdbc
1 parent 8e55a58 commit 4cbcd6b

File tree

3 files changed

+15
-12
lines changed

3 files changed

+15
-12
lines changed

linkis-engineconn-plugins/jdbc/src/main/scala/org/apache/linkis/manager/engineplugin/jdbc/executor/JDBCEngineConnExecutor.scala

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -310,26 +310,25 @@ class JDBCEngineConnExecutor(override val outputPrintLimit: Int, val id: Int)
310310
JDBCEngineConnConstant.JDBC_ENGINE_RUN_TIME_DS_PARAM_USERNAME,
311311
""
312312
)
313+
val proxyUser: String =
314+
globalConfig.getOrDefault(JDBCEngineConnConstant.JDBC_PROXY_USER, execSqlUser)
313315
logger.info(
314-
s"use conn param get dataSourceInfo: executeUser:${execSqlUser} ip:${connHost}, port:${connPort}, dsType:${connDsType}, connUser: ${userName}"
316+
s"use conn param get dataSourceInfo: executeUser:${execSqlUser} ip:${connHost}, port:${connPort}, dsType:${connDsType}, createUser:${execSqlUser} connUser: ${proxyUser}"
315317
)
316318
if (
317319
StringUtils.isBlank(connHost) || StringUtils
318-
.isBlank(connPort) || StringUtils.isBlank(connDsType) || StringUtils.isBlank(userName)
320+
.isBlank(connPort) || StringUtils.isBlank(connDsType) || StringUtils.isBlank(
321+
execSqlUser
322+
) || StringUtils.isBlank(proxyUser)
319323
) {
320324
throw new JDBCGetDatasourceInfoException(
321325
JDBC_GET_DATASOURCEINFO_ERROR.getErrorCode,
322326
JDBC_GET_DATASOURCEINFO_ERROR.getErrorDesc + " 缺失部分连接参数"
323327
)
324328
}
325-
if (!execSqlUser.equals(userName)) {
326-
throw new JDBCGetDatasourceInfoException(
327-
JDBC_GET_DATASOURCEINFO_ERROR.getErrorCode,
328-
JDBC_GET_DATASOURCEINFO_ERROR.getErrorDesc + " 执行用户和连接用户不匹配"
329-
)
330-
}
331329
dataSourceInfo = JDBCMultiDatasourceParser.queryDatasourceInfoByConnParams(
332-
userName,
330+
execSqlUser,
331+
proxyUser,
333332
connHost,
334333
connPort,
335334
connDsType

linkis-engineconn-plugins/jdbc/src/main/scala/org/apache/linkis/manager/engineplugin/jdbc/executor/JDBCMultiDatasourceParser.scala

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -293,15 +293,17 @@ object JDBCMultiDatasourceParser extends Logging {
293293
}
294294

295295
def queryDatasourceInfoByConnParams(
296-
userName: String,
296+
createUser: String,
297+
proxyUser: String,
297298
ip: String,
298299
port: String,
299300
datasourceTypeName: String
300301
): util.Map[String, String] = {
301302
val dataSourceClient = new LinkisDataSourceRemoteClient()
302303
val action: GetInfoPublishedByUserIpPortAction = GetInfoPublishedByUserIpPortAction.builder
303304
.setDatasourceTypeName(datasourceTypeName)
304-
.setUser(userName)
305+
.setUser(createUser)
306+
.setDatasourceUser(proxyUser)
305307
.setIp(ip)
306308
.setPort(port)
307309
.build // ignore parameter 'system'

linkis-public-enhancements/linkis-pes-client/src/main/scala/org/apache/linkis/datasource/client/request/GetInfoPublishedByUserIpPortAction.scala

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ object GetInfoPublishedByUserIpPortAction {
6262
this.datasourceUser = datasourceUser
6363
this
6464
}
65+
6566
def setOwner(owner: String): Builder = {
6667
this.owner = owner
6768
this
@@ -103,7 +104,8 @@ object GetInfoPublishedByUserIpPortAction {
103104
// if (system == null) throw new DataSourceClientBuilderException(SYSTEM_NEEDED.getErrorDesc)
104105
if (user == null) throw new DataSourceClientBuilderException(USER_NEEDED.getErrorDesc)
105106

106-
if (datasourceUser == null) throw new DataSourceClientBuilderException(USER_NEEDED.getErrorDesc)
107+
if (datasourceUser == null)
108+
throw new DataSourceClientBuilderException(USER_NEEDED.getErrorDesc)
107109

108110
val GetInfoPublishedByUserIpPortAction = new GetInfoPublishedByUserIpPortAction
109111
GetInfoPublishedByUserIpPortAction.datasourceTypeName = this.datasourceTypeName

0 commit comments

Comments
 (0)