Examples with MysqlConnection com.mysql.cj.MysqlConnection used on opensource projects

Example 11 with MysqlConnection

use of com.mysql.cj.MysqlConnection in project aws-mysql-jdbc by awslabs.

the class StatementRegressionTest method testBug11540.

@Test
public void testBug11540() throws Exception {
    Locale originalLocale = Locale.getDefault();
    Connection thaiConn = null;
    Statement thaiStmt = null;
    PreparedStatement thaiPrepStmt = null;
    try {
        createTable("testBug11540", "(field1 DATE, field2 DATETIME)");
        this.stmt.executeUpdate("INSERT INTO testBug11540 VALUES (NOW(), NOW())");
        Locale.setDefault(new Locale("th", "TH"));
        Properties props = new Properties();
        props.setProperty(PropertyKey.sslMode.getKeyName(), SslMode.DISABLED.name());
        props.setProperty(PropertyKey.allowPublicKeyRetrieval.getKeyName(), "true");
        props.setProperty(PropertyKey.jdbcCompliantTruncation.getKeyName(), "false");
        props.setProperty(PropertyKey.connectionTimeZone.getKeyName(), "LOCAL");
        thaiConn = getConnectionWithProps(props);
        thaiStmt = thaiConn.createStatement();
        this.rs = thaiStmt.executeQuery("SELECT field1, field2 FROM testBug11540");
        this.rs.next();
        Date origDate = this.rs.getDate(1);
        Timestamp origTimestamp = this.rs.getTimestamp(2);
        this.rs.close();
        thaiStmt.executeUpdate("TRUNCATE TABLE testBug11540");
        thaiPrepStmt = ((com.mysql.cj.jdbc.JdbcConnection) thaiConn).clientPrepareStatement("INSERT INTO testBug11540 VALUES (?,?)");
        thaiPrepStmt.setDate(1, origDate);
        thaiPrepStmt.setTimestamp(2, origTimestamp);
        thaiPrepStmt.executeUpdate();
        this.rs = thaiStmt.executeQuery("SELECT field1, field2 FROM testBug11540");
        this.rs.next();
        Date testDate = this.rs.getDate(1);
        Timestamp testTimestamp = this.rs.getTimestamp(2);
        this.rs.close();
        props = new Properties();
        props.setProperty(PropertyKey.sslMode.getKeyName(), SslMode.DISABLED.name());
        props.setProperty(PropertyKey.allowPublicKeyRetrieval.getKeyName(), "true");
        props.setProperty(PropertyKey.connectionTimeZone.getKeyName(), "LOCAL");
        Connection testConn = getConnectionWithProps(props);
        TimeZone serverTz = ((MysqlConnection) testConn).getSession().getServerSession().getSessionTimeZone();
        Timestamp expTs = Timestamp.from(origTimestamp.toInstant().atZone(ZoneId.systemDefault()).withZoneSameInstant(serverTz.toZoneId()).withZoneSameLocal(ZoneId.systemDefault()).toInstant());
        assertEquals(origDate, testDate);
        assertEquals(expTs, testTimestamp);
    } finally {
        Locale.setDefault(originalLocale);
    }
}

Example 12 with MysqlConnection

use of com.mysql.cj.MysqlConnection in project aws-mysql-jdbc by awslabs.

the class StatementRegressionTest method testLoadData.

