Extract utility method resolveExecutionProfile()

Co-authored-by: Kevin Gallardo <kevin.gallardo@datastax.com>

Author: olim7t

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

@@ -20,7 +20,9 @@
 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.DriverConfig;
 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.cql.AsyncResultSet;
 import com.datastax.oss.driver.api.core.cql.BatchStatement;
 import com.datastax.oss.driver.api.core.cql.BatchableStatement;
@@ -51,6 +53,7 @@
 import com.datastax.oss.driver.api.core.servererrors.UnavailableException;
 import com.datastax.oss.driver.api.core.servererrors.WriteFailureException;
 import com.datastax.oss.driver.api.core.servererrors.WriteTimeoutException;
+import com.datastax.oss.driver.api.core.session.Request;
 import com.datastax.oss.driver.api.core.type.codec.TypeCodecs;
 import com.datastax.oss.driver.api.core.type.codec.registry.CodecRegistry;
 import com.datastax.oss.driver.internal.core.DefaultProtocolFeature;
@@ -94,6 +97,19 @@
  */
 public class Conversions {
 
+  public static DriverExecutionProfile resolveExecutionProfile(
+      Request request, DriverContext context) {
+    if (request.getExecutionProfile() != null) {
+      return request.getExecutionProfile();
+    } else {
+      DriverConfig config = context.getConfig();
+      String profileName = request.getExecutionProfileName();
+      return (profileName == null || profileName.isEmpty())
+          ? config.getDefaultProfile()
+          : config.getProfile(profileName);
+    }
+  }
+
   public static Message toMessage(
       Statement<?> statement, DriverExecutionProfile config, InternalDriverContext context) {
     ConsistencyLevel consistency =

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

@@ -21,7 +21,6 @@
 import com.datastax.oss.driver.api.core.ProtocolVersion;
 import com.datastax.oss.driver.api.core.RequestThrottlingException;
 import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
-import com.datastax.oss.driver.api.core.config.DriverConfig;
 import com.datastax.oss.driver.api.core.config.DriverExecutionProfile;
 import com.datastax.oss.driver.api.core.cql.PrepareRequest;
 import com.datastax.oss.driver.api.core.cql.PreparedStatement;
@@ -116,17 +115,7 @@ protected CqlPrepareHandlerBase(
     this.preparedStatementsCache = preparedStatementsCache;
     this.session = session;
     this.context = context;
-
-    if (request.getExecutionProfile() != null) {
-      this.executionProfile = request.getExecutionProfile();
-    } else {
-      DriverConfig config = context.getConfig();
-      String profileName = request.getExecutionProfileName();
-      this.executionProfile =
-          (profileName == null || profileName.isEmpty())
-              ? config.getDefaultProfile()
-              : config.getProfile(profileName);
-    }
+    this.executionProfile = Conversions.resolveExecutionProfile(request, context);
     this.queryPlan =
         context
             .getLoadBalancingPolicyWrapper()

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

@@ -21,7 +21,6 @@
 import com.datastax.oss.driver.api.core.DriverTimeoutException;
 import com.datastax.oss.driver.api.core.RequestThrottlingException;
 import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
-import com.datastax.oss.driver.api.core.config.DriverConfig;
 import com.datastax.oss.driver.api.core.config.DriverExecutionProfile;
 import com.datastax.oss.driver.api.core.connection.FrameTooLongException;
 import com.datastax.oss.driver.api.core.cql.AsyncResultSet;
@@ -143,17 +142,7 @@ protected CqlRequestHandlerBase(
     this.session = session;
     this.keyspace = session.getKeyspace().orElse(null);
     this.context = context;
-
-    if (statement.getExecutionProfile() != null) {
-      this.executionProfile = statement.getExecutionProfile();
-    } else {
-      DriverConfig config = context.getConfig();
-      String profileName = statement.getExecutionProfileName();
-      this.executionProfile =
-          (profileName == null || profileName.isEmpty())
-              ? config.getDefaultProfile()
-              : config.getProfile(profileName);
-    }
+    this.executionProfile = Conversions.resolveExecutionProfile(statement, context);
     if (this.statement.getNode() != null) {
       this.queryPlan = new QueryPlan(this.statement.getNode());