Search in sources :

Example 6 with DatabaseServiceException

use of com.google.refine.extension.database.DatabaseServiceException in project OpenRefine by OpenRefine.

the class SQLiteConnectionManager method getConnection.

/**
 * Get a connection form the connection pool.
 *
 * @return connection from the pool
 */
public Connection getConnection(DatabaseConfiguration databaseConfiguration) throws DatabaseServiceException {
    try {
        if (connection != null) {
            connection.close();
        }
        Class.forName(type.getClassPath());
        String dbURL = getDatabaseUrl(databaseConfiguration);
        connection = DriverManager.getConnection(dbURL);
        logger.debug("*** Acquired New  connection for ::{} **** ", dbURL);
        return connection;
    } catch (ClassNotFoundException e) {
        logger.error("Jdbc Driver not found", e);
        throw new DatabaseServiceException(e.getMessage());
    } catch (SQLException e) {
        logger.error("SQLException::Couldn't get a Connection!", e);
        throw new DatabaseServiceException(true, e.getSQLState(), e.getErrorCode(), e.getMessage());
    }
}
Also used : DatabaseServiceException(com.google.refine.extension.database.DatabaseServiceException) SQLException(java.sql.SQLException)

Example 7 with DatabaseServiceException

use of com.google.refine.extension.database.DatabaseServiceException in project OpenRefine by OpenRefine.

the class TestQueryCommand method doPost.

@Override
public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    if (!hasValidCSRFToken(request)) {
        respondCSRFError(response);
        return;
    }
    DatabaseConfiguration dbConfig = getJdbcConfiguration(request);
    String query = request.getParameter("query");
    if (logger.isDebugEnabled()) {
        logger.debug("TestQueryCommand::Post::DatabaseConfiguration::{}::Query::{} ", dbConfig, query);
    }
    // ProjectManager.singleton.setBusy(true);
    try {
        response.setCharacterEncoding("UTF-8");
        response.setHeader("Content-Type", "application/json");
        Writer w = response.getWriter();
        JsonGenerator writer = ParsingUtilities.mapper.getFactory().createGenerator(w);
        try {
            DatabaseInfo databaseInfo = DatabaseService.get(dbConfig.getDatabaseType()).testQuery(dbConfig, query);
            ObjectMapper mapperObj = new ObjectMapper();
            response.setStatus(HttpStatus.SC_OK);
            String jsonStr = mapperObj.writeValueAsString(databaseInfo);
            if (logger.isDebugEnabled()) {
                logger.debug("TestQueryCommand::Post::Result::{} ", jsonStr);
            }
            writer.writeStartObject();
            writer.writeStringField("code", "ok");
            writer.writeStringField("QueryResult", jsonStr);
            writer.writeEndObject();
        } catch (DatabaseServiceException e) {
            logger.error("TestQueryCommand::Post::DatabaseServiceException::{}", e);
            sendError(HttpStatus.SC_BAD_REQUEST, response, e);
        } catch (Exception e) {
            logger.error("TestQueryCommand::Post::Exception::{}", e);
            sendError(HttpStatus.SC_BAD_REQUEST, response, e);
        } finally {
            writer.flush();
            writer.close();
            w.close();
        }
    } catch (Exception e) {
        logger.error("TestQueryCommand::Post::Exception::{}", e);
        throw new ServletException(e);
    }
// finally {
// // ProjectManager.singleton.setBusy(false);
// }
}
Also used : ServletException(javax.servlet.ServletException) DatabaseServiceException(com.google.refine.extension.database.DatabaseServiceException) DatabaseConfiguration(com.google.refine.extension.database.DatabaseConfiguration) DatabaseInfo(com.google.refine.extension.database.model.DatabaseInfo) JsonGenerator(com.fasterxml.jackson.core.JsonGenerator) Writer(java.io.Writer) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) ServletException(javax.servlet.ServletException) DatabaseServiceException(com.google.refine.extension.database.DatabaseServiceException) IOException(java.io.IOException)

Example 8 with DatabaseServiceException

use of com.google.refine.extension.database.DatabaseServiceException in project OpenRefine by OpenRefine.

the class MariaDBDatabaseService method getMetadata.

/**
 * @param connectionInfo
 * @return
 * @throws DatabaseServiceException
 */
private DatabaseInfo getMetadata(DatabaseConfiguration connectionInfo) throws DatabaseServiceException {
    try {
        Connection connection = MariaDBConnectionManager.getInstance().getConnection(connectionInfo, true);
        if (connection != null) {
            java.sql.DatabaseMetaData metadata = connection.getMetaData();
            int dbMajorVersion = metadata.getDatabaseMajorVersion();
            int dbMinorVersion = metadata.getDatabaseMinorVersion();
            String dbProductVersion = metadata.getDatabaseProductVersion();
            String dbProductName = metadata.getDatabaseProductName();
            DatabaseInfo dbInfo = new DatabaseInfo();
            dbInfo.setDatabaseMajorVersion(dbMajorVersion);
            dbInfo.setDatabaseMinorVersion(dbMinorVersion);
            dbInfo.setDatabaseProductVersion(dbProductVersion);
            dbInfo.setDatabaseProductName(dbProductName);
            return dbInfo;
        }
    } catch (SQLException e) {
        logger.error("SQLException::", e);
        throw new DatabaseServiceException(true, e.getSQLState(), e.getErrorCode(), e.getMessage());
    }
    return null;
}
Also used : DatabaseServiceException(com.google.refine.extension.database.DatabaseServiceException) DatabaseInfo(com.google.refine.extension.database.model.DatabaseInfo) SQLException(java.sql.SQLException) Connection(java.sql.Connection)

