Override skip method

ChengJie1053 · Jan 22, 2024 · 5c4c20f · 5c4c20f
1 parent 8d5c5ac
commit 5c4c20f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/.../linkis-storage/src/main/java/org/apache/linkis/storage/resultset/OrcResultSetReader.java b/.../linkis-storage/src/main/java/org/apache/linkis/storage/resultset/OrcResultSetReader.java
@@ -101,7 +101,16 @@ public MetaData getMetaData() {
 
   @Override
   public int skip(int recordNum) throws IOException {
-    throw new UnsupportedOperationException("Storeage Unsupported type: skip");
+    if (recordNum < 0) return -1;
+
+    for (int i = recordNum; i > 0; i--) {
+      try {
+        hasNext();
+      } catch (Throwable t) {
+        return recordNum - i;
+      }
+    }
+    return recordNum;
   }
 
   @Override

diff --git a/...kis-storage/src/main/java/org/apache/linkis/storage/resultset/ParquetResultSetReader.java b/...kis-storage/src/main/java/org/apache/linkis/storage/resultset/ParquetResultSetReader.java
@@ -98,7 +98,16 @@ public MetaData getMetaData() {
 
   @Override
   public int skip(int recordNum) throws IOException {
-    throw new UnsupportedOperationException("Storeage Unsupported type: skip");
+    if (recordNum < 0) return -1;
+
+    for (int i = recordNum; i > 0; i--) {
+      try {
+        this.record = parquetReader.read();
+      } catch (Throwable t) {
+        return recordNum - i;
+      }
+    }
+    return recordNum;
   }
 
   @Override