Make fetchSize changable for outputBuffer

pesse · pesse · commit 6806f2be1c25 · 2018-03-27T21:14:52.000+02:00
diff --git a/src/main/java/org/utplsql/api/outputBuffer/AbstractOutputBuffer.java b/src/main/java/org/utplsql/api/outputBuffer/AbstractOutputBuffer.java
@@ -20,6 +20,7 @@
 abstract class AbstractOutputBuffer implements OutputBuffer {
 
     private Reporter reporter;
+    private int fetchSize = 100;
 
     /**
      * Creates a new DefaultOutputBuffer.
@@ -40,6 +41,11 @@ public Reporter getReporter() {
         return reporter;
     }
 
+    @Override
+    public void setFetchSize(int fetchSize) {
+        this.fetchSize = fetchSize;
+    }
+
     /**
      * Print the lines as soon as they are produced and write to a PrintStream.
      * @param conn DB connection
@@ -77,7 +83,7 @@ public void fetchAvailable(Connection conn, Consumer<String> onLineFetched) thro
 
         try (CallableStatement cstmt = getLinesCursorStatement(conn)) {
             cstmt.execute();
-            cstmt.setFetchSize(1);
+            cstmt.setFetchSize(fetchSize);
 
             try ( ResultSet resultSet = (ResultSet) cstmt.getObject(1)) {
                 while (resultSet.next())
@@ -97,6 +103,7 @@ public List<String> fetchAll(Connection conn) throws SQLException {
         try (CallableStatement cstmt = getLinesCursorStatement(conn)) {
 
             cstmt.execute();
+            cstmt.setFetchSize(fetchSize);
 
             try ( ResultSet resultSet = (ResultSet) cstmt.getObject(1)) {
 
diff --git a/src/main/java/org/utplsql/api/outputBuffer/NonOutputBuffer.java b/src/main/java/org/utplsql/api/outputBuffer/NonOutputBuffer.java
@@ -26,6 +26,11 @@ public Reporter getReporter() {
         return reporter;
     }
 
+    @Override
+    public void setFetchSize(int fetchSize) {
+
+    }
+
     @Override
     public void printAvailable(Connection conn, PrintStream ps) throws SQLException {
         List<PrintStream> printStreams = new ArrayList<>(1);
diff --git a/src/main/java/org/utplsql/api/outputBuffer/OutputBuffer.java b/src/main/java/org/utplsql/api/outputBuffer/OutputBuffer.java
@@ -12,6 +12,12 @@ public interface OutputBuffer {
 
     Reporter getReporter();
 
+    /** Override the fetchSize of the OutputBuffer
+     *
+     * @param fetchSize the ResultSet fetch-size.
+     */
+    void setFetchSize( int fetchSize );
+
     /**
      * Print the lines as soon as they are produced and write to a PrintStream.
      * @param conn DB connection
diff --git a/src/test/java/org/utplsql/api/OutputBufferIT.java b/src/test/java/org/utplsql/api/OutputBufferIT.java
@@ -59,6 +59,7 @@ public void printAvailableLines() throws SQLException {
                     printStreams.add(System.out);
                     printStreams.add(new PrintStream(fileOutStream));
 
+                    reporter.getOutputBuffer().setFetchSize(1);
                     reporter.getOutputBuffer().printAvailable(newConnection(), printStreams);
 
                     return Boolean.TRUE;
