Skip to content

Commit 30427b3

Browse files
nicktorwaldnicktorwald
committed
Add support for JDBC result set type
Provide support for FORWARD_ONLY and INSENSITIVE scroll types. Now statements as well as theirs resultSets can be built using a forward only iterator or full implemented insensitive iterator. To achieve this iteration logic was extracted into two separate classes to support two scroll types respectively. This is a cross-cutting support through SQLMetadata, SQLConnection, SQL(Prepared)Statement, SQLResultSet. Extend SQLStates constants. '24000' state was added in scope of iterator support. Add missed implementation of a closing for SQLStatement and SQLResultSet. Deprecate JDBCBridge. This redundant class should be removed completely in scope of another task. Mapping of field labels was moved to SQLResultSet. Closes: #85 Affectes: #119, #108
1 parent 06755d5 commit 30427b3

20 files changed

+1328
-186
lines changed

Diff for: ‎src/main/java/org/tarantool/JDBCBridge.java

+12-34
Original file line numberDiff line numberDiff line change
@@ -1,34 +1,26 @@
11
package org.tarantool;
22

3+
import org.tarantool.protocol.TarantoolPacket;
4+
35
import java.util.ArrayList;
46
import java.util.Collections;
5-
import java.util.LinkedHashMap;
67
import java.util.List;
7-
import java.util.ListIterator;
8-
import java.util.Map;
9-
10-
import org.tarantool.jdbc.SQLResultSet;
11-
import org.tarantool.protocol.TarantoolPacket;
128

