From 728477c284c9386f7cff9fac9e10e48dd0ca4c94 Mon Sep 17 00:00:00 2001
From: Siyuan Zhang <siyuanzhang.zsy@alibaba-inc.com>
Date: Mon, 14 Oct 2024 14:06:11 +0800
Subject: [PATCH] add callback option to IO interface (#2001)

Enable user to process the results of `ReadTable` in a streaming way
---------
Signed-off-by: siyuan0322 <siyuanzhang.zsy@alibaba-inc.com>
---
 modules/io/io/i_io_adaptor.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/modules/io/io/i_io_adaptor.h b/modules/io/io/i_io_adaptor.h
index 5502991a4..fd05101e8 100644
--- a/modules/io/io/i_io_adaptor.h
+++ b/modules/io/io/i_io_adaptor.h
@@ -80,6 +80,11 @@ class IIOAdaptor {
     return Status::OK();
   }
 
+  virtual Status ReadTable(
+      std::function<void(std::shared_ptr<arrow::RecordBatch>)> callback) {
+    return Status::OK();
+  }
+
   virtual Status WriteTable(std::shared_ptr<arrow::Table> table) {
     return Status::OK();
   }