From 875d6268a7525b0c558ed2b834f9581a46c9bb4c Mon Sep 17 00:00:00 2001 From: Andrey Kuzin Date: Wed, 20 Mar 2024 18:55:52 +0000 Subject: [PATCH] SPILL_QUEUE_CAPACITY config property can now be declared as all lowercase as well --- .../athena/connector/lambda/data/S3BlockSpiller.java | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/athena-federation-sdk/src/main/java/com/amazonaws/athena/connector/lambda/data/S3BlockSpiller.java b/athena-federation-sdk/src/main/java/com/amazonaws/athena/connector/lambda/data/S3BlockSpiller.java index 502453ed0b..2604b5e228 100644 --- a/athena-federation-sdk/src/main/java/com/amazonaws/athena/connector/lambda/data/S3BlockSpiller.java +++ b/athena-federation-sdk/src/main/java/com/amazonaws/athena/connector/lambda/data/S3BlockSpiller.java @@ -35,6 +35,7 @@ import com.fasterxml.jackson.databind.ObjectMapper; import com.google.common.io.ByteStreams; import org.apache.arrow.vector.types.pojo.Schema; +import org.apache.commons.lang3.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -499,8 +500,10 @@ private void safeClose(AutoCloseable block) private ThreadPoolExecutor makeAsyncSpillPool(SpillConfig config) { int spillQueueCapacity = config.getNumSpillThreads(); - if (configOptions.get(SPILL_QUEUE_CAPACITY) != null) { - spillQueueCapacity = Integer.parseInt(configOptions.get(SPILL_QUEUE_CAPACITY)); + + String capacity = StringUtils.isNotBlank(configOptions.get(SPILL_QUEUE_CAPACITY)) ? configOptions.get(SPILL_QUEUE_CAPACITY) : configOptions.get(SPILL_QUEUE_CAPACITY.toLowerCase()); + if (capacity != null) { + spillQueueCapacity = Integer.parseInt(capacity); logger.debug("Setting Spill Queue Capacity to {}", spillQueueCapacity); }