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 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 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;