Support in GoogleSQL dialect

Author: sakthivelmanii

google-cloud-spanner/src/main/resources/com/google/cloud/spanner/connection/ClientSideStatements.json

@@ -753,6 +753,24 @@
 	    "converterName": "ClientSideStatementValueConverters$BooleanConverter"
 	  }
 	},
+  {
+    "name": "SET LOCAL DML_BATCH_UPDATE_COUNT = <INT64>",
+    "executorName": "ClientSideStatementSetExecutor",
+    "resultType": "NO_RESULT",
+    "statementType": "SET_DML_BATCH_UPDATE_COUNT",
+    "regex": "(?is)\\A\\s*set\\s+((?:session|local)\\s+)dml_batch_update_count\\s*(?:=)\\s*(.*)\\z",
+    "method": "statementSetDmlBatchUpdateCount",
+    "exampleStatements": [
+      "set local dml_batch_update_count = 0",
+      "set local dml_batch_update_count = 100"
+    ],
+    "setStatement": {
+      "propertyName": "DML_BATCH_UPDATE_COUNT",
+      "separator": "(?:=|\\s+TO\\s+)",
+      "allowedValues": "(\\d{1,19})",
+      "converterName": "ClientSideStatementValueConverters$LongConverter"
+    }
+  },
 	{
 		"name": "SHOW VARIABLE READ_LOCK_MODE",
 		"executorName": "ClientSideStatementNoParamExecutor",

google-cloud-spanner/src/test/java/com/google/cloud/spanner/connection/ConnectionAsyncApiTest.java

@@ -365,6 +365,30 @@ public void testDmlBatchUpdateCount() {
     }
   }
 
+  @Test
+  public void testDmlBatchUpdateCountGoogleSql() {
+    SpannerPool.closeSpannerPool();
+    try {
+      try (Connection connection = createConnection()) {
+        connection.execute(Statement.of("set local dml_batch_update_count = 1"));
+        connection.execute(Statement.of("START BATCH DML"));
+        List<Statement> statements = Arrays.asList(INSERT_STATEMENT, INSERT_STATEMENT);
+        long[] updateCounts = connection.executeBatchUpdate(statements);
+        assertThat(updateCounts).asList().containsExactly(1L, 1L);
+      }
+      try (Connection connection = createConnection()) {
+        connection.execute(Statement.of("START BATCH DML"));
+        List<Statement> statements = Arrays.asList(INSERT_STATEMENT, INSERT_STATEMENT);
+        long[] updateCounts = connection.executeBatchUpdate(statements);
+        assertThat(updateCounts).asList().containsExactly(-1L, -1L);
+      }
+    } finally {
+      SpannerPool.closeSpannerPool();
+      mockSpanner.putStatementResult(
+          MockSpannerServiceImpl.StatementResult.detectDialectResult(Dialect.GOOGLE_STANDARD_SQL));
+    }
+  }
+
   @Test
   public void testAutocommitRunBatchAsync() {
     try (Connection connection = createConnection()) {