Temparary revert old behaviour

Author: alex268

query/src/main/java/tech/ydb/query/tools/QueryReader.java

@@ -176,12 +176,14 @@ public int getRowCount() {
 
         @Override
         public void setRowIndex(int index) {
-            if (index < 0 || index >= rowsCount) {
-                throw new IndexOutOfBoundsException(String.format("Index %s out of bounds for length %s",
-                        index, rowsCount));
-            }
+            // TODO: Enable after JDBC fixing
+//            if (index < 0 || index >= rowsCount) {
+//                throw new IndexOutOfBoundsException(String.format("Index %s out of bounds for length %s",
+//                        index, rowsCount));
+//            }
+//            int currentIdx = index;
+            int currentIdx = Math.max(0, index);
             partIndex = 0;
-            int currentIdx = index;
             while (partIndex < parts.length) {
                 int readerRows = parts[partIndex].getRowCount();
                 if (currentIdx < readerRows) {
@@ -193,6 +195,11 @@ public void setRowIndex(int index) {
                 partIndex++;
             }
 
+            // TODO: remove after JDBC fixing
+            if (partIndex >= parts.length) {
+                partIndex = parts.length - 1;
+            }
+
             for (int partStep = partIndex + 1; partStep < parts.length; partStep++) {
                 parts[partStep].setRowIndex(0);
             }

query/src/test/java/tech/ydb/query/tools/QueryReaderTest.java

@@ -142,11 +142,16 @@ public void compositeResultSetTest() {
         rsr.setRowIndex(5);
         Assert.assertEquals(1, readAll(rsr, 5));
 
-        IndexOutOfBoundsException ex1 = Assert.assertThrows(IndexOutOfBoundsException.class, () -> rsr.setRowIndex(6));
-        Assert.assertEquals("Index 6 out of bounds for length 6", ex1.getMessage());
+        rsr.setRowIndex(-1);
+        Assert.assertEquals(6, readAll(rsr, 0));
 
-        IndexOutOfBoundsException ex2 = Assert.assertThrows(IndexOutOfBoundsException.class, () -> rsr.setRowIndex(-1));
-        Assert.assertEquals("Index -1 out of bounds for length 6", ex2.getMessage());
+        rsr.setRowIndex(6);
+        Assert.assertEquals(0, readAll(rsr, 0));
+//        IndexOutOfBoundsException ex1 = Assert.assertThrows(IndexOutOfBoundsException.class, () -> rsr.setRowIndex(6));
+//        Assert.assertEquals("Index 6 out of bounds for length 6", ex1.getMessage());
+//
+//        IndexOutOfBoundsException ex2 = Assert.assertThrows(IndexOutOfBoundsException.class, () -> rsr.setRowIndex(-1));
+//        Assert.assertEquals("Index -1 out of bounds for length 6", ex2.getMessage());
     }
 
     private int readAll(ResultSetReader rsr, int startKey) {