9+
@Deprecated
1310
public class JDBCBridge {
1411

15-
public static final JDBCBridge EMPTY = new JDBCBridge(Collections.<TarantoolBase.SQLMetaData>emptyList(), Collections.<List<Object>>emptyList());
12+
public static final JDBCBridge EMPTY = new JDBCBridge(Collections.emptyList(), Collections.emptyList());
1613

17-
final List<TarantoolBase.SQLMetaData> sqlMetadata;
18-
final Map<String,Integer> columnsByName;
14+
final List<SqlProtoUtils.SQLMetaData> sqlMetadata;
1915
final List<List<Object>> rows;
2016

2117
protected JDBCBridge(TarantoolPacket pack) {
2218
this(SqlProtoUtils.getSQLMetadata(pack), SqlProtoUtils.getSQLData(pack));
2319
}
2420

25-
protected JDBCBridge(List<TarantoolBase.SQLMetaData> sqlMetadata, List<List<Object>> rows) {
21+
protected JDBCBridge(List<SqlProtoUtils.SQLMetaData> sqlMetadata, List<List<Object>> rows) {
2622
this.sqlMetadata = sqlMetadata;
2723
this.rows = rows;
28-
columnsByName = new LinkedHashMap<String, Integer>((int) Math.ceil(sqlMetadata.size() / 0.75), 0.75f);
29-
for (int i = 0; i < sqlMetadata.size(); i++) {
30-
columnsByName.put(sqlMetadata.get(i).getName(), i + 1);
31-
}
3224
}
3325

3426
public static JDBCBridge query(TarantoolConnection connection, String sql, Object ... params) {
@@ -41,9 +33,9 @@ public static int update(TarantoolConnection connection, String sql, Object ...
4133
}
4234

4335
public static JDBCBridge mock(List<String> fields, List<List<Object>> values) {
44-
List<TarantoolBase.SQLMetaData> meta = new ArrayList<TarantoolBase.SQLMetaData>(fields.size());
36+
List<SqlProtoUtils.SQLMetaData> meta = new ArrayList<>(fields.size());
4537
for(String field : fields) {
46-
meta.add(new TarantoolBase.SQLMetaData(field));
38+
meta.add(new SqlProtoUtils.SQLMetaData(field));
4739
}
4840
return new JDBCBridge(meta, values);
4941
}
@@ -57,32 +49,18 @@ public static Object execute(TarantoolConnection connection, String sql, Object
5749
return rowCount.intValue();
5850
}
5951

60-
61-
public String getColumnName(int columnIndex) {
62-
return columnIndex > sqlMetadata.size() ? null : sqlMetadata.get(columnIndex - 1).getName();
63-
}
64-
65-
public Integer getColumnIndex(String columnName) {
66-
return columnsByName.get(columnName);
67-
}
68-
69-
public int getColumnCount() {
70-
return columnsByName.size();
71-
}
72-
73-
public ListIterator<List<Object>> iterator() {
74-
return rows.listIterator();
52+
public List<List<Object>> getRows() {
53+
return rows;
7554
}
7655

77-
public int size() {
78-
return rows.size();
56+
public List<SqlProtoUtils.SQLMetaData> getSqlMetadata() {
57+
return sqlMetadata;
7958
}
8059

8160
@Override
8261
public String toString() {
8362
return "JDBCBridge{" +
8463
"sqlMetadata=" + sqlMetadata +
85-
", columnsByName=" + columnsByName +
8664
", rows=" + rows +
8765
'}';
8866
}

Diff for: ‎src/main/java/org/tarantool/SqlProtoUtils.java

+23-4
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ public static List<Map<String, Object>> readSqlResult(TarantoolPacket pack) {
1212
List<List<?>> data = (List<List<?>>) pack.getBody().get(Key.DATA.getId());
1313

1414
List<Map<String, Object>> values = new ArrayList<>(data.size());
15-
List<TarantoolBase.SQLMetaData> metaData = getSQLMetadata(pack);
15+
List<SQLMetaData> metaData = getSQLMetadata(pack);
1616
for (List row : data) {
1717
LinkedHashMap<String, Object> value = new LinkedHashMap<>();
1818
for (int i = 0; i < row.size(); i++) {
@@ -27,11 +27,11 @@ public static List<List<Object>> getSQLData(TarantoolPacket pack) {
2727
return (List<List<Object>>) pack.getBody().get(Key.DATA.getId());
2828
}
2929

30-
public static List<TarantoolBase.SQLMetaData> getSQLMetadata(TarantoolPacket pack) {
30+
public static List<SQLMetaData> getSQLMetadata(TarantoolPacket pack) {
3131
List<Map<Integer, Object>> meta = (List<Map<Integer, Object>>) pack.getBody().get(Key.SQL_METADATA.getId());
32-
List<TarantoolBase.SQLMetaData> values = new ArrayList<TarantoolBase.SQLMetaData>(meta.size());
32+
List<SQLMetaData> values = new ArrayList<>(meta.size());
3333
for (Map<Integer, Object> c : meta) {
34-
values.add(new TarantoolBase.SQLMetaData((String) c.get(Key.SQL_FIELD_NAME.getId())));
34+
values.add(new SQLMetaData((String) c.get(Key.SQL_FIELD_NAME.getId())));
3535
}
3636
return values;
3737
}
@@ -44,4 +44,23 @@ public static Long getSqlRowCount(TarantoolPacket pack) {
4444
}
4545
return null;
4646
}
47+
48+
public static class SQLMetaData {
49+
protected String name;
50+
51+
public SQLMetaData(String name) {
52+
this.name = name;
53+
}
54+
55+
public String getName() {
56+
return name;
57+
}
58+
59+
@Override
60+
public String toString() {
61+
return "SQLMetaData{" +
62+
"name='" + name + '\'' +
63+
'}';
64+
}
65+
}
4766
}

Diff for: ‎src/main/java/org/tarantool/TarantoolBase.java

-19
Original file line numberDiff line numberDiff line change
@@ -31,25 +31,6 @@ public TarantoolBase(String username, String password, Socket socket) {
3131
}
3232
}
3333

34-
protected static class SQLMetaData {
35-
protected String name;
36-
37-
public SQLMetaData(String name) {
38-
this.name = name;
39-
}
40-
41-
public String getName() {
42-
return name;
43-
}
44-
45-
@Override
46-
public String toString() {
47-
return "SQLMetaData{" +
48-
"name='" + name + '\'' +
49-
'}';
50-
}
51-
}
52-
5334
protected TarantoolException serverError(long code, Object error) {
5435
return new TarantoolException(code, error instanceof String ? (String) error : new String((byte[]) error));
5536
}

Diff for: ‎src/main/java/org/tarantool/jdbc/SQLConnection.java

+33-3
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,11 @@
4040
import static org.tarantool.jdbc.SQLDriver.PROP_SOCKET_TIMEOUT;
4141
import static org.tarantool.jdbc.SQLDriver.PROP_USER;
4242

43-
@SuppressWarnings("Since15")
43+
/**
44+
* Tarantool {@link Connection} implementation
45+
*
46+
* Supports creating {@link Statement} and {@link PreparedStatement} instances
47+
*/
4448
public class SQLConnection implements Connection {
4549

4650
private static final int UNSET_HOLDABILITY = 0;
@@ -91,6 +95,7 @@ public class SQLConnection implements Connection {
9195
* to honor this timeout for the following read/write operations as well.
9296
*
9397
* @return Connected socket.
98+
*
9499
* @throws SQLException if failed.
95100
*/
96101
protected Socket getConnectedSocket() throws SQLException {
@@ -140,7 +145,9 @@ protected Socket makeSocket() {
140145
* @param user User name.
141146
* @param pass Password.
142147
* @param socket Connected socket.
148+
*
143149
* @return Native tarantool connection.
150+
*
144151
* @throws IOException if failed.
145152
*/
146153
protected TarantoolConnection makeConnection(String user, String pass, Socket socket) throws IOException {
@@ -171,7 +178,7 @@ public String nativeSQL(String sql) throws SQLException {
171178

172179
@Override
173180
public void setAutoCommit(boolean autoCommit) throws SQLException {
174-
if (autoCommit == false) {
181+
if (!autoCommit) {
175182
throw new SQLFeatureNotSupportedException();
176183
}
177184
}
@@ -319,6 +326,7 @@ public Statement createStatement(int resultSetType,
319326
int resultSetConcurrency,
320327
int resultSetHoldability) throws SQLException {
321328
checkNotClosed();
329+
checkResultSetType(resultSetType);
322330
checkHoldabilitySupport(resultSetHoldability);
323331
return new SQLStatement(this, resultSetType, resultSetConcurrency, resultSetHoldability);
324332
}
@@ -329,6 +337,7 @@ public PreparedStatement prepareStatement(String sql,
329337
int resultSetConcurrency,
330338
int resultSetHoldability) throws SQLException {
331339
checkNotClosed();
340+
checkResultSetType(resultSetType);
332341
checkHoldabilitySupport(resultSetHoldability);
333342
return new SQLPreparedStatement(this, sql, resultSetType, resultSetConcurrency, resultSetHoldability);
334343
}
@@ -540,12 +549,32 @@ private void handleException(Exception e) {
540549
}
541550
}
542551

552+
/**
553+
* Checks wether <code>resultSetType</code> is supported
554+
*
555+
* @param resultSetType param to be checked
556+
*
557+
* @throws SQLFeatureNotSupportedException param is not supported
558+
* @throws SQLNonTransientException param has invalid value
559+
*/
560+
private void checkResultSetType(int resultSetType) throws SQLException {
561+
if (resultSetType != ResultSet.TYPE_FORWARD_ONLY
562+
&& resultSetType != ResultSet.TYPE_SCROLL_INSENSITIVE
563+
&& resultSetType != ResultSet.TYPE_SCROLL_SENSITIVE) {
564+
throw new SQLNonTransientException("", SQLStates.INVALID_PARAMETER_VALUE.getSqlState());
565+
}
566+
if (!getMetaData().supportsResultSetType(resultSetType)) {
567+
throw new SQLFeatureNotSupportedException();
568+
}
569+
}
570+
543571
/**
544572
* Checks whether <code>holdability</code> is supported
545573
*
546574
* @param holdability param to be checked
575+
*
547576
* @throws SQLFeatureNotSupportedException param is not supported
548-
* @throws SQLNonTransientException param has invalid value
577+
* @throws SQLNonTransientException param has invalid value
549578
*/
550579
private void checkHoldabilitySupport(int holdability) throws SQLException {
551580
if (holdability != ResultSet.CLOSE_CURSORS_AT_COMMIT
@@ -562,6 +591,7 @@ private void checkHoldabilitySupport(int holdability) throws SQLException {
562591
*
563592
* @param sql SQL Text.
564593
* @param params Parameters of the SQL statement.
594+
*
565595
* @return Formatted error message.
566596
*/
567597
private static String formatError(String sql, Object... params) {

Diff for: ‎src/main/java/org/tarantool/jdbc/SQLDatabaseMetadata.java

+8-7
Original file line numberDiff line numberDiff line change
@@ -36,17 +36,17 @@ public SQLNullResultSet(JDBCBridge bridge, SQLStatement ownerStatement) throws S
3636
}
3737

3838
@Override
39-
protected Object getRaw(int columnIndex) {
40-
return columnIndex > getCurrentRow().size() ? null : getCurrentRow().get(columnIndex - 1);
39+
protected Object getRaw(int columnIndex) throws SQLException {
40+
List<Object> row = getCurrentRow();
41+
return columnIndex > row.size() ? null : row.get(columnIndex - 1);
4142
}
4243

4344
@Override
44-
protected Integer getColumnIndex(String columnLabel) {
45-
Integer idx = super.getColumnIndex(columnLabel);
46-
return idx == null ? Integer.MAX_VALUE : idx;
45+
protected int findColumnIndex(String columnLabel) throws SQLException {
46+
int index = super.findColumnIndex(columnLabel);
47+
return index == 0 ? Integer.MAX_VALUE : index;
4748
}
4849

49-
5050
}
5151

5252
public SQLDatabaseMetadata(SQLConnection connection) {
@@ -880,7 +880,8 @@ public ResultSet getIndexInfo(String catalog, String schema, String table, boole
880880

881881
@Override
882882
public boolean supportsResultSetType(int type) throws SQLException {
883-
return false;
883+
return type == ResultSet.TYPE_FORWARD_ONLY
884+
|| type == ResultSet.TYPE_SCROLL_INSENSITIVE;
884885
}
885886

886887
@Override

0 commit comments

Comments
 (0)