Fix okPointNum statistics for table models. (#467)

* Fix okPointNum statistics for table models. * fix okpointNum for table.
thulab · Nov 28, 2024 · 456fa81 · 456fa81
1 parent e1a45c1
commit 456fa81
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/...2.0/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb200/DMLStrategy/SessionStrategy.java b/...2.0/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb200/DMLStrategy/SessionStrategy.java
@@ -31,6 +31,7 @@
 import cn.edu.tsinghua.iot.benchmark.entity.DeviceSummary;
 import cn.edu.tsinghua.iot.benchmark.entity.Record;
 import cn.edu.tsinghua.iot.benchmark.entity.Sensor;
+import cn.edu.tsinghua.iot.benchmark.entity.enums.SQLDialect;
 import cn.edu.tsinghua.iot.benchmark.entity.enums.SensorType;
 import cn.edu.tsinghua.iot.benchmark.exception.OperationFailException;
 import cn.edu.tsinghua.iot.benchmark.iotdb200.IoTDB;
@@ -364,9 +365,14 @@ public long executeQueryAndGetStatusImpl(
                 isOk.set(false);
               }
               long resultPointNum = line.get();
-              if (!Operation.LATEST_POINT_QUERY.equals(operation)) {
+
+              if (config.getIoTDB_DIALECT_MODE().equals(SQLDialect.TABLE)) {
                 resultPointNum *= config.getQUERY_SENSOR_NUM();
-                resultPointNum *= config.getQUERY_DEVICE_NUM();
+              } else {
+                if (!Operation.LATEST_POINT_QUERY.equals(operation)) {
+                  resultPointNum *= config.getQUERY_DEVICE_NUM();
+                  resultPointNum *= config.getQUERY_SENSOR_NUM();
+                }
               }
               queryResultPointNum.set(resultPointNum);
             });