Set an appropriate List HashMap capacity to reduce the number of capa…

…city expansions. (#445)
thulab · Sep 18, 2024 · 9fe6209 · 9fe6209
1 parent bddf3d0
commit 9fe6209
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/core/src/main/java/cn/edu/tsinghua/iot/benchmark/workload/GenerateDataWorkLoad.java b/core/src/main/java/cn/edu/tsinghua/iot/benchmark/workload/GenerateDataWorkLoad.java
@@ -67,7 +67,7 @@ public long getBatchNumber() {
   /** Add one row into batch, row contains data from all sensors */
   protected List<Object> generateOneRow(int deviceIndex, int colIndex, long stepOffset)
       throws WorkloadException {
-    List<Object> values = new ArrayList<>();
+    List<Object> values = new ArrayList<>(config.getSENSOR_NUMBER());
     int index = (int) (Math.abs(stepOffset * (deviceIndex + 1)) % config.getWORKLOAD_BUFFER_SIZE());
     if (colIndex == -1) {
       for (int i = 0; i < config.getSENSOR_NUMBER(); i++) {

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
@@ -73,7 +73,7 @@ public class SessionStrategy extends DMLStrategy {
   static final Config config = ConfigDescriptor.getInstance().getConfig();
 
   private static final Map<String, Binary> binaryCache =
-      new ConcurrentHashMap<>(config.getWORKLOAD_BUFFER_SIZE());
+      new ConcurrentHashMap<>(config.getWORKLOAD_BUFFER_SIZE(), 1.00f);
   private final Session session;
   private final IoTDB iotdb;