diff --git a/collector/src/main/java/com/usthe/collector/collect/database/JdbcCommonCollect.java b/collector/src/main/java/com/usthe/collector/collect/database/JdbcCommonCollect.java index 300a838..116d5c8 100644 --- a/collector/src/main/java/com/usthe/collector/collect/database/JdbcCommonCollect.java +++ b/collector/src/main/java/com/usthe/collector/collect/database/JdbcCommonCollect.java @@ -52,9 +52,11 @@ public class JdbcCommonCollect extends AbstractCollect { } JdbcProtocol jdbcProtocol = metrics.getJdbc(); String databaseUrl = constructDatabaseUrl(jdbcProtocol); + //获取查询语句/获取链接最大超时时间 + int timeout = metrics.getJdbc().getTimeout().intValue(); try { Statement statement = getConnection(jdbcProtocol.getUsername(), - jdbcProtocol.getPassword(), databaseUrl); + jdbcProtocol.getPassword(), databaseUrl,timeout); switch (jdbcProtocol.getQueryType()) { case QUERY_TYPE_ONE_ROW: queryOneRow(statement, jdbcProtocol.getSql(), metrics.getAliasFields(), builder, startTime); @@ -95,7 +97,7 @@ public class JdbcCommonCollect extends AbstractCollect { } - private Statement getConnection(String username, String password, String url) throws Exception { + private Statement getConnection(String username, String password, String url,Integer timeout) throws Exception { CacheIdentifier identifier = CacheIdentifier.builder() .ip(url) .username(username).password(password).build(); @@ -106,7 +108,7 @@ public class JdbcCommonCollect extends AbstractCollect { try { statement = jdbcConnect.getConnection().createStatement(); // 设置查询超时时间10秒 - statement.setQueryTimeout(10); + statement.setQueryTimeout(timeout); // 设置查询最大行数1000行 statement.setMaxRows(1000); } catch (Exception e) { @@ -130,7 +132,7 @@ public class JdbcCommonCollect extends AbstractCollect { Connection connection = DriverManager.getConnection(url, username, password); statement = connection.createStatement(); // 设置查询超时时间10秒 - statement.setQueryTimeout(10); + statement.setQueryTimeout(timeout); // 设置查询最大行数1000行 statement.setMaxRows(1000); JdbcConnect jdbcConnect = new JdbcConnect(connection); diff --git a/common/src/main/java/com/usthe/common/entity/job/protocol/JdbcProtocol.java b/common/src/main/java/com/usthe/common/entity/job/protocol/JdbcProtocol.java index 19a4685..642326a 100644 --- a/common/src/main/java/com/usthe/common/entity/job/protocol/JdbcProtocol.java +++ b/common/src/main/java/com/usthe/common/entity/job/protocol/JdbcProtocol.java @@ -35,6 +35,10 @@ public class JdbcProtocol { * 数据库 */ private String database; + /** + * 超时时间 + */ + private Long timeout; /** * 数据库类型 mysql oracle ... */ diff --git a/manager/src/main/resources/define/app/mysql.yml b/manager/src/main/resources/define/app/mysql.yml index bbc3784..951f0b5 100644 --- a/manager/src/main/resources/define/app/mysql.yml +++ b/manager/src/main/resources/define/app/mysql.yml @@ -16,6 +16,10 @@ configmap: type: 2 - key: database type: 1 + - key: timeout + type: 0 + required: false + defaultValue: 3 - key: url type: 1 # 指标组列表 @@ -61,6 +65,7 @@ metrics: username: ^_^username^_^ password: ^_^password^_^ database: ^_^database^_^ + timeout: ^_^timeout^_^ # SQL查询方式: oneRow, multiRow, columns queryType: columns # sql diff --git a/manager/src/main/resources/define/param/mysql.yml b/manager/src/main/resources/define/param/mysql.yml index bc1dd3a..a7e4fa1 100644 --- a/manager/src/main/resources/define/param/mysql.yml +++ b/manager/src/main/resources/define/param/mysql.yml @@ -27,4 +27,10 @@ param: - field: url name: URL type: text - required: false \ No newline at end of file + required: false + - field: timeout + name: 超时时间 + type: number + required: false + defaultValue: 3 + placeholder: '请输入mysql可接受的超时时间' \ No newline at end of file