betetr math

hms-dbmi · Jul 20, 2023 · 84146ed · 84146ed
1 parent fd15f0f
commit 84146ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/processing/src/main/java/edu/harvard/hms/dbmi/avillach/hpds/processing/QueryProcessor.java b/processing/src/main/java/edu/harvard/hms/dbmi/avillach/hpds/processing/QueryProcessor.java
@@ -66,7 +66,8 @@ public void runQuery(Query query, AsyncResult result) {
 			.map(ColumnMeta::getName)
 			.collect(Collectors.toList());
 
-		int size = Integer.BYTES + columns.size() * 8 * ID_BATCH_SIZE;
+		int rowWidth = Integer.BYTES + columns.stream().mapToInt(ColumnMeta::getWidthInBytes).sum() * ID_BATCH_SIZE;
+		int size = rowWidth * ID_BATCH_SIZE;
 		GenericObjectPool<byte[]> pool = new GenericObjectPool<>(new ByteArrayFactory(size));
 
 		Lists.partition(new ArrayList<>(idList), ID_BATCH_SIZE).stream().parallel().forEach((list) -> {