代码优化

jerrytaotz · Apr 11, 2020 · ed4ec48 · ed4ec48
1 parent ce4bdf7
commit ed4ec48
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.../main/java/org/spiderflow/core/executor/function/extension/ResponseFunctionExtension.java b/.../main/java/org/spiderflow/core/executor/function/extension/ResponseFunctionExtension.java
@@ -27,7 +27,7 @@ public Class<?> support() {
     @Comment("将请求结果转为Element对象")
     @Example("${resp.element()}")
     public static Element element(SpiderResponse response) {
-        return Jsoup.parse(response.getHtml());
+        return Jsoup.parse(response.getHtml(),response.getUrl());
     }
 
     @Comment("根据xpath在请求结果中查找")

diff --git a/...ore/src/main/java/org/spiderflow/core/executor/function/extension/SqlRowSetExtension.java b/...ore/src/main/java/org/spiderflow/core/executor/function/extension/SqlRowSetExtension.java
@@ -21,8 +21,7 @@ public Class<?> support() {
     @Example("${rs.nextToMap()}")
     public static Map<String, Object> nextToMap(SqlRowSet sqlRowSet) {
         try {
-            boolean next = sqlRowSet.next();
-            if (!next) {
+            if (!sqlRowSet.next()) {
                 return null;
             }
             String[] columnNames = sqlRowSet.getMetaData().getColumnNames();