Make test time limit more precise (TEST_MAX_TIME) (#385)

* done * revert
thulab · Dec 19, 2023 · 7c49c17 · 7c49c17
1 parent ba26ed8
commit 7c49c17
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/configuration/conf/config.properties b/configuration/conf/config.properties
@@ -50,6 +50,8 @@
 # BENCHMARK_WORK_MODE=testWithDefaultPath
 
 # 对于数据写入或查询，限制最长耗时，设置为0表示无限制，单位为ms
+# 对数据写入，最大误差约为WRITE_OPERATION_TIMEOUT_MS
+# 对数据查询，最大误差约为READ_OPERATION_TIMEOUT_MS
 # 此参数不限制预先注册元数据的耗时
 # TEST_MAX_TIME=0
 

diff --git a/core/src/main/java/cn/edu/tsinghua/iot/benchmark/client/generate/GenerateDataMixClient.java b/core/src/main/java/cn/edu/tsinghua/iot/benchmark/client/generate/GenerateDataMixClient.java
@@ -140,6 +140,9 @@ private boolean ingestionOperation() {
           }
         }
         for (int j = 0; j < innerLoop; j++) {
+          if (isStop.get()) {
+            return true;
+          }
           IBatch batch = dataWorkLoad.getOneBatch();
           if (checkBatch(batch)) {
             dbWrapper.insertOneBatchWithCheck(batch);