/*
     * public void testBug9595() throws Exception { double[] vals = new double[]
     * {52.21, 52.22, 52.23, 52.24};
     * 
     * createTable("testBug9595", "(field1 DECIMAL(10,2), sortField INT)");
     * 
     * this.pstmt = this.conn.prepareStatement("INSERT INTO testBug9595 VALUES
     * (?, ?)"); // Try setting as doubles for (int i = 0; i < vals.length; i++)
     * { this.pstmt.setDouble(1, vals[i]); this.pstmt.setInt(2, i);
     * this.pstmt.executeUpdate(); }
     * 
     * this.pstmt = this.conn.prepareStatement("SELECT field1 FROM testBug9595
     * ORDER BY sortField"); this.rs = this.pstmt.executeQuery();
     * 
     * int i = 0;
     * 
     * while (this.rs.next()) { double valToTest = vals[i++];
     * 
     * assertEquals(this.rs.getDouble(1), valToTest, 0.001);
     * assertEquals(this.rs.getBigDecimal(1).doubleValue(), valToTest, 0.001); }
     * 
     * this.pstmt = this.conn.prepareStatement("INSERT INTO testBug9595 VALUES
     * (?, ?)");
     * 
     * this.stmt.executeUpdate("TRUNCATE TABLE testBug9595"); // Now, as
     * BigDecimals for (i = 0; i < vals.length; i++) { BigDecimal foo = new
     * BigDecimal(vals[i]);
     * 
     * this.pstmt.setObject(1, foo, Types.DECIMAL, 2); this.pstmt.setInt(2, i);
     * this.pstmt.executeUpdate(); }
     * 
     * this.pstmt = this.conn.prepareStatement("SELECT field1 FROM testBug9595
     * ORDER BY sortField"); this.rs = this.pstmt.executeQuery();
     * 
     * i = 0;
     * 
     * while (this.rs.next()) { double valToTest = vals[i++];
     * System.out.println(this.rs.getString(1));
     * assertEquals(this.rs.getDouble(1), valToTest, 0.001);
     * assertEquals(this.rs.getBigDecimal(1).doubleValue(), valToTest, 0.001); }
     * }
     */
/**
 * Tests that 'LOAD DATA LOCAL INFILE' works
 *
 * @throws Exception
 */
@Test
public void testLoadData() throws Exception {
    assumeTrue(supportsLoadLocalInfile(this.stmt), "This test requires the server started with --local-infile=ON");
    createTable("loadDataRegress", "(field1 int, field2 int)");
    // int maxAllowedPacket = 1048576;
    File tempFile = File.createTempFile("mysql", ".txt");
    // tempFile.deleteOnExit();
    System.out.println(tempFile);
    Writer out = new FileWriter(tempFile);
    int localCount = 0;
    // maxAllowedPacket * 4;
    int rowCount = 128;
    for (int i = 0; i < rowCount; i++) {
        out.write((localCount++) + "\t" + (localCount++) + "\n");
    }
    out.close();
    StringBuilder fileNameBuf = null;
    if (File.separatorChar == '\\') {
        fileNameBuf = new StringBuilder();
        String fileName = tempFile.getAbsolutePath();
        int fileNameLength = fileName.length();
        for (int i = 0; i < fileNameLength; i++) {
            char c = fileName.charAt(i);
            if (c == '\\') {
                fileNameBuf.append("/");
            } else {
                fileNameBuf.append(c);
            }
        }
    } else {
        fileNameBuf = new StringBuilder(tempFile.getAbsolutePath());
    }
    final String fileName = fileNameBuf.toString();
    assertThrows(SQLSyntaxErrorException.class, versionMeetsMinimum(8, 0, 19) ? "Loading local data is disabled;.*" : "The used command is not allowed with this MySQL version", () -> {
        this.stmt.executeUpdate("LOAD DATA LOCAL INFILE '" + fileName + "' INTO TABLE loadDataRegress CHARACTER SET " + CharsetMappingWrapper.getStaticMysqlCharsetForJavaEncoding(((MysqlConnection) this.conn).getPropertySet().getStringProperty(PropertyKey.characterEncoding).getValue(), this.serverVersion));
        return null;
    });
    Properties props = new Properties();
    props.setProperty(PropertyKey.sslMode.getKeyName(), SslMode.DISABLED.name());
    props.setProperty(PropertyKey.allowPublicKeyRetrieval.getKeyName(), "true");
    props.setProperty(PropertyKey.allowLoadLocalInfile.getKeyName(), "true");
    Connection testConn = getConnectionWithProps(props);
    int updateCount = testConn.createStatement().executeUpdate("LOAD DATA LOCAL INFILE '" + fileNameBuf.toString() + "' INTO TABLE loadDataRegress CHARACTER SET " + CharsetMappingWrapper.getStaticMysqlCharsetForJavaEncoding(((MysqlConnection) this.conn).getPropertySet().getStringProperty(PropertyKey.characterEncoding).getValue(), this.serverVersion));
    assertTrue(updateCount == rowCount);
}

