com.orientechnologies.orient.jdbc.OrientJdbcStatement Maven / Gradle / Ivy
Show all versions of orientdb-jdbc Show documentation
/**
* Copyright 2010-2016 OrientDB LTD (http://orientdb.com)
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of the License at
*
*
http://www.apache.org/licenses/LICENSE-2.0
*
*
Unless required by applicable law or agreed to in writing, software distributed under the
* License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing permissions and
* limitations under the License.
*
*
For more information: http://orientdb.com
*/
package com.orientechnologies.orient.jdbc;
import static java.lang.Boolean.parseBoolean;
import com.orientechnologies.common.exception.OException;
import com.orientechnologies.orient.core.db.document.ODatabaseDocument;
import com.orientechnologies.orient.core.exception.OQueryParsingException;
import com.orientechnologies.orient.core.sql.OCommandSQL;
import com.orientechnologies.orient.core.sql.executor.OInternalResultSet;
import com.orientechnologies.orient.core.sql.executor.OResult;
import com.orientechnologies.orient.core.sql.executor.OResultInternal;
import com.orientechnologies.orient.core.sql.executor.OResultSet;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLSyntaxErrorException;
import java.sql.SQLWarning;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.Properties;
/**
* @author Roberto Franchini (CELI Srl - [email protected])
* @author Salvatore Piccione (TXT e-solutions SpA - [email protected])
*/
public class OrientJdbcStatement implements Statement {
protected final OrientJdbcConnection connection;
protected final ODatabaseDocument database;
protected final List batches;
protected final int resultSetType;
protected final int resultSetConcurrency;
protected final int resultSetHoldability;
protected final Properties info;
// protected OCommandSQL sql;
protected String sql;
// protected List documents;
protected boolean closed;
protected OResultSet oResultSet;
protected OrientJdbcResultSet resultSet;
public OrientJdbcStatement(final OrientJdbcConnection iConnection) {
this(
iConnection,
ResultSet.TYPE_FORWARD_ONLY,
ResultSet.CONCUR_READ_ONLY,
ResultSet.HOLD_CURSORS_OVER_COMMIT);
}
/**
* @param iConnection
* @param resultSetType
* @param resultSetConcurrency
* @throws SQLException
*/
public OrientJdbcStatement(
OrientJdbcConnection iConnection, int resultSetType, int resultSetConcurrency) {
this(iConnection, resultSetType, resultSetConcurrency, resultSetType);
}
/**
* @param iConnection
* @param resultSetType
* @param resultSetConcurrency
* @param resultSetHoldability
*/
public OrientJdbcStatement(
OrientJdbcConnection iConnection,
int resultSetType,
int resultSetConcurrency,
int resultSetHoldability) {
this.connection = iConnection;
this.database = iConnection.getDatabase();
database.activateOnCurrentThread();
// documents = emptyList();
batches = new ArrayList<>();
this.resultSetType = resultSetType;
this.resultSetConcurrency = resultSetConcurrency;
this.resultSetHoldability = resultSetHoldability;
info = connection.getInfo();
}
@Override
public boolean execute(final String sqlCommand) throws SQLException {
if ("".equals(sqlCommand)) return false;
sql = mayCleanForSpark(sqlCommand);
if (sql.equalsIgnoreCase("select 1")) {
OResultInternal element = new OResultInternal();
element.setProperty("1", 1);
OInternalResultSet rs = new OInternalResultSet();
rs.add(element);
oResultSet = rs;
} else {
try {
oResultSet = executeCommand(sql);
} catch (OQueryParsingException e) {
throw new SQLSyntaxErrorException("Error while parsing query", e);
} catch (OException e) {
throw new SQLException("Error while executing query", e);
}
}
resultSet =
new OrientJdbcResultSet(
this, oResultSet, resultSetType, resultSetConcurrency, resultSetHoldability);
return true;
}
public ResultSet executeQuery(final String sql) throws SQLException {
if (execute(sql)) return resultSet;
else return null;
}
@Override
public int executeUpdate(final String sql) throws SQLException {
try {
oResultSet = executeCommand(sql);
Optional res = oResultSet.stream().findFirst();
if (res.isPresent()) {
if (res.get().getProperty("count") != null) {
return Math.toIntExact((Long) res.get().getProperty("count"));
} else return 1;
} else {
return 0;
}
} finally {
oResultSet.close();
}
}
protected OResultSet executeCommand(String query) throws SQLException {
try {
return database.command(query);
} catch (OQueryParsingException e) {
throw new SQLSyntaxErrorException("Error while parsing command", e);
} catch (OException e) {
throw new SQLException("Error while executing command", e);
}
}
public int executeUpdate(final String sql, int autoGeneratedKeys) throws SQLException {
return 0;
}
public int executeUpdate(final String sql, int[] columnIndexes) throws SQLException {
return 0;
}
public int executeUpdate(final String sql, String[] columnNames) throws SQLException {
return 0;
}
public Connection getConnection() throws SQLException {
return connection;
}
public void close() throws SQLException {
closed = true;
}
public boolean execute(final String sql, int autoGeneratedKeys) throws SQLException {
return false;
}
public boolean execute(final String sql, int[] columnIndexes) throws SQLException {
return false;
}
public boolean execute(final String sql, String[] columnNames) throws SQLException {
return false;
}
public void addBatch(final String sql) throws SQLException {
batches.add(sql);
}
public void cancel() throws SQLException {}
public void clearBatch() throws SQLException {
batches.clear();
}
public void clearWarnings() throws SQLException {}
public int[] executeBatch() throws SQLException {
int[] results = new int[batches.size()];
int i = 0;
for (String sql : batches) {
results[i++] = executeUpdate(sql);
}
return results;
}
public int getFetchDirection() throws SQLException {
return 0;
}
public void setFetchDirection(final int direction) throws SQLException {}
public int getFetchSize() throws SQLException {
return 0;
}
public void setFetchSize(final int rows) throws SQLException {}
public ResultSet getGeneratedKeys() throws SQLException {
return null;
}
public int getMaxFieldSize() throws SQLException {
return 0;
}
public void setMaxFieldSize(final int max) throws SQLException {}
public int getMaxRows() throws SQLException {
return 0;
}
public void setMaxRows(final int max) throws SQLException {}
public boolean getMoreResults() throws SQLException {
return false;
}
public boolean getMoreResults(final int current) throws SQLException {
return false;
}
public int getQueryTimeout() throws SQLException {
return 0;
}
public void setQueryTimeout(final int seconds) throws SQLException {}
public ResultSet getResultSet() throws SQLException {
return resultSet;
}
public int getResultSetConcurrency() throws SQLException {
return resultSet.getConcurrency();
}
public int getResultSetHoldability() throws SQLException {
return resultSet.getHoldability();
}
public int getResultSetType() throws SQLException {
return resultSet.getType();
}
public int getUpdateCount() throws SQLException {
if (isClosed()) throw new SQLException("Statement already closed");
return -1;
}
public SQLWarning getWarnings() throws SQLException {
return null;
}
public boolean isClosed() throws SQLException {
return closed;
}
public boolean isPoolable() throws SQLException {
return false;
}
public void setPoolable(final boolean poolable) throws SQLException {}
public void setCursorName(final String name) throws SQLException {}
public void setEscapeProcessing(final boolean enable) throws SQLException {}
public boolean isWrapperFor(Class iface) throws SQLException {
try {
// the following if-then structure makes sense if the query can be a
// subclass of OCommandSQL.
if (this.sql == null) {
return OCommandSQL.class.isAssignableFrom(iface);
} else {
return this.sql.getClass().isAssignableFrom(iface);
}
} catch (NullPointerException e) {
throw new SQLException(e);
}
}
public T unwrap(Class iface) throws SQLException {
try {
return iface.cast(sql);
} catch (ClassCastException e) {
throw new SQLException(e);
}
}
public void closeOnCompletion() throws SQLException {}
public boolean isCloseOnCompletion() throws SQLException {
return false;
}
protected String mayCleanForSpark(String sql) {
// SPARK support
if (parseBoolean(info.getProperty("spark", "false"))) {
if (sql.endsWith("WHERE 1=0")) {
sql = sql.replace("WHERE 1=0", " LIMIT 1");
}
return sql.replace('"', ' ');
}
return sql;
}
}