Merge pull request #103 from jumpserver/pr@dev@feat_affectrows

feat: fix affected rows

Author: Aaron3S

backend/framework/src/main/java/org/jumpserver/chen/framework/datasource/base/BaseSQLActuator.java

@@ -56,29 +56,26 @@ public int getAffectedRows(SQL sql) throws SQLException {
 
         var sqlStmts = SQLUtils.parseStatements(sql.getSql(), this.druidDbType);
 
-        if (sqlStmts.size() != 1) {
-            return -1;
-        }
-
-        var sqlStmt = sqlStmts.get(0);
-
-        if (sqlStmt instanceof SQLUpdateStatement || sqlStmt instanceof SQLDeleteStatement || sqlStmt instanceof SQLInsertStatement) {
-            var conn = this.getConnection();
-            try {
-                conn.setAutoCommit(false);
-                var stmt = conn.createStatement();
-                stmt.execute(sqlStmt.toString());
-
-                result = stmt.getUpdateCount();
-
-                conn.rollback();
-                stmt.close();
-            } finally {
-                if (this.connection == null) {
-                    conn.close();
-                } else {
-                    conn.setAutoCommit(true);
+        for (var sqlStmt : sqlStmts) {
+            if (sqlStmt instanceof SQLUpdateStatement || sqlStmt instanceof SQLDeleteStatement || sqlStmt instanceof SQLInsertStatement) {
+                var conn = this.getConnection();
+                try {
+                    conn.setAutoCommit(false);
+                    var stmt = conn.createStatement();
+                    stmt.execute(sqlStmt.toString());
+
+                    result += stmt.getUpdateCount();
+
+                    conn.rollback();
+                    stmt.close();
+                } finally {
+                    if (this.connection == null) {
+                        conn.close();
+                    } else {
+                        conn.setAutoCommit(true);
+                    }
                 }
+
             }
         }
         return result;