[doris-flink-connector] branch master updated: get connection response improvement (#124)

[di...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

diwu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris-flink-connector.git


The following commit(s) were added to refs/heads/master by this push:
     new 3be9dc9  get connection response improvement (#124)
3be9dc9 is described below

commit 3be9dc9ab62c4966e55d971390c7ef2d6218b56b
Author: benjobs <be...@apache.org>
AuthorDate: Mon Mar 27 10:14:22 2023 +0800

    get connection response improvement (#124)
    
    Co-authored-by: benjobs <be...@gmail.com>
---
 .../java/org/apache/doris/flink/rest/RestService.java  | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

diff --git a/flink-doris-connector/src/main/java/org/apache/doris/flink/rest/RestService.java b/flink-doris-connector/src/main/java/org/apache/doris/flink/rest/RestService.java
index c774d68..7db5da3 100644
--- a/flink-doris-connector/src/main/java/org/apache/doris/flink/rest/RestService.java
+++ b/flink-doris-connector/src/main/java/org/apache/doris/flink/rest/RestService.java
@@ -45,10 +45,8 @@ import org.apache.http.client.methods.HttpRequestBase;
 import org.apache.http.entity.StringEntity;
 import org.slf4j.Logger;
 
-import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStream;
-import java.io.InputStreamReader;
 import java.io.PrintWriter;
 import java.io.Serializable;
 import java.net.HttpURLConnection;
@@ -62,6 +60,7 @@ import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Scanner;
 import java.util.Set;
 import java.util.stream.Collectors;
 
@@ -192,16 +191,13 @@ public class RestService implements Serializable {
                     connection.getURL(), connection.getResponseCode());
             throw new IOException("Failed to get response from Doris");
         }
-        String result = "";
-        BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream(), "utf-8"));
-        String line;
-        while ((line = in.readLine()) != null) {
-            result += line;
-        }
-        if (in != null) {
-            in.close();
+        StringBuffer result = new StringBuffer();
+        try (Scanner scanner = new Scanner(connection.getInputStream(), "utf-8")) {
+            while (scanner.hasNext()) {
+                result.append(scanner.next());
+            }
+            return result.toString();
         }
-        return result;
     }
 
     /**


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@doris.apache.org
For additional commands, e-mail: commits-help@doris.apache.org