Example 13 with MysqlConnection

use of com.mysql.cj.MysqlConnection in project aws-mysql-jdbc by awslabs.

the class ConnectionRegressionTest method assertCurrentUser.

private void assertCurrentUser(String url, Properties props, String expectedUser, boolean sslRequired) throws SQLException {
    Connection connection = url == null ? getConnectionWithProps(props) : getConnectionWithProps(url, props);
    if (sslRequired) {
        assertTrue(((MysqlConnection) connection).getSession().isSSLEstablished(), "SSL connection isn't established!");
    }
    Statement st = connection.createStatement();
    ResultSet rset = st.executeQuery("SELECT USER(), CURRENT_USER()");
    rset.next();
    assertEquals(expectedUser, rset.getString(1).split("@")[0]);
    assertEquals(expectedUser, rset.getString(2).split("@")[0]);
    connection.close();
}

Example 14 with MysqlConnection

use of com.mysql.cj.MysqlConnection in project aws-mysql-jdbc by awslabs.

the class ConnectionRegressionTest method testBug88242.

/**
 * Tests fix for Bug#88242 - autoReconnect and socketTimeout JDBC option makes wrong order of client packet.
 *
 * The wrong behavior may not be observed in all systems or configurations. It seems to be easier to reproduce when SSL is enabled. Without it, the data
 * packets flow faster and desynchronization occurs rarely, which is the root cause for this problem.
 *
 * @throws Exception
 */
@Test
public void testBug88242() throws Exception {
    assumeTrue((((MysqlConnection) this.conn).getSession().getServerSession().getCapabilities().getCapabilityFlags() & NativeServerSession.CLIENT_SSL) != 0, "This test requires server with SSL support.");
    assumeTrue(supportsTLSv1_2(((MysqlConnection) this.conn).getSession().getServerSession().getServerVersion()), "This test requires server with TLSv1.2+ support.");
    assumeTrue(supportsTestCertificates(this.stmt), "This test requires the server configured with SSL certificates from ConnectorJ/src/test/config/ssl-test-certs");
    Properties props = new Properties();
    props.setProperty(PropertyKey.sslMode.getKeyName(), SslMode.REQUIRED.name());
    props.setProperty(PropertyKey.autoReconnect.getKeyName(), "true");
    props.setProperty(PropertyKey.socketTimeout.getKeyName(), "1500");
    Connection testConn = getConnectionWithProps(props);
    PreparedStatement ps = testConn.prepareStatement("SELECT ?, SLEEP(?)");
    int key = 0;
    for (int i = 0; i < 5; i++) {
        // Execute a query that runs faster than the socket timeout limit.
        ps.setInt(1, ++key);
        ps.setInt(2, 0);
        try {
            ResultSet rset = ps.executeQuery();
            assertTrue(rset.next());
            assertEquals(key, rset.getInt(1));
        } catch (SQLException e) {
            e.printStackTrace();
            fail("Exception [" + e.getClass().getName() + ": " + e.getMessage() + "] caught when no exception was expected.");
        }
        // Execute a query that runs slower than the socket timeout limit.
        ps.setInt(1, ++key);
        ps.setInt(2, 2);
        final PreparedStatement localPstmt = ps;
        assertThrows("Communications link failure.*", SQLException.class, new Callable<Void>() {

            public Void call() throws Exception {
                localPstmt.executeQuery();
                return null;
            }
        });
    }
    testConn.close();
}

Example 15 with MysqlConnection

use of com.mysql.cj.MysqlConnection in project aws-mysql-jdbc by awslabs.

the class ConnectionRegressionTest method testAuthCleartextPlugin.