Example 9 with DatabaseServiceException

use of com.google.refine.extension.database.DatabaseServiceException in project OpenRefine by OpenRefine.

the class MariaDBDatabaseService method getRows.

@Override
public List<DatabaseRow> getRows(DatabaseConfiguration dbConfig, String query) throws DatabaseServiceException {
    try {
        Connection connection = MariaDBConnectionManager.getInstance().getConnection(dbConfig, false);
        Statement statement = connection.createStatement();
        ResultSet queryResult = statement.executeQuery(query);
        MariaDbResultSetMetaData metadata = (MariaDbResultSetMetaData) queryResult.getMetaData();
        int columnCount = metadata.getColumnCount();
        int index = 0;
        List<DatabaseRow> rows = new ArrayList<DatabaseRow>();
        while (queryResult.next()) {
            DatabaseRow row = new DatabaseRow();
            row.setIndex(index);
            List<String> values = new ArrayList<String>(columnCount);
            for (int i = 1; i <= columnCount; i++) {
                values.add(queryResult.getString(i));
            }
            row.setValues(values);
            rows.add(row);
            index++;
        }
        return rows;
    } catch (SQLException e) {
        logger.error("SQLException::", e);
        throw new DatabaseServiceException(true, e.getSQLState(), e.getErrorCode(), e.getMessage());
    }
}
Also used : DatabaseServiceException(com.google.refine.extension.database.DatabaseServiceException) SQLException(java.sql.SQLException) Statement(java.sql.Statement) Connection(java.sql.Connection) ResultSet(java.sql.ResultSet) ArrayList(java.util.ArrayList) DatabaseRow(com.google.refine.extension.database.model.DatabaseRow) MariaDbResultSetMetaData(org.mariadb.jdbc.MariaDbResultSetMetaData)

Example 10 with DatabaseServiceException

use of com.google.refine.extension.database.DatabaseServiceException in project OpenRefine by OpenRefine.

the class MariaDBDatabaseService method testQuery.

@Override
public DatabaseInfo testQuery(DatabaseConfiguration dbConfig, String query) throws DatabaseServiceException {
    Statement statement = null;
    ResultSet queryResult = null;
    try {
        Connection connection = MariaDBConnectionManager.getInstance().getConnection(dbConfig, true);
        statement = connection.createStatement();
        queryResult = statement.executeQuery(query);
        DatabaseInfo dbInfo = new DatabaseInfo();
        return dbInfo;
    } catch (SQLException e) {
        logger.error("SQLException::", e);
        throw new DatabaseServiceException(true, e.getSQLState(), e.getErrorCode(), e.getMessage());
    } finally {
        try {
            if (queryResult != null) {
                queryResult.close();
            }
            if (statement != null) {
                statement.close();
            }
        } catch (SQLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        MariaDBConnectionManager.getInstance().shutdown();
    }
}
Also used : DatabaseServiceException(com.google.refine.extension.database.DatabaseServiceException) DatabaseInfo(com.google.refine.extension.database.model.DatabaseInfo) SQLException(java.sql.SQLException) Statement(java.sql.Statement) ResultSet(java.sql.ResultSet) Connection(java.sql.Connection)

Aggregations

DatabaseServiceException (com.google.refine.extension.database.DatabaseServiceException)27 SQLException (java.sql.SQLException)23 Connection (java.sql.Connection)19 DatabaseInfo (com.google.refine.extension.database.model.DatabaseInfo)12 ResultSet (java.sql.ResultSet)12 Statement (java.sql.Statement)12 ArrayList (java.util.ArrayList)9 DatabaseColumn (com.google.refine.extension.database.model.DatabaseColumn)6 DatabaseRow (com.google.refine.extension.database.model.DatabaseRow)6 JsonGenerator (com.fasterxml.jackson.core.JsonGenerator)4 DatabaseConfiguration (com.google.refine.extension.database.DatabaseConfiguration)4 IOException (java.io.IOException)4 Writer (java.io.Writer)4 ServletException (javax.servlet.ServletException)4 ObjectMapper (com.fasterxml.jackson.databind.ObjectMapper)3 MariaDbResultSetMetaData (org.mariadb.jdbc.MariaDbResultSetMetaData)3 PgResultSetMetaData (org.postgresql.jdbc.PgResultSetMetaData)3