com.mongodb.jdbc.MongoStatement Maven / Gradle / Ivy
package com.mongodb.jdbc;
import com.google.common.base.Preconditions;
import com.mongodb.MongoExecutionTimeoutException;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import java.sql.*;
import java.util.Collections;
import java.util.concurrent.TimeUnit;
import org.bson.BsonDocument;
import org.bson.BsonInt32;
import org.bson.BsonString;
public class MongoStatement implements Statement {
// Likely, the actual mongo sql command will not
// need a database or collection, since those
// must be parsed from the query.
private MongoDatabase currentDB;
private MongoResultSet resultSet;
private MongoConnection conn;
private boolean relaxed;
private boolean isClosed = false;
private boolean closeOnCompletion = false;
private int fetchSize = 10;
private int maxQuerySec = 0;
private String currentDBName;
private final BsonInt32 formatVersion = new BsonInt32(1);
public MongoStatement(MongoConnection conn, String databaseName, boolean relaxed)
throws SQLException {
Preconditions.checkNotNull(conn);
Preconditions.checkNotNull(databaseName);
this.conn = conn;
currentDBName = databaseName;
this.relaxed = relaxed;
try {
currentDB = conn.getDatabase(databaseName);
} catch (IllegalArgumentException e) {
throw new SQLException("Database name %s is invalid", databaseName);
}
}
private void checkClosed() throws SQLException {
if (isClosed()) {
throw new SQLException("Connection is closed.");
}
}
@SuppressWarnings("unchecked")
public ResultSet executeQuery(String sql) throws SQLException {
checkClosed();
closeExistingResultSet();
BsonDocument stage = new BsonDocument();
BsonDocument sqlDoc = new BsonDocument();
sqlDoc.put("statement", new BsonString(sql));
sqlDoc.put("formatVersion", formatVersion);
sqlDoc.put("format", new BsonString("jdbc"));
sqlDoc.put("dialect", new BsonString("mysql"));
stage.put("$sql", sqlDoc);
try {
MongoCursor cursor =
currentDB
.withCodecRegistry(MongoDriver.registry)
.aggregate(Collections.singletonList(stage), MongoResultDoc.class)
.batchSize(fetchSize)
.maxTime(maxQuerySec, TimeUnit.SECONDS)
.cursor();
resultSet = new MongoResultSet(this, cursor, relaxed);
return resultSet;
} catch (MongoExecutionTimeoutException e) {
throw new SQLTimeoutException(e);
} catch (Exception e) {
throw new SQLException(e);
}
}
@Override
public int executeUpdate(String sql) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public void close() {
// closing an already closed Statement is a no-op.
if (isClosed) {
return;
}
isClosed = true;
closeExistingResultSet();
}
// ----------------------------------------------------------------------
@Override
public int getMaxFieldSize() throws SQLException {
checkClosed();
return 0;
}
@Override
public void setMaxFieldSize(int max) throws SQLException {
checkClosed();
}
@Override
public int getMaxRows() throws SQLException {
checkClosed();
return 0;
}
@Override
public void setMaxRows(int max) throws SQLException {
checkClosed();
}
@Override
public void setEscapeProcessing(boolean enable) throws SQLException {
checkClosed();
}
@Override
public int getQueryTimeout() throws SQLException {
checkClosed();
return maxQuerySec;
}
@Override
public void setQueryTimeout(int seconds) throws SQLException {
checkClosed();
maxQuerySec = seconds;
}
// Close any existing resultsets associated with this statement.
private void closeExistingResultSet() {
try {
if (resultSet != null) {
resultSet.close();
}
} catch (SQLException ignored) {
// The cursor might have already been closed by the server. Ignore exceptiong
} finally {
resultSet = null;
}
}
@Override
public SQLWarning getWarnings() throws SQLException {
checkClosed();
return null;
}
@Override
public void clearWarnings() throws SQLException {
checkClosed();
}
@Override
public void setCursorName(String name) throws SQLException {
checkClosed();
}
// ----------------------- Multiple Results --------------------------
@Override
public boolean execute(String sql) throws SQLException {
executeQuery(sql);
return resultSet != null;
}
@Override
public ResultSet getResultSet() throws SQLException {
checkClosed();
return resultSet;
}
@Override
public int getUpdateCount() throws SQLException {
checkClosed();
return -1;
}
@Override
public boolean getMoreResults() throws SQLException {
checkClosed();
// We only support one SQL query every time and no stored procedure support
return false;
}
// --------------------------JDBC 2.0-----------------------------
@Override
public void setFetchDirection(int direction) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public int getFetchDirection() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public void setFetchSize(int rows) throws SQLException {
checkClosed();
fetchSize = rows;
}
@Override
public int getFetchSize() throws SQLException {
checkClosed();
return fetchSize;
}
@Override
public int getResultSetConcurrency() throws SQLException {
checkClosed();
return ResultSet.CONCUR_READ_ONLY;
}
@Override
public int getResultSetType() throws SQLException {
checkClosed();
return ResultSet.TYPE_FORWARD_ONLY;
}
@Override
public void addBatch(String sql) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public void clearBatch() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public int[] executeBatch() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public void cancel() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public Connection getConnection() throws SQLException {
checkClosed();
return conn;
}
// --------------------------JDBC 3.0-----------------------------
@Override
public boolean getMoreResults(int current) throws SQLException {
checkClosed();
if (current != CLOSE_CURRENT_RESULT
&& current != KEEP_CURRENT_RESULT
&& current != CLOSE_ALL_RESULTS) {
throw new SQLException("Invalid input.");
}
if (current == KEEP_CURRENT_RESULT || current == CLOSE_ALL_RESULTS) {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
if (current == CLOSE_CURRENT_RESULT) {
closeExistingResultSet();
}
return false;
}
@Override
public ResultSet getGeneratedKeys() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public int executeUpdate(String sql, int autoGeneratedKeys) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public int executeUpdate(String sql, int columnIndexes[]) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public int executeUpdate(String sql, String columnNames[]) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public boolean execute(String sql, int autoGeneratedKeys) throws SQLException {
checkClosed();
if (autoGeneratedKeys == NO_GENERATED_KEYS) {
return execute(sql);
}
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public boolean execute(String sql, int columnIndexes[]) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public boolean execute(String sql, String columnNames[]) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public int getResultSetHoldability() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public boolean isClosed() {
return isClosed;
}
@Override
public void setPoolable(boolean poolable) throws SQLException {
checkClosed();
}
@Override
public boolean isPoolable() throws SQLException {
checkClosed();
return false;
}
// --------------------------JDBC 4.1 -----------------------------
@Override
public void closeOnCompletion() throws SQLException {
checkClosed();
closeOnCompletion = true;
}
@Override
public boolean isCloseOnCompletion() throws SQLException {
checkClosed();
return closeOnCompletion;
}
// --------------------------JDBC 4.2 -----------------------------
@Override
public long getLargeUpdateCount() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public void setLargeMaxRows(long max) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public long getLargeMaxRows() throws SQLException {
checkClosed();
return 0;
}
@Override
public long[] executeLargeBatch() throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public long executeLargeUpdate(String sql) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public long executeLargeUpdate(String sql, int autoGeneratedKeys) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public long executeLargeUpdate(String sql, int columnIndexes[]) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
@Override
public long executeLargeUpdate(String sql, String columnNames[]) throws SQLException {
throw new SQLFeatureNotSupportedException(
Thread.currentThread().getStackTrace()[1].toString());
}
// java.sql.Wrapper impl
public boolean isWrapperFor(Class iface) throws SQLException {
return iface.isInstance(this);
}
@SuppressWarnings("unchecked")
public T unwrap(Class iface) throws SQLException {
return (T) this;
}
}