@Test
public void testAuthCleartextPlugin() throws Exception {
    assumeTrue((((MysqlConnection) this.conn).getSession().getServerSession().getCapabilities().getCapabilityFlags() & NativeServerSession.CLIENT_SSL) != 0, "This test requires server with SSL support.");
    assumeTrue(supportsTLSv1_2(((MysqlConnection) this.conn).getSession().getServerSession().getServerVersion()), "This test requires server with TLSv1.2+ support.");
    assumeTrue(supportsTestCertificates(this.stmt), "This test requires the server configured with SSL certificates from ConnectorJ/src/test/config/ssl-test-certs");
    boolean installPluginInRuntime = false;
    try {
        // Install plugin if required.
        this.rs = this.stmt.executeQuery("SELECT (PLUGIN_LIBRARY LIKE 'auth_test_plugin%') as `TRUE` FROM INFORMATION_SCHEMA.PLUGINS " + "WHERE PLUGIN_NAME='cleartext_plugin_server'");
        if (this.rs.next()) {
            installPluginInRuntime = !this.rs.getBoolean(1);
        } else {
            installPluginInRuntime = true;
        }
        if (installPluginInRuntime) {
            try {
                String ext = isServerRunningOnWindows() ? ".dll" : ".so";
                this.stmt.executeUpdate("INSTALL PLUGIN cleartext_plugin_server SONAME 'auth_test_plugin" + ext + "'");
            } catch (SQLException e) {
                if (e.getErrorCode() == MysqlErrorNumbers.ER_CANT_OPEN_LIBRARY) {
                    // To disable plugin deinstallation attempt in the finally block.
                    installPluginInRuntime = false;
                    assumeTrue(false, "This test requires a server installed with the test package.");
                } else {
                    throw e;
                }
            }
        }
        String dbname = getPropertiesFromTestsuiteUrl().getProperty(PropertyKey.DBNAME.getKeyName());
        assertFalse(StringUtils.isNullOrEmpty(dbname), "No database selected");
        createUser("'wl5735user'@'%'", "identified WITH cleartext_plugin_server AS ''");
        this.stmt.executeUpdate("DELETE FROM mysql.db WHERE user='wl5735user'");
        this.stmt.executeUpdate("INSERT INTO mysql.db (Host, Db, User, Select_priv, Insert_priv, Update_priv, Delete_priv, Create_priv, " + "Drop_priv, Grant_priv, References_priv, Index_priv, Alter_priv, Create_tmp_table_priv, Lock_tables_priv, Create_view_priv, " + "Show_view_priv, Create_routine_priv, Alter_routine_priv, Execute_priv, Event_priv, Trigger_priv) VALUES ('%', '" + dbname + "', 'wl5735user', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'N', 'N')");
        this.stmt.executeUpdate("FLUSH PRIVILEGES");
        Properties props = new Properties();
        props.setProperty(PropertyKey.USER.getKeyName(), "wl5735user");
        props.setProperty(PropertyKey.PASSWORD.getKeyName(), "");
        props.setProperty(PropertyKey.sslMode.getKeyName(), SslMode.DISABLED.name());
        assertThrows(SQLException.class, "SSL connection required for plugin \"mysql_clear_password\"\\. Check if 'sslMode' is enabled\\.", () -> getConnectionWithProps(props));
        String trustStorePath = "src/test/config/ssl-test-certs/ca-truststore";
        System.setProperty("javax.net.ssl.trustStore", trustStorePath);
        System.setProperty("javax.net.ssl.trustStorePassword", "password");
        props.setProperty(PropertyKey.sslMode.getKeyName(), SslMode.REQUIRED.name());
        try (Connection testConn = getConnectionWithProps(props)) {
            assertTrue(((MysqlConnection) testConn).getSession().isSSLEstablished(), "SSL connection isn't actually established!");
            Statement testSt = testConn.createStatement();
            ResultSet testRs = testSt.executeQuery("SELECT USER(), CURRENT_USER()");
            assertTrue(testRs.next());
            assertEquals("wl5735user", testRs.getString(1).split("@")[0]);
            assertEquals("wl5735user", testRs.getString(2).split("@")[0]);
        }
    } finally {
        if (installPluginInRuntime) {
            this.stmt.executeUpdate("UNINSTALL PLUGIN cleartext_plugin_server");
        }
    }
}