From ddd34b4b161334f81b62fc930e941156e5701fca Mon Sep 17 00:00:00 2001
From: chenshizhi <chenshizhi@bilibili.com>
Date: Wed, 15 Jun 2022 20:24:46 +0800
Subject: [PATCH] [HUDI-4277] supoort flink table source with computed column

---
 .../src/main/java/org/apache/hudi/table/HoodieTableSource.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/table/HoodieTableSource.java b/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/table/HoodieTableSource.java
index 1836857383554..e76bb29bdf50a 100644
--- a/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/table/HoodieTableSource.java
+++ b/hudi-flink-datasource/hudi-flink/src/main/java/org/apache/hudi/table/HoodieTableSource.java
@@ -156,7 +156,7 @@ public HoodieTableSource(
     this.fileIndex = FileIndex.instance(this.path, this.conf);
     this.requiredPartitions = requiredPartitions;
     this.requiredPos = requiredPos == null
-        ? IntStream.range(0, schema.getColumnCount()).toArray()
+        ? IntStream.range(0, schema.toPhysicalRowDataType().getChildren().size()).toArray()
         : requiredPos;
     this.limit = limit == null ? NO_LIMIT_CONSTANT : limit;
     this.filters = filters == null ? Collections.emptyList() : filters;