collect-tablet

Author: Caideyipi

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/pipe/event/TsFileInsertionEventParserTest.java

@@ -617,18 +617,15 @@ private void testTsFilePointNum(
                                       })
                                   .forEach(
                                       tabletInsertionEvent2 ->
-                                          tabletInsertionEvent2.processTablet(
-                                              (tablet, rowCollector) ->
-                                                  new PipeRawTabletInsertionEvent(tablet, false)
-                                                      .processRowByRow(
-                                                          (row, collector) -> {
+                                          tabletInsertionEvent2.processTabletWithCollect(
+                                              (tablet, collector) -> {
                                                             try {
-                                                              rowCollector.collectRow(row);
-                                                              count3.addAndGet(getNonNullSize(row));
+                                                              collector.collectTablet(tablet);
+                                                              count3.addAndGet(getNonNullSize(tablet));
                                                             } catch (final IOException e) {
                                                               throw new RuntimeException(e);
                                                             }
-                                                          })))));
+                                                          }))));
 
       Assert.assertEquals(expectedCount, count1.get());
       Assert.assertEquals(expectedCount, count2.get());
@@ -648,4 +645,16 @@ private int getNonNullSize(final Row row) {
     }
     return count;
   }
+
+  private int getNonNullSize(final Tablet tablet) {
+    int count = 0;
+    for (int i=0; i<tablet.getRowSize(); ++i) {
+      for (int j=0; j<tablet.getSchemas().size(); ++j) {
+        if (!tablet.isNull(i, j)) {
+          ++count;
+        }
+      }
+    }
+    return count;
+  }
 }