From 84146ed1f54d70ae5322c9c3a2c2ebb4f8d4fdc1 Mon Sep 17 00:00:00 2001
From: Luke Sikina <lucas.sikina@gmail.com>
Date: Thu, 20 Jul 2023 14:44:34 -0400
Subject: [PATCH] betetr math

---
 .../hms/dbmi/avillach/hpds/processing/QueryProcessor.java      | 3 ++-
 1 file changed, 2 insertions(+), 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
index 556d0fef..5fe7f81b 100644
--- 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) -> {