Fix for wrong connection count on remote nodes

core/src/main/java/com/datastax/oss/driver/internal/core/cql/CqlPrepareHandler.java

@@ -188,6 +188,13 @@ private void sendRequest(PrepareRequest request, Node node, int retryCount) {
       while (!result.isDone() && (node = queryPlan.poll()) != null) {
         channel = session.getChannel(node, logPrefix);
         if (channel != null) {
+          if (LOG.isTraceEnabled()) {
+            LOG.trace(
+                "[{}] Polled node {} from queryPlan with hashCode = {}",
+                logPrefix,
+                node.getEndPoint(),
+                System.identityHashCode(queryPlan));
+          }
           break;
         } else {
           recordError(node, new NodeUnavailableException(node));
@@ -363,7 +370,13 @@ public void operationComplete(Future<java.lang.Void> future) {
           // Might happen if the timeout just fired
           cancel();
         } else {
-          LOG.trace("[{}] Request sent to {}", logPrefix, node);
+          if (LOG.isDebugEnabled()) {
+            LOG.debug(
+                "[{}] Request sent to {} for a queryPlan with hashCode = {}",
+                logPrefix,
+                node.getEndPoint(),
+                System.identityHashCode(queryPlan));
+          }
           initialCallback = this;
         }
       }

core/src/main/java/com/datastax/oss/driver/internal/core/cql/CqlRequestHandler.java

@@ -251,6 +251,13 @@ private void sendRequest(
       while (!result.isDone() && (node = queryPlan.poll()) != null) {
         channel = session.getChannel(node, logPrefix);
         if (channel != null) {
+          if (LOG.isTraceEnabled()) {
+            LOG.trace(
+                "[{}] Polled node {} from queryPlan with hashCode = {}",
+                logPrefix,
+                node.getEndPoint(),
+                System.identityHashCode(queryPlan));
+          }
           break;
         } else {
           recordError(node, new NodeUnavailableException(node));
@@ -535,7 +542,13 @@ public void operationComplete(Future<java.lang.Void> future) throws Exception {
               scheduleNextExecution); // try next node
         }
       } else {
-        LOG.trace("[{}] Request sent on {}", logPrefix, channel);
+        if (LOG.isDebugEnabled()) {
+          LOG.debug(
+              "[{}] Request sent on {} for a queryPlan with hashCode = {}",
+              logPrefix,
+              channel.getEndPoint(),
+              System.identityHashCode(queryPlan));
+        }
         if (result.isDone()) {
           // If the handler completed since the last time we checked, cancel directly because we
           // don't know if cancelScheduledTasks() has run yet

core/src/main/java/com/datastax/oss/driver/internal/core/loadbalancing/BasicLoadBalancingPolicy.java

@@ -111,9 +111,11 @@ public class BasicLoadBalancingPolicy implements LoadBalancingPolicy {
   private final ConsistencyLevel defaultConsistencyLevel;
 
   // private because they should be set in init() and never be modified after
-  private volatile DistanceReporter distanceReporter;
+  // Yugabyte specific: protected because distanceReporter is to be used in
+  // YugabyteDefaultLoadBalancingPolicy class
+  protected volatile DistanceReporter distanceReporter;
   protected volatile NodeDistanceEvaluator nodeDistanceEvaluator;
-  private volatile String localDc;
+  protected volatile String localDc;
   protected volatile NodeSet liveNodes;
 
   public BasicLoadBalancingPolicy(@NonNull DriverContext context, @NonNull String profileName) {

core/src/main/java/com/datastax/oss/driver/internal/core/session/DefaultSession.java

@@ -20,6 +20,7 @@
 import com.datastax.oss.driver.api.core.CqlSession;
 import com.datastax.oss.driver.api.core.ProtocolVersion;
 import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
+import com.datastax.oss.driver.api.core.config.DriverExecutionProfile;
 import com.datastax.oss.driver.api.core.context.DriverContext;
 import com.datastax.oss.driver.api.core.loadbalancing.LoadBalancingPolicy;
 import com.datastax.oss.driver.api.core.metadata.EndPoint;
@@ -140,6 +141,55 @@ private DefaultSession(InternalDriverContext context, Set<EndPoint> contactPoint
       }
       throw t;
     }
+    if (LOG.isInfoEnabled()) {
+      if (context.getConfig() != null) {
+        Map<String, ?> profiles = context.getConfig().getProfiles();
+        if (profiles != null) {
+          for (Map.Entry e : profiles.entrySet()) {
+            DriverExecutionProfile dep = (DriverExecutionProfile) e.getValue();
+            String cl =
+                dep.isDefined(DefaultDriverOption.REQUEST_CONSISTENCY)
+                    ? dep.getString(DefaultDriverOption.REQUEST_CONSISTENCY)
+                    : "UNDEFINED";
+            String localDC =
+                dep.isDefined(DefaultDriverOption.LOAD_BALANCING_LOCAL_DATACENTER)
+                    ? dep.getString(DefaultDriverOption.LOAD_BALANCING_LOCAL_DATACENTER)
+                    : "UNDEFINED";
+            String lbClass =
+                dep.isDefined(DefaultDriverOption.LOAD_BALANCING_POLICY_CLASS)
+                    ? dep.getString(DefaultDriverOption.LOAD_BALANCING_POLICY_CLASS)
+                    : "UNDEFINED";
+            String rpClass =
+                dep.isDefined(DefaultDriverOption.RETRY_POLICY_CLASS)
+                    ? dep.getString(DefaultDriverOption.RETRY_POLICY_CLASS)
+                    : "UNDEFINED";
+            String localDCViaAPI =
+                context.getLocalDatacenter((String) e.getKey()) != null
+                    ? context.getLocalDatacenter((String) e.getKey())
+                    : "UNDEFINED";
+            String remotePoolSize =
+                dep.isDefined(DefaultDriverOption.CONNECTION_POOL_REMOTE_SIZE)
+                    ? String.valueOf(dep.getInt(DefaultDriverOption.CONNECTION_POOL_REMOTE_SIZE))
+                    : "UNDEFINED";
+            String localPoolSize =
+                dep.isDefined(DefaultDriverOption.CONNECTION_POOL_LOCAL_SIZE)
+                    ? String.valueOf(dep.getInt(DefaultDriverOption.CONNECTION_POOL_LOCAL_SIZE))
+                    : "UNDEFINED";
+            LOG.info(
+                "Driver Setting for profile {}: global CL = {}, localDC (conf) = {}, localDC (API) = {}, LB class = {},"
+                    + " RetryPolicy class = {}, remote pool size = {}, local pool size = {}",
+                e.getKey(),
+                cl,
+                localDC,
+                localDCViaAPI,
+                lbClass,
+                rpClass,
+                remotePoolSize,
+                localPoolSize);
+          }
+        }
+      }
+    }
   }
 
   private CompletionStage<CqlSession> init(CqlIdentifier keyspace) {

core/src/main/java/com/yugabyte/oss/driver/api/core/DefaultPartitionMetadata.java

@@ -142,6 +142,7 @@ private Optional<Map<QualifiedTableName, TableSplitMetadata>> createPartitionMap
 
     Map<QualifiedTableName, TableSplitMetadata> tableSplits =
         new HashMap<QualifiedTableName, TableSplitMetadata>();
+    StringBuilder msg = new StringBuilder();
 
     for (AdminRow row : result) {
 
@@ -156,6 +157,7 @@ private Optional<Map<QualifiedTableName, TableSplitMetadata>> createPartitionMap
       Map<InetAddress, String> replicaAddresses =
           row.getMapOfInetAddressToString("replica_addresses");
 
+      boolean leaderFound = false;
       List<Node> hosts = new ArrayList<Node>();
       for (Map.Entry<InetAddress, String> entry : replicaAddresses.entrySet()) {
 
@@ -175,17 +177,23 @@ private Optional<Map<QualifiedTableName, TableSplitMetadata>> createPartitionMap
         // Put the leader at the beginning and the rest after.
         String role = entry.getValue();
         if (role.equals("LEADER")) {
+          leaderFound = true;
           hosts.add(0, host);
         } else if (role.equals("FOLLOWER") || role.equals("READ_REPLICA")) {
           hosts.add(host);
         }
       }
       int startKey = getKey(row.getByteBuffer("start_key"));
       int endKey = getKey(row.getByteBuffer("end_key"));
+      if (!leaderFound && LOG.isDebugEnabled()) {
+        msg.append(
+            tableId.getKeyspaceName() + "." + tableId.getTableName() + ": " + startKey + ", ");
+      }
       tableSplitMetadata
           .getPartitionMap()
           .put(startKey, new PartitionMetadata(startKey, endKey, hosts));
     }
+    LOG.debug("Created partition map. Tablets without leaders: {}", msg);
     return Optional.ofNullable(tableSplits);
   }