Collect all table names.

thulab · Nov 28, 2024 · fe7b5f6 · fe7b5f6
1 parent 99bcaf2
commit fe7b5f6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/core/src/main/java/cn/edu/tsinghua/iot/benchmark/schema/MetaUtil.java b/core/src/main/java/cn/edu/tsinghua/iot/benchmark/schema/MetaUtil.java
@@ -110,7 +110,8 @@ public static void distributeDevices(
       Map<Integer, List<DeviceSchema>> clientDataSchema,
       List<Sensor> sensors,
       Map<String, DeviceSchema> nameDataSchema,
-      Set<String> groups) {
+      Set<String> groups,
+      boolean collectTables) {
     Map<Integer, Integer> deviceDistributionForClient =
         CommonAlgorithms.distributeDevicesToClients(config.getDEVICE_NUMBER(), clientNumber);
     int deviceIndex = MetaUtil.getDeviceId(0);
@@ -125,6 +126,9 @@ public static void distributeDevices(
         deviceSchemasList.add(deviceSchema);
         nameDataSchema.putIfAbsent(deviceSchema.getDevice(), deviceSchema);
         groups.add(deviceSchema.getGroup());
+        if(collectTables){
+          MetaDataSchema.TABLES.add(deviceSchema.getTable());
+        }
         deviceIndex++;
       }
       clientDataSchema.put(clientId, deviceSchemasList);

diff --git a/...src/main/java/cn/edu/tsinghua/iot/benchmark/schema/schemaImpl/GenerateMetaDataSchema.java b/...src/main/java/cn/edu/tsinghua/iot/benchmark/schema/schemaImpl/GenerateMetaDataSchema.java
@@ -38,16 +38,17 @@ public boolean createMetaDataSchema() {
     if (sensors == null) {
       return false;
     }
-    // schemaClient
+    // schemaClient, Collect all table names and register them at one time.
     MetaUtil.distributeDevices(
         config.getSCHEMA_CLIENT_NUMBER(),
         SCHEMA_CLIENT_DATA_SCHEMA,
         sensors,
         NAME_DATA_SCHEMA,
-        GROUPS);
+        GROUPS,
+        true);
     // dataClient
     MetaUtil.distributeDevices(
-        config.getDATA_CLIENT_NUMBER(), DATA_CLIENT_DATA_SCHEMA, sensors, NAME_DATA_SCHEMA, GROUPS);
+        config.getDATA_CLIENT_NUMBER(), DATA_CLIENT_DATA_SCHEMA, sensors, NAME_DATA_SCHEMA, GROUPS, false);
     return true;
   }
 }