Merge 98482a0 into 9c1ac1d

tsfile/src/test/java/org/apache/iotdb/tsfile/read/reader/FakedMultiBatchReader.java

@@ -0,0 +1,62 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.iotdb.tsfile.read.reader;
+
+import java.io.IOException;
+import org.apache.iotdb.tsfile.file.metadata.enums.TSDataType;
+import org.apache.iotdb.tsfile.read.common.BatchData;
+import org.apache.iotdb.tsfile.read.filter.basic.Filter;
+
+public class FakedMultiBatchReader implements IBatchReader {
+
+  private long time = 0;
+  private int batchSize;
+  private int batches;
+  private int batchIndex = 0;
+  private Filter filter;
+
+  FakedMultiBatchReader(int batchSize, int batches, Filter filter) {
+    this.batchSize = batchSize;
+    this.batches = batches;
+    this.filter = filter;
+  }
+
+  @Override
+  public boolean hasNextBatch() {
+    return batchIndex < batches;
+  }
+
+  @Override
+  public BatchData nextBatch() {
+    batchIndex++;
+    BatchData batchData = new BatchData(TSDataType.INT64);
+    for (int i = 0; i < batchSize; i++) {
+      if (filter.satisfy(time, time)) {
+        batchData.putLong(time, time);
+      }
+      time++;
+    }
+    return batchData;
+  }
+
+  @Override
+  public void close() throws IOException {
+
+  }
+}

tsfile/src/test/java/org/apache/iotdb/tsfile/read/reader/FakedTimeGenerator.java

@@ -0,0 +1,73 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+package org.apache.iotdb.tsfile.read.reader;
+
+import java.io.IOException;
+import org.apache.iotdb.tsfile.read.common.Path;
+import org.apache.iotdb.tsfile.read.expression.IExpression;
+import org.apache.iotdb.tsfile.read.expression.impl.BinaryExpression;
+import org.apache.iotdb.tsfile.read.expression.impl.SingleSeriesExpression;
+import org.apache.iotdb.tsfile.read.filter.TimeFilter;
+import org.apache.iotdb.tsfile.read.filter.factory.FilterFactory;
+import org.apache.iotdb.tsfile.read.query.timegenerator.TimeGenerator;
+import org.apache.iotdb.tsfile.read.query.timegenerator.node.AndNode;
+import org.apache.iotdb.tsfile.read.query.timegenerator.node.LeafNode;
+import org.junit.Assert;
+import org.junit.Test;
+
+public class FakedTimeGenerator extends TimeGenerator {
+
+  public FakedTimeGenerator() throws IOException {
+
+    // or(and(d1.s1, d2.s2), d2.s2)
+    IExpression expression =
+        BinaryExpression.or(
+            BinaryExpression.and(
+                new SingleSeriesExpression(new Path("d1.s1"),
+                    FilterFactory.and(TimeFilter.gtEq(1L), TimeFilter.ltEq(5L))),
+                new SingleSeriesExpression(new Path("d2.s2"),
+                    FilterFactory.and(TimeFilter.gtEq(1L), TimeFilter.ltEq(10L)))
+            ),
+            new SingleSeriesExpression(new Path("d2.s2"),
+                FilterFactory.and(TimeFilter.gtEq(11L), TimeFilter.ltEq(15L)))
+        );
+
+    super.constructNode(expression);
+  }
+
+  @Override
+  protected IBatchReader generateNewBatchReader(SingleSeriesExpression expression) {
+    return new FakedMultiBatchReader(10, 10, expression.getFilter());
+  }
+
+
+  @Test
+  public void testTimeGenerator() throws IOException {
+    FakedTimeGenerator fakedTimeGenerator = new FakedTimeGenerator();
+    Path path = new Path("d1.s1");
+    long count = 0;
+    while (fakedTimeGenerator.hasNext()) {
+      long time = fakedTimeGenerator.next();
+      fakedTimeGenerator.getValue(path, time);
+      count++;
+    }
+    Assert.assertEquals(10L, count);
+  }
+
+}