jdbc: add connection timeout configuration and handling

Added connection property `socketTimeout` to allow user control over
network timeout before actual connection is returned by the driver.
This is only done for default socket provider. The default timeout is
is left to be infinite.

Implemented `Connection.setNetworkTimeout` API to make it possible to
change the maximum amount of time to wait for server replies after the
connection is established.

The connection that has timed out is forced to close. New subsequent
operations requested on such connection must fail right away.
The corresponding checks are embedded into relevant APIs.

Closes tarantool#38

Author: ztarvos

src/main/java/org/tarantool/TarantoolConnection.java

@@ -4,6 +4,7 @@
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.net.Socket;
+import java.net.SocketException;
 import java.nio.ByteBuffer;
 import java.util.List;
 import java.util.Map;
@@ -80,4 +81,24 @@ protected void sql(String sql, Object[] bind) {
     public boolean isClosed() {
         return socket.isClosed();
     }
+
+    /**
+     * Sets given timeout value on underlying socket.
+     *
+     * @param timeout Timeout in milliseconds.
+     * @throws SocketException If failed.
+     */
+    public void setSocketTimeout(int timeout) throws SocketException {
+        socket.setSoTimeout(timeout);
+    }
+
+    /**
+     * Retrieves timeout value from underlying socket.
+     *
+     * @return Timeout in milliseconds.
+     * @throws SocketException If failed.
+     */
+    public int getSocketTimeout() throws SocketException {
+        return socket.getSoTimeout();
+    }
 }

src/main/java/org/tarantool/jdbc/SQLConnection.java

@@ -1,5 +1,7 @@
 package org.tarantool.jdbc;
 
+import java.io.IOException;
+import java.net.SocketException;
 import java.sql.Array;
 import java.sql.Blob;
 import java.sql.CallableStatement;
@@ -20,6 +22,7 @@
 import java.util.Properties;
 import java.util.concurrent.Executor;
 
+import org.tarantool.CommunicationException;
 import org.tarantool.TarantoolConnection;
 
 @SuppressWarnings("Since15")
@@ -36,12 +39,14 @@ public SQLConnection(TarantoolConnection connection, String url, Properties prop
 
     @Override
     public Statement createStatement() throws SQLException {
-        return new SQLStatement(connection, this);
+        checkNotClosed();
+        return new SQLStatement(this);
     }
 
     @Override
     public PreparedStatement prepareStatement(String sql) throws SQLException {
-        return new SQLPreparedStatement(connection, this, sql);
+        checkNotClosed();
+        return new SQLPreparedStatement(this, sql);
     }
 
     @Override
@@ -89,6 +94,7 @@ public boolean isClosed() throws SQLException {
 
     @Override
     public DatabaseMetaData getMetaData() throws SQLException {
+        checkNotClosed();
         return new SQLDatabaseMetadata(this);
     }
 
@@ -293,15 +299,28 @@ public void abort(Executor executor) throws SQLException {
 
     @Override
     public void setNetworkTimeout(Executor executor, int milliseconds) throws SQLException {
-        throw new SQLFeatureNotSupportedException();
+        checkNotClosed();
+
+        if (milliseconds < 0)
+            throw new SQLException("Network timeout cannot be negative.");
+
+        try {
+            connection.setSocketTimeout(milliseconds);
+        } catch (SocketException e) {
+            throw new SQLException("Failed to set socket timeout: timeout=" + milliseconds, e);
+        }
     }
 
     @Override
     public int getNetworkTimeout() throws SQLException {
-        throw new SQLFeatureNotSupportedException();
+        checkNotClosed();
+        try {
+            return connection.getSocketTimeout();
+        } catch (SocketException e) {
+            throw new SQLException("Failed to retrieve socket timeout", e);
+        }
     }
 
-
     @Override
     public <T> T unwrap(Class<T> iface) throws SQLException {
         throw new SQLFeatureNotSupportedException();
@@ -311,4 +330,28 @@ public <T> T unwrap(Class<T> iface) throws SQLException {
     public boolean isWrapperFor(Class<?> iface) throws SQLException {
         throw new SQLFeatureNotSupportedException();
     }
+
+    /**
+     * @throws SQLException If connection is closed.
+     */
+    protected void checkNotClosed() throws SQLException {
+        if (isClosed())
+            throw new SQLException("Connection is closed.");
+    }
+
+    /**
+     * Inspects passed exception and closes the connection if appropriate.
+     *
+     * @param e Exception to process.
+     */
+    protected void handleException(Exception e) {
+        if (CommunicationException.class.isAssignableFrom(e.getClass()) ||
+            IOException.class.isAssignableFrom(e.getClass())) {
+            try {
+                close();
+            } catch (SQLException ignored) {
+                // No-op.
+            }
+        }
+    }
 }

src/main/java/org/tarantool/jdbc/SQLDatabaseMetadata.java

@@ -672,23 +672,30 @@ protected boolean like(String value, String[] parts) {
 
     @Override
     public ResultSet getTables(String catalog, String schemaPattern, String tableNamePattern, String[] types)
-            throws SQLException {
-        if (types != null && !Arrays.asList(types).contains("TABLE")) {
-            return new SQLResultSet(JDBCBridge.EMPTY);
-        }
-        String[] parts = tableNamePattern == null ? new String[]{""} : tableNamePattern.split("%");
-        List<List<Object>> spaces = (List<List<Object>>) connection.connection.select(_VSPACE, 0, Arrays.asList(), 0, SPACES_MAX, 0);
-        List<List<Object>> rows = new ArrayList<List<Object>>();
-        for (List<Object> space : spaces) {
-            String name = (String) space.get(NAME_IDX);
-            Map flags = (Map) space.get(FLAGS_IDX);
-            if (flags != null && flags.containsKey("sql") && like(name, parts)) {
-                rows.add(Arrays.asList(name, "TABLE", flags.get("sql")));
+        throws SQLException {
+        connection.checkNotClosed();
+        try {
+            if (types != null && !Arrays.asList(types).contains("TABLE")) {
+                return new SQLResultSet(JDBCBridge.EMPTY);
+            }
+            String[] parts = tableNamePattern == null ? new String[]{""} : tableNamePattern.split("%");
+            List<List<Object>> spaces = (List<List<Object>>) connection.connection.select(_VSPACE, 0, Arrays.asList(), 0, SPACES_MAX, 0);
+            List<List<Object>> rows = new ArrayList<List<Object>>();
+            for (List<Object> space : spaces) {
+                String name = (String) space.get(NAME_IDX);
+                Map flags = (Map) space.get(FLAGS_IDX);
+                if (flags != null && flags.containsKey("sql") && like(name, parts)) {
+                    rows.add(Arrays.asList(name, "TABLE", flags.get("sql")));
+                }
             }
+            return new SQLNullResultSet(JDBCBridge.mock(Arrays.asList("TABLE_NAME", "TABLE_TYPE", "REMARKS",
+                    //nulls
+                    "TABLE_CAT", "TABLE_SCHEM", "TABLE_TYPE", "TYPE_CAT", "TYPE_SCHEM", "TYPE_NAME", "SELF_REFERENCING_COL_NAME", "REF_GENERATION"), rows));
+        } catch (Exception e) {
+            connection.handleException(e);
+            throw new SQLException("Failed to retrieve table(s) description: " +
+                "tableNamePattern=\"" + tableNamePattern + "\".", e);
         }
-        return new SQLNullResultSet(JDBCBridge.mock(Arrays.asList("TABLE_NAME", "TABLE_TYPE", "REMARKS",
-                //nulls
-                "TABLE_CAT", "TABLE_SCHEM", "TABLE_TYPE", "TYPE_CAT", "TYPE_SCHEM", "TYPE_NAME", "SELF_REFERENCING_COL_NAME", "REF_GENERATION"), rows));
     }
 
     @Override
@@ -713,32 +720,40 @@ public ResultSet getTableTypes() throws SQLException {
     @Override
     public ResultSet getColumns(String catalog, String schemaPattern, String tableNamePattern, String columnNamePattern)
             throws SQLException {
-        String[] tableParts = tableNamePattern == null ? new String[]{""} : tableNamePattern.split("%");
-        String[] colParts = columnNamePattern == null ? new String[]{""} : columnNamePattern.split("%");
-        List<List<Object>> spaces = (List<List<Object>>) connection.connection.select(_VSPACE, 0, Arrays.asList(), 0, SPACES_MAX, 0);
-        List<List<Object>> rows = new ArrayList<List<Object>>();
-        for (List<Object> space : spaces) {
-            String tableName = (String) space.get(NAME_IDX);
-            Map flags = (Map) space.get(FLAGS_IDX);
-            if (flags != null && flags.containsKey("sql") && like(tableName, tableParts)) {
-                List<Map<String, Object>> format = (List<Map<String, Object>>) space.get(FORMAT_IDX);
-                for (int columnIdx = 1; columnIdx <= format.size(); columnIdx++) {
-                    Map<String, Object> f = format.get(columnIdx - 1);
-                    String columnName = (String) f.get("name");
-                    String dbType = (String) f.get("type");
-                    if (like(columnName, colParts)) {
-                        rows.add(Arrays.<Object>asList(tableName, columnName, columnIdx, Types.OTHER, dbType, 10, 1, "YES", Types.OTHER, "NO", "NO"));
+        connection.checkNotClosed();
+        try {
+            String[] tableParts = tableNamePattern == null ? new String[]{""} : tableNamePattern.split("%");
+            String[] colParts = columnNamePattern == null ? new String[]{""} : columnNamePattern.split("%");
+            List<List<Object>> spaces = (List<List<Object>>) connection.connection.select(_VSPACE, 0, Arrays.asList(), 0, SPACES_MAX, 0);
+            List<List<Object>> rows = new ArrayList<List<Object>>();
+            for (List<Object> space : spaces) {
+                String tableName = (String) space.get(NAME_IDX);
+                Map flags = (Map) space.get(FLAGS_IDX);
+                if (flags != null && flags.containsKey("sql") && like(tableName, tableParts)) {
+                    List<Map<String, Object>> format = (List<Map<String, Object>>) space.get(FORMAT_IDX);
+                    for (int columnIdx = 1; columnIdx <= format.size(); columnIdx++) {
+                        Map<String, Object> f = format.get(columnIdx - 1);
+                        String columnName = (String) f.get("name");
+                        String dbType = (String) f.get("type");
+                        if (like(columnName, colParts)) {
+                            rows.add(Arrays.<Object>asList(tableName, columnName, columnIdx, Types.OTHER, dbType, 10, 1, "YES", Types.OTHER, "NO", "NO"));
+                        }
                     }
                 }
             }
-        }
 
-        return new SQLNullResultSet((JDBCBridge.mock(
-                Arrays.asList("TABLE_NAME", "COLUMN_NAME", "ORDINAL_POSITION", "DATA_TYPE", "TYPE_NAME", "NUM_PREC_RADIX", "NULLABLE", "IS_NULLABLE", "SOURCE_DATA_TYPE", "IS_AUTOINCREMENT", "IS_GENERATEDCOLUMN",
-                        //nulls
-                        "TABLE_CAT", "TABLE_SCHEM", "COLUMN_SIZE", "BUFFER_LENGTH", "DECIMAL_DIGITS", "REMARKS", "COLUMN_DEF", "SQL_DATA_TYPE", "SQL_DATETIME_SUB", "CHAR_OCTET_LENGTH", "SCOPE_CATALOG", "SCOPE_SCHEMA", "SCOPE_TABLE"
-                ),
-                rows)));
+            return new SQLNullResultSet((JDBCBridge.mock(
+                    Arrays.asList("TABLE_NAME", "COLUMN_NAME", "ORDINAL_POSITION", "DATA_TYPE", "TYPE_NAME", "NUM_PREC_RADIX", "NULLABLE", "IS_NULLABLE", "SOURCE_DATA_TYPE", "IS_AUTOINCREMENT", "IS_GENERATEDCOLUMN",
+                            //nulls
+                            "TABLE_CAT", "TABLE_SCHEM", "COLUMN_SIZE", "BUFFER_LENGTH", "DECIMAL_DIGITS", "REMARKS", "COLUMN_DEF", "SQL_DATA_TYPE", "SQL_DATETIME_SUB", "CHAR_OCTET_LENGTH", "SCOPE_CATALOG", "SCOPE_SCHEMA", "SCOPE_TABLE"
+                    ),
+                    rows)));
+        } catch (Exception e) {
+            connection.handleException(e);
+            throw new SQLException("Error processing table column metadata: " +
+                "tableNamePattern=\"" + tableNamePattern + "\"; " +
+                "columnNamePattern=\"" + columnNamePattern + "\".", e);
+        }
     }
 
     @Override
@@ -766,6 +781,8 @@ public ResultSet getVersionColumns(String catalog, String schema, String table)
 
     @Override
     public ResultSet getPrimaryKeys(String catalog, String schema, String table) throws SQLException {
+        connection.checkNotClosed();
+
         final List<String> colNames = Arrays.asList(
                 "TABLE_CAT", "TABLE_SCHEM", "TABLE_NAME", "COLUMN_NAME", "KEY_SEQ", "PK_NAME");
 
@@ -809,9 +826,9 @@ public int compare(List<Object> row0, List<Object> row1) {
                 }
             });
             return new SQLNullResultSet((JDBCBridge.mock(colNames, rows)));
-        }
-        catch (Throwable t) {
-            throw new SQLException("Error processing metadata for table \"" + table + "\".", t);
+        } catch (Exception e) {
+            connection.handleException(e);
+            throw new SQLException("Error processing metadata for table \"" + table + "\".", e);
         }
     }