added mysq jdbc driver test

Signed-off-by: Harshit Gangal <harshit@planetscale.com>

java/example/pom.xml

@@ -29,6 +29,13 @@
       <artifactId>vitess-jdbc</artifactId>
     </dependency>
 
+    <dependency>
+      <groupId>mysql</groupId>
+      <artifactId>mysql-connector-java</artifactId>
+      <version>8.0.25</version>
+      <optional>false</optional>
+    </dependency>
+
     <dependency>
       <groupId>joda-time</groupId>
       <artifactId>joda-time</artifactId>

java/example/src/main/java/io/vitess/example/MysqlJDBCExample.java

@@ -0,0 +1,179 @@
+/*
+ * Copyright 2021 The Vitess Authors.
+
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+
+ *     http://www.apache.org/licenses/LICENSE-2.0
+
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.vitess.example;
+
+import org.joda.time.Instant;
+
+import java.sql.BatchUpdateException;
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.Random;
+
+
+public class MysqlJDBCExample {
+
+  public static void main(String[] args) throws Exception {
+    if (args.length != 1) {
+      System.out.println("usage: VitessJDBCExample <vtgate-host:port>");
+      System.exit(1);
+    }
+
+    // Connect to vtgate.
+    String dbURL = "jdbc:mysql://" + args[0];
+    try (Connection conn = DriverManager.getConnection(dbURL, null)) {
+      // Setting AutoCommit to false as VTTablet was not up with enable-autocommit
+      // Not Required if enable-autocommit flag is set in VTTablet
+      conn.setAutoCommit(false);
+
+      // Insert some messages on random pages.
+      System.out.println("Inserting into master...");
+      insertData(conn);
+
+      // To Commit Open Transaction
+      conn.commit();
+
+      // Read it back from master.
+      System.out.println("Reading from master...");
+      readData(conn);
+
+      // To Commit Open Transaction,
+      // as select was made on master with autocommit false a transaction was open
+      conn.commit();
+
+      // Read it back from replica.
+      dbURL += "/test_keyspace@replica";
+      try (Connection connReplica = DriverManager.getConnection(dbURL, null)) {
+        System.out.println("Validate it is connected to replica");
+        validateReplica(connReplica);
+        System.out.println("Reading from replica...");
+        readData(connReplica);
+      }
+
+      // Execute DML Queries in a Batch
+      batchedQueries(conn);
+
+      // To Commit Open Transaction
+      conn.commit();
+
+    } catch (Exception exc) {
+      System.out.println("Vitess JDBC example failed.");
+      System.out.println("Error Details:");
+      exc.printStackTrace();
+      System.exit(2);
+    }
+  }
+
+  private static void insertData(Connection conn) throws SQLException {
+    Random rand = new Random();
+    try (PreparedStatement stmt = conn
+        .prepareStatement("INSERT INTO messages (page,time_created_ns,message) VALUES (?,?,?)")) {
+      for (int i = 0; i < 3; i++) {
+        Instant timeCreated = Instant.now();
+        int page = rand.nextInt(100) + 1;
+        stmt.setInt(1, page);
+        stmt.setLong(2, timeCreated.getMillis() * 1000000);
+        stmt.setString(3, "V is for speed");
+        stmt.execute();
+      }
+    }
+  }
+
+  private static void readData(Connection conn) throws SQLException {
+    String sql = "SELECT page, time_created_ns, message FROM messages";
+    try (Statement stmt = conn.createStatement(); ResultSet rs = stmt.executeQuery(sql)) {
+      while (rs.next()) {
+        long page = rs.getLong("page");
+        long timeCreated = rs.getLong("time_created_ns");
+        String message = rs.getString("message");
+        System.out.format("(%s, %s, %s)\n", page, timeCreated, message);
+      }
+    }
+  }
+
+  private static void validateReplica(Connection conn) throws SQLException {
+    String sql = "show slave status";
+    try (Statement stmt = conn.createStatement(); ResultSet rs = stmt.executeQuery(sql)) {
+      if (!rs.next()) {
+        throw new RuntimeException("connected to wrong tablet");
+      }
+    }
+  }
+
+  private static void batchedQueries(Connection conn) throws SQLException {
+    Random rand = new Random();
+    try (PreparedStatement stmt = conn
+        .prepareStatement("INSERT INTO messages (page,time_created_ns,message) VALUES (?,?,?)")) {
+      for (int i = 0; i < 3; i++) {
+        Instant timeCreated = Instant.now();
+        int page = rand.nextInt(100) + 1;
+        stmt.setInt(1, page);
+        stmt.setLong(2, timeCreated.getMillis() * 1000000);
+        stmt.setString(3, "V is for speed");
+        stmt.addBatch();
+      }
+      int[] updateCounts;
+      try {
+        updateCounts = stmt.executeBatch();
+      } catch (BatchUpdateException ex) {
+        updateCounts = ex.getUpdateCounts();
+      }
+      if (null != updateCounts) {
+        evalBatchResult(updateCounts);
+      }
+    }
+
+    try (Statement stmt = conn.createStatement()) {
+      Instant timeCreated = Instant.now();
+      int page = rand.nextInt(100) + 1;
+      System.out.println("Page selected for all dml operation: " + page);
+      stmt.addBatch("INSERT INTO messages (page,time_created_ns,message) VALUES (" + page + ","
+          + timeCreated.getMillis() * 1000000 + ",'V is for speed')");
+      stmt.addBatch(
+          "UPDATE messages set message = 'V Batch is for more speed' where page = " + page);
+      stmt.addBatch("DELETE FROM messages where page = " + page);
+      int[] updateCounts;
+      try {
+        updateCounts = stmt.executeBatch();
+      } catch (BatchUpdateException ex) {
+        updateCounts = ex.getUpdateCounts();
+      }
+      if (null != updateCounts) {
+        evalBatchResult(updateCounts);
+      }
+    }
+
+  }
+
+  private static void evalBatchResult(int[] updateCounts) {
+    for (int i = 0; i < updateCounts.length; i++) {
+      switch (updateCounts[i]) {
+        case Statement.EXECUTE_FAILED:
+          System.out.println("execution failed");
+          break;
+        case Statement.SUCCESS_NO_INFO:
+          System.out.println("execution success with no result");
+          break;
+        default:
+          System.out.println("execution success with rows changed: " + updateCounts[i]);
+      }
+    }
+  }
+}

java/pom.xml

@@ -253,6 +253,9 @@
             <configuration>
               <failOnWarning>true</failOnWarning>
               <outputXML>true</outputXML>
+              <usedDependencies>
+                <dependency>mysql:mysql-connector-java</dependency>
+              </usedDependencies>
             </configuration>
           </execution>
         </executions>

test/client_jdbc.sh

@@ -46,7 +46,14 @@ fi
 # them when we run "exec:java". See also: http://stackoverflow.com/questions/11091311/maven-execjava-goal-on-a-multi-module-project
 # Install only "example". See also: http://stackoverflow.com/questions/1114026/maven-modules-building-a-single-specific-module
 mvn -f $VTROOT/java/pom.xml -pl example -am install -DskipTests
+# running tests using vitess-jdbc through grpc
 mvn -f $VTROOT/java/example/pom.xml exec:java \
     -Dexec.cleanupDaemonThreads=false \
     -Dexec.mainClass="io.vitess.example.VitessJDBCExample" \
     -Dexec.args="localhost:15991$optional_tls_args"
+
+# running tests using mysql protocol
+mvn -f $VTROOT/java/example/pom.xml exec:java \
+    -Dexec.cleanupDaemonThreads=false \
+    -Dexec.mainClass="io.vitess.example.MysqlJDBCExample" \
+    -Dexec.args="localhost:15306$optional_tls_args"