try (#382)

thulab · Nov 28, 2023 · 8dd7a11 · 8dd7a11
1 parent f28ea50
commit 8dd7a11
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/iotdb-1.1/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb110/IoTDBSessionBase.java b/iotdb-1.1/src/main/java/cn/edu/tsinghua/iot/benchmark/iotdb110/IoTDBSessionBase.java
@@ -52,6 +52,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
@@ -66,6 +67,9 @@ public class IoTDBSessionBase extends IoTDB {
   static final Config config = ConfigDescriptor.getInstance().getConfig();
   IBenchmarkSession sessionWrapper;
 
+  private static final Map<String, Binary> binaryCache =
+      new ConcurrentHashMap<>(config.getWORKLOAD_BUFFER_SIZE());
+
   public IoTDBSessionBase(DBConfig dbConfig) {
     super(dbConfig);
   }
@@ -376,6 +380,7 @@ public DeviceSummary deviceSummary(DeviceQuery deviceQuery) throws TsdbException
   }
 
   protected Tablet genTablet(IBatch batch) {
+    config.getWORKLOAD_BUFFER_SIZE();
     List<MeasurementSchema> schemaList = new ArrayList<>();
     int sensorIndex = 0;
     for (Sensor sensor : batch.getDeviceSchema().getSensors()) {
@@ -428,10 +433,11 @@ protected Tablet genTablet(IBatch batch) {
           case TEXT:
             Binary[] sensorsText = (Binary[]) values[recordValueIndex];
             sensorsText[recordIndex] =
-                Binary.valueOf((String) record.getRecordDataValue().get(recordValueIndex));
+                binaryCache.computeIfAbsent(
+                    (String) record.getRecordDataValue().get(recordValueIndex), Binary::valueOf);
             break;
           default:
-            LOGGER.error("Unsupported Type:" + sensors.get(sensorIndex).getSensorType());
+            LOGGER.error("Unsupported Type: {}", sensors.get(sensorIndex).getSensorType());
         }
         sensorIndex++;
       }