add null check

snowflakedb · Mar 14, 2024 · 540e2df · 540e2df
1 parent 8f4514d
commit 540e2df
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/main/java/net/snowflake/client/core/arrow/AbstractArrowVectorConverter.java b/src/main/java/net/snowflake/client/core/arrow/AbstractArrowVectorConverter.java
@@ -148,8 +148,10 @@ public BigDecimal toBigDecimal(int index) throws SFException {
   }
 
   public boolean shouldTreatDecimalAsInt(SFBaseSession session) {
-    if (!session.isJdbcArrowTreatDecimalAsInt() && !session.isJdbcTreatDecimalAsInt()) {
-      return false;
+    if (session != null) {
+      if (!session.isJdbcArrowTreatDecimalAsInt() && !session.isJdbcTreatDecimalAsInt()) {
+        return false;
+      }
     }
     return true;
   }