@@ -107,10 +107,10 @@ public void testVectorSearchEndToEnd() throws Exception {
107107 assertThat (result .results ().isEmpty ()).isFalse ();
108108
109109 // Read using the search result
110- ReadBuilder readBuilder = table .newReadBuilder (). withGlobalIndexResult ( result ) ;
110+ ReadBuilder readBuilder = table .newReadBuilder ();
111111 List <Integer > ids = new ArrayList <>();
112- try ( RecordReader < InternalRow > reader =
113- readBuilder .newRead ().createReader (readBuilder . newScan (). plan () )) {
112+ TableScan . Plan plan = readBuilder . newScan (). withGlobalIndexResult ( result ). plan ();
113+ try ( RecordReader < InternalRow > reader = readBuilder .newRead ().createReader (plan )) {
114114 reader .forEachRemaining (row -> ids .add (row .getInt (0 )));
115115 }
116116
@@ -157,10 +157,10 @@ public void testVectorSearchWithCosineMetric() throws Exception {
157157
158158 assertThat (result ).isInstanceOf (ScoredGlobalIndexResult .class );
159159
160- ReadBuilder readBuilder = table .newReadBuilder ().withGlobalIndexResult (result );
160+ ReadBuilder readBuilder = table .newReadBuilder ();
161+ TableScan .Plan plan = readBuilder .newScan ().withGlobalIndexResult (result ).plan ();
161162 List <Integer > ids = new ArrayList <>();
162- try (RecordReader <InternalRow > reader =
163- readBuilder .newRead ().createReader (readBuilder .newScan ().plan ())) {
163+ try (RecordReader <InternalRow > reader = readBuilder .newRead ().createReader (plan )) {
164164 reader .forEachRemaining (row -> ids .add (row .getInt (0 )));
165165 }
166166
@@ -209,10 +209,10 @@ public void testVectorSearchTopKLimit() throws Exception {
209209 .withVectorColumn (VECTOR_FIELD_NAME )
210210 .executeLocal ();
211211
212- ReadBuilder readBuilder = table .newReadBuilder ().withGlobalIndexResult (result );
212+ ReadBuilder readBuilder = table .newReadBuilder ();
213+ TableScan .Plan plan = readBuilder .newScan ().withGlobalIndexResult (result ).plan ();
213214 List <Integer > ids = new ArrayList <>();
214- try (RecordReader <InternalRow > reader =
215- readBuilder .newRead ().createReader (readBuilder .newScan ().plan ())) {
215+ try (RecordReader <InternalRow > reader = readBuilder .newRead ().createReader (plan )) {
216216 reader .forEachRemaining (row -> ids .add (row .getInt (0 )));
217217 }
218218
@@ -250,10 +250,10 @@ public void testVectorSearchWithMultipleIndexFiles() throws Exception {
250250 assertThat (result ).isInstanceOf (ScoredGlobalIndexResult .class );
251251 assertThat (result .results ().isEmpty ()).isFalse ();
252252
253- ReadBuilder readBuilder = table .newReadBuilder ().withGlobalIndexResult (result );
253+ ReadBuilder readBuilder = table .newReadBuilder ();
254+ TableScan .Plan plan = readBuilder .newScan ().withGlobalIndexResult (result ).plan ();
254255 List <Integer > ids = new ArrayList <>();
255- try (RecordReader <InternalRow > reader =
256- readBuilder .newRead ().createReader (readBuilder .newScan ().plan ())) {
256+ try (RecordReader <InternalRow > reader = readBuilder .newRead ().createReader (plan )) {
257257 reader .forEachRemaining (row -> ids .add (row .getInt (0 )));
258258 }
259259
0 commit comments