From cab309fccba3f46882d8351e22cd920b49e23ec8 Mon Sep 17 00:00:00 2001
From: Fabian Sobanski <fasobanski@gmail.com>
Date: Fri, 27 Jun 2025 12:01:24 +0200
Subject: [PATCH] Fix NPE if trace logging is enabled

---
 .../AmazonKeyspacesRoundRobinLoadBalancingPolicy.java         | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/aws/ssa/keyspaces/loadbalancing/AmazonKeyspacesRoundRobinLoadBalancingPolicy.java b/src/main/java/com/aws/ssa/keyspaces/loadbalancing/AmazonKeyspacesRoundRobinLoadBalancingPolicy.java
index fbee7e3..31341c2 100644
--- a/src/main/java/com/aws/ssa/keyspaces/loadbalancing/AmazonKeyspacesRoundRobinLoadBalancingPolicy.java
+++ b/src/main/java/com/aws/ssa/keyspaces/loadbalancing/AmazonKeyspacesRoundRobinLoadBalancingPolicy.java
@@ -76,7 +76,7 @@ public Queue<Node> newQueryPlan(@Nullable Request request, @Nullable Session ses
         int totalNodes = currentNodes.length;
 
         if (LOG.isTraceEnabled()) {
-            if (totalNodes > 0) {
+            if (totalNodes > 0 && session != null) {
                 //int currentSize = getSize((Node) currentNodes[0], session);
 
                 int inflight = getInFlight((Node) currentNodes[0], session);
@@ -88,6 +88,8 @@ public Queue<Node> newQueryPlan(@Nullable Request request, @Nullable Session ses
                 int requestPerMostUsedConnection = (openConnections > 0) ? (inflight / openConnections) : 0;
 
                 LOG.trace(" Total local nodes: [{}], First Node [{}], Number of Connections: [{}], Total inflight:[{}], Number of Request per connection: [{}]", totalNodes, firstNode, openConnections, inflight, requestPerMostUsedConnection);
+            } else if (session == null) {
+                LOG.trace(" Total local nodes: [{}], Session is null", totalNodes);
             }
         }
         return queryPlan;