From cc8b9c0b6e0e631d0c1cdfd512006bdd36039431 Mon Sep 17 00:00:00 2001
From: somu-imply <93540295+somu-imply@users.noreply.github.com>
Date: Mon, 24 Jan 2022 08:37:13 -0800
Subject: [PATCH] Handling OOM error in ExpressionVector setup by reducing
 number of rows (#12186)

* Handling OOM error in ExpressionVector setup by reducing number of rows

* Removing row size to 10K in sanity tests
---
 .../druid/segment/virtual/ExpressionVectorSelectorsTest.java    | 2 +-
 .../druid/sql/calcite/SqlVectorizedExpressionSanityTest.java    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/processing/src/test/java/org/apache/druid/segment/virtual/ExpressionVectorSelectorsTest.java b/processing/src/test/java/org/apache/druid/segment/virtual/ExpressionVectorSelectorsTest.java
index e940e68fc12..e0295eb6fb2 100644
--- a/processing/src/test/java/org/apache/druid/segment/virtual/ExpressionVectorSelectorsTest.java
+++ b/processing/src/test/java/org/apache/druid/segment/virtual/ExpressionVectorSelectorsTest.java
@@ -95,7 +95,7 @@ public class ExpressionVectorSelectorsTest
       "concat(string1, nonexistent)"
   );
 
-  private static final int ROWS_PER_SEGMENT = 100_000;
+  private static final int ROWS_PER_SEGMENT = 10_000;
 
   private static QueryableIndex INDEX;
   private static Closer CLOSER;
diff --git a/sql/src/test/java/org/apache/druid/sql/calcite/SqlVectorizedExpressionSanityTest.java b/sql/src/test/java/org/apache/druid/sql/calcite/SqlVectorizedExpressionSanityTest.java
index 2339cdd29bb..4dcfc182187 100644
--- a/sql/src/test/java/org/apache/druid/sql/calcite/SqlVectorizedExpressionSanityTest.java
+++ b/sql/src/test/java/org/apache/druid/sql/calcite/SqlVectorizedExpressionSanityTest.java
@@ -93,7 +93,7 @@ public class SqlVectorizedExpressionSanityTest extends InitializedNullHandlingTe
       "SELECT long5 IS NULL, long3 IS NOT NULL, count(*) FROM foo GROUP BY 1,2 ORDER BY 3"
   );
 
-  private static final int ROWS_PER_SEGMENT = 100_000;
+  private static final int ROWS_PER_SEGMENT = 10_000;
 
   private static QueryableIndex INDEX;
   private static Closer CLOSER;