Question

我正在尝试模拟我的服务失去与数据库的连接并且无法通过阻止与iptables的连接来执行INSERT的情况，但是我无法使executeQuery()方法超时。

我所做的是为PreparedStatement设置超时，例如statement.setQueryTimeout(5)。这是代码。

HikariConfig config = new HikariConfig();

config.setJdbcUrl("jdbc:mysql://db-url/db");
config.setUsername("user");
config.setPassword("passwd");

config.setMaximumPoolSize(10);
config.setAutoCommit(false);
config.setConnectionTimeout(5000);
config.setDriverClassName("com.mysql.jdbc.Driver");
config.addDataSourceProperty("cachePrepStmts", "true");
config.addDataSourceProperty("prepStmtCacheSize", "250");
config.addDataSourceProperty("prepStmtCacheSqlLimit", "2048");
config.addDataSourceProperty("autoReconnect", "true");

final HikariDataSource pool = new HikariDataSource(config);

final String query = "INSERT INTO xtable VALUES (?, ?, ?, ?, ?)";

try ( Connection connection = pool.getConnection() )
{
    try ( PreparedStatement statement = connection.prepareStatement(query) )
    {
        // this is what I expect to work
        statement.setQueryTimeout(5);

        for ( Info info : infos )
        {
            statement.setString(1, info.getValue1());
            statement.setString(2, info.getValue2());
            statement.setString(3, info.getValue3());
            statement.setString(4, info.getValue4());
            statement.setString(5, info.getValue5());

            try
            {
                System.out.println("Waiting");
                Thread.sleep(5000);
                // I use this sleep to ban the database url with iptables
                // to simulate a disconnection
                System.out.println("Waited");
            }
            catch ( InterruptedException e )
            {
                e.printStackTrace();
            }

            System.out.println("Before executeQuery");
            statement.executeQuery();
            // I assumed that this would timeout after 5 seconds
            // But it never reaches the next System.out.print
            System.out.println("After executeQuery");
        }
    }

    System.out.println("Before commit");
    connection.commit();
    System.out.println("After commit");
}
catch ( SQLException e )
{
    log.error("Couldn't execute query", e);
}

输出结果为：

Waiting
Waited
Before executeQuery

然后它永远挂起......我该怎么办才能让它抛出异常？

Answer 1

在try-finally 中调用Connection.setNetworkTimeout() 。

private final static Executor immediateExecutor = Runnable::run; try ( Connection connection = pool.getConnection() ) { int timeout = connection.getNetworkTimeout(); connection.setNetworkTimeout(immediateExecutor, TimeUnit.SECONDS.toMillis(5)); ... try (PreparedStatement...) { ... } finally { connection.setNetworkTimeout(timeout); } } finally { ... }

您遇到了未经确认的TCP流量，如果未设置网络超时，则可能会挂断连接。

Answer 2

你总是可以：

try{
     preparedstatement = connection.prepareStatement(query);
     preparedstatement.setQueryTimeout(seconds);
}

而不是你做了什么。也许这会更好。同样的事情：

Connection connection = pool.getConnection()

即使明确设置，PreparedStatement也不会超时

2 个答案: