com.mongodb.jdbc.MongoStatement Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongodb-jdbc Show documentation
Show all versions of mongodb-jdbc Show documentation
JDBC Driver for MongoDB Atlas SQL interface
package com.mongodb.jdbc;
import com.google.common.base.Preconditions;
import com.mongodb.client.MongoDatabase;
import java.sql.*;
import org.bson.BsonDocument;
import org.bson.BsonInt32;
import org.bson.BsonString;
public abstract 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.
protected MongoDatabase currentDB;
protected MongoResultSet resultSet;
protected MongoConnection conn;
protected boolean isClosed = false;
protected boolean closeOnCompletion = false;
protected int fetchSize = 0;
protected int maxQuerySec = 0;
protected String currentDBName;
public MongoStatement(MongoConnection conn, String databaseName) throws SQLException {
Preconditions.checkNotNull(conn);
Preconditions.checkNotNull(databaseName);
this.conn = conn;
currentDBName = databaseName;
try {
currentDB = conn.getDatabase(databaseName);
} catch (IllegalArgumentException e) {
throw new SQLException("Database name %s is invalid", databaseName);
}
}
protected BsonDocument constructQueryDocument(
String sql, String dialect, BsonInt32 formatVersion) {
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(dialect));
stage.put("$sql", sqlDoc);
return stage;
}
protected void checkClosed() throws SQLException {
if (isClosed()) {
throw new SQLException("Connection is closed.");
}
}
@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.
protected 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;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy