From 3ea36425b8c9204198cffc89b271c916c445ea7d Mon Sep 17 00:00:00 2001
From: Brian Bleakley <brianbleakley@gmail.com>
Date: Fri, 20 Jun 2025 16:08:43 -0700
Subject: [PATCH] add getcolumn to parquetdataframe

---
 src/lib/tableProvider.ts | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/lib/tableProvider.ts b/src/lib/tableProvider.ts
index 122b1fe6..596c8a7a 100644
--- a/src/lib/tableProvider.ts
+++ b/src/lib/tableProvider.ts
@@ -188,6 +188,19 @@ export function parquetDataFrame(from: AsyncBufferFrom, metadata: FileMetaData):
       }
     },
     sortable: true,
-    // TODO(SL): implement getColumn({column, start, end}): any[]
+    getColumn({ column, start, end }) {
+      if (!header.includes(column)) {
+        return Promise.reject(new Error(`Column "${column}" not found in header`))
+      }
+
+      return parquetQueryWorker({
+        from,
+        metadata,
+        rowStart: start,
+        rowEnd: end,
+      }).then(rows => {
+        return rows.map(row => row[column])
+      })
+    },
   }
 }