support run sql script file in jdbc protocol config (#1117)

apache · Jul 27, 2023 · a9c3ef3 · a9c3ef3
1 parent 6e508eb
commit a9c3ef3
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/collector/pom.xml b/collector/pom.xml
@@ -56,6 +56,12 @@
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-validation</artifactId>
         </dependency>
+        <!-- run sql script-->
+        <dependency>
+            <groupId>org.springframework</groupId>
+            <artifactId>spring-jdbc</artifactId>
+            <version>5.3.23</version>
+        </dependency>
         <!-- kafka -->
         <dependency>
             <groupId>org.apache.kafka</groupId>

diff --git a/...tor/src/main/java/org/dromara/hertzbeat/collector/collect/database/JdbcCommonCollect.java b/...tor/src/main/java/org/dromara/hertzbeat/collector/collect/database/JdbcCommonCollect.java
@@ -32,6 +32,8 @@
 import org.dromara.hertzbeat.common.util.CommonUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.postgresql.util.PSQLException;
+import org.springframework.core.io.FileSystemResource;
+import org.springframework.jdbc.datasource.init.ScriptUtils;
 
 import java.sql.Connection;
 import java.sql.DriverManager;
@@ -54,6 +56,8 @@ public class JdbcCommonCollect extends AbstractCollect {
     private static final String QUERY_TYPE_ONE_ROW = "oneRow";
     private static final String QUERY_TYPE_MULTI_ROW = "multiRow";
     private static final String QUERY_TYPE_COLUMNS = "columns";
+
+    private static final String RUN_SCRIPT = "runScript";
 
     public JdbcCommonCollect(){}
 
@@ -84,6 +88,11 @@ public void collect(CollectRep.MetricsData.Builder builder, long appId, String a
                 case QUERY_TYPE_COLUMNS:
                     queryOneRowByMatchTwoColumns(statement, jdbcProtocol.getSql(), metrics.getAliasFields(), builder, startTime);
                     break;
+                case RUN_SCRIPT:
+                    Connection connection = statement.getConnection();
+                    FileSystemResource rc = new FileSystemResource(jdbcProtocol.getSql());
+                    ScriptUtils.executeSqlScript(connection, rc);
+                    break;
                 default:
                     builder.setCode(CollectRep.Code.FAIL);
                     builder.setMsg("Not support database query type: " + jdbcProtocol.getQueryType());

diff --git a/common/src/main/java/org/dromara/hertzbeat/common/entity/job/protocol/JdbcProtocol.java b/common/src/main/java/org/dromara/hertzbeat/common/entity/job/protocol/JdbcProtocol.java
@@ -61,7 +61,7 @@ public class JdbcProtocol {
      */
     private String platform;
     /**
-     * SQL查询方式： oneRow, multiRow, columns
+     * SQL查询方式： oneRow, multiRow, columns, runScript
      */
     private String queryType;
     /**