Batch read connection api native adjustments (#2569)

* chore: add ClientSideStatementPartitionExecutor to SpannerFeature

* chore: wrap AbstractStatementParser static initialization in try/catch

* chore: add ClientSideStatementRunPartitionExecutor to SpannerFeature

* chore: add ClientSideStatementRunPartitionedQueryExecutor to SpannerFeature

* chore: lint formatting

Author: burkedavison

google-cloud-spanner/src/main/java/com/google/cloud/spanner/connection/AbstractStatementParser.java

@@ -34,6 +34,8 @@
 import java.util.Objects;
 import java.util.Set;
 import java.util.concurrent.Callable;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
 /**
  * Internal class for the Spanner Connection API.
@@ -91,8 +93,7 @@ public static AbstractStatementParser getInstance(Dialect dialect) {
    */
 
   /** Begins a transaction. */
-  static final ParsedStatement BEGIN_STATEMENT =
-      AbstractStatementParser.getInstance(Dialect.GOOGLE_STANDARD_SQL).parse(Statement.of("BEGIN"));
+  static final ParsedStatement BEGIN_STATEMENT;
 
   /**
    * Create a COMMIT statement to use with the {@link #commit()} method to allow it to be cancelled,
@@ -104,14 +105,10 @@ public static AbstractStatementParser getInstance(Dialect dialect) {
    * #commit()} method is called directly, we do not have a {@link ParsedStatement}, and the method
    * uses this statement instead in order to use the same logic as the other statements.
    */
-  static final ParsedStatement COMMIT_STATEMENT =
-      AbstractStatementParser.getInstance(Dialect.GOOGLE_STANDARD_SQL)
-          .parse(Statement.of("COMMIT"));
+  static final ParsedStatement COMMIT_STATEMENT;
 
   /** The {@link Statement} and {@link Callable} for rollbacks */
-  static final ParsedStatement ROLLBACK_STATEMENT =
-      AbstractStatementParser.getInstance(Dialect.GOOGLE_STANDARD_SQL)
-          .parse(Statement.of("ROLLBACK"));
+  static final ParsedStatement ROLLBACK_STATEMENT;
 
   /**
    * Create a RUN BATCH statement to use with the {@link #executeBatchUpdate(Iterable)} method to
@@ -124,9 +121,22 @@ public static AbstractStatementParser getInstance(Dialect dialect) {
    * and the method uses this statement instead in order to use the same logic as the other
    * statements.
    */
-  static final ParsedStatement RUN_BATCH_STATEMENT =
-      AbstractStatementParser.getInstance(Dialect.GOOGLE_STANDARD_SQL)
-          .parse(Statement.of("RUN BATCH"));
+  static final ParsedStatement RUN_BATCH_STATEMENT;
+
+  static {
+    try {
+      BEGIN_STATEMENT = getInstance(Dialect.GOOGLE_STANDARD_SQL).parse(Statement.of("BEGIN"));
+      COMMIT_STATEMENT = getInstance(Dialect.GOOGLE_STANDARD_SQL).parse(Statement.of("COMMIT"));
+      ROLLBACK_STATEMENT = getInstance(Dialect.GOOGLE_STANDARD_SQL).parse(Statement.of("ROLLBACK"));
+      RUN_BATCH_STATEMENT =
+          getInstance(Dialect.GOOGLE_STANDARD_SQL).parse(Statement.of("RUN BATCH"));
+
+    } catch (Throwable ex) {
+      Logger logger = Logger.getLogger(AbstractStatementParser.class.getName());
+      logger.log(Level.SEVERE, "Static initialization failure.", ex);
+      throw ex;
+    }
+  }
 
   /** The type of statement that has been recognized by the parser. */
   @InternalApi

google-cloud-spanner/src/main/java/com/google/cloud/spanner/nativeimage/SpannerFeature.java

@@ -43,6 +43,12 @@ final class SpannerFeature implements Feature {
       "com.google.cloud.spanner.connection.ClientSideStatementPgBeginExecutor";
   private static final String CLIENT_SIDE_STATEMENT_EXPLAIN_EXECUTOR =
       "com.google.cloud.spanner.connection.ClientSideStatementExplainExecutor";
+  private static final String CLIENT_SIDE_STATEMENT_PARTITION_EXECUTOR =
+      "com.google.cloud.spanner.connection.ClientSideStatementPartitionExecutor";
+  private static final String CLIENT_SIDE_STATEMENT_RUN_PARTITION_EXECUTOR =
+      "com.google.cloud.spanner.connection.ClientSideStatementRunPartitionExecutor";
+  private static final String CLIENT_SIDE_STATEMENT_RUN_PARTITIONED_QUERY_EXECUTOR =
+      "com.google.cloud.spanner.connection.ClientSideStatementRunPartitionedQueryExecutor";
   private static final String ABSTRACT_STATEMENT_PARSER =
       "com.google.cloud.spanner.connection.AbstractStatementParser";
   private static final String STATEMENT_PARSER =
@@ -70,6 +76,17 @@ public void beforeAnalysis(BeforeAnalysisAccess access) {
     if (access.findClassByName(CLIENT_SIDE_STATEMENT_SET_EXECUTOR) != null) {
       NativeImageUtils.registerClassForReflection(access, CLIENT_SIDE_STATEMENT_SET_EXECUTOR);
     }
+    if (access.findClassByName(CLIENT_SIDE_STATEMENT_PARTITION_EXECUTOR) != null) {
+      NativeImageUtils.registerClassForReflection(access, CLIENT_SIDE_STATEMENT_PARTITION_EXECUTOR);
+    }
+    if (access.findClassByName(CLIENT_SIDE_STATEMENT_RUN_PARTITION_EXECUTOR) != null) {
+      NativeImageUtils.registerClassForReflection(
+          access, CLIENT_SIDE_STATEMENT_RUN_PARTITION_EXECUTOR);
+    }
+    if (access.findClassByName(CLIENT_SIDE_STATEMENT_RUN_PARTITIONED_QUERY_EXECUTOR) != null) {
+      NativeImageUtils.registerClassForReflection(
+          access, CLIENT_SIDE_STATEMENT_RUN_PARTITIONED_QUERY_EXECUTOR);
+    }
     if (access.findClassByName(CLIENT_SIDE_VALUE_CONVERTER) != null) {
       NativeImageUtils.registerClassHierarchyForReflection(access, CLIENT_SIDE_VALUE_CONVERTER);
     }