Examples with DatabaseServiceException com.google.refine.extension.database.DatabaseServiceException used on opensource projects

Example 16 with DatabaseServiceException

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

the class PgSQLDatabaseService method executeQuery.

@Override
public DatabaseInfo executeQuery(DatabaseConfiguration dbConfig, String query) throws DatabaseServiceException {
    try {
        Connection connection = PgSQLConnectionManager.getInstance().getConnection(dbConfig, false);
        Statement statement = connection.createStatement();
        ResultSet queryResult = statement.executeQuery(query);
        PgResultSetMetaData metadata = (PgResultSetMetaData) queryResult.getMetaData();
        int columnCount = metadata.getColumnCount();
        ArrayList<DatabaseColumn> columns = new ArrayList<DatabaseColumn>(columnCount);
        for (int i = 1; i <= columnCount; i++) {
            DatabaseColumn dc = new DatabaseColumn(metadata.getColumnName(i), metadata.getColumnLabel(i), DatabaseUtils.getDbColumnType(metadata.getColumnType(i)), metadata.getColumnDisplaySize(i));
            columns.add(dc);
        }
        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++;
        }
        DatabaseInfo dbInfo = new DatabaseInfo();
        dbInfo.setColumns(columns);
        dbInfo.setRows(rows);
        return dbInfo;
    } catch (SQLException e) {
        logger.error("SQLException::", e);
        throw new DatabaseServiceException(true, e.getSQLState(), e.getErrorCode(), e.getMessage());
    } finally {
        PgSQLConnectionManager.getInstance().shutdown();
    }
}

Example 17 with DatabaseServiceException

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

the class ConnectCommand method doPost.

@Override
public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    if (!hasValidCSRFToken(request)) {
        respondCSRFError(response);
        return;
    }
    DatabaseConfiguration databaseConfiguration = getJdbcConfiguration(request);
    if (logger.isDebugEnabled()) {
        logger.debug("ConnectCommand::Post::{}", databaseConfiguration);
    }
    // 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);
        ObjectMapper mapperObj = new ObjectMapper();
        try {
            DatabaseInfo databaseInfo = DatabaseService.get(databaseConfiguration.getDatabaseType()).connect(databaseConfiguration);
            String databaseInfoString = mapperObj.writeValueAsString(databaseInfo);
            response.setStatus(HttpStatus.SC_OK);
            writer.writeStartObject();
            writer.writeStringField("code", "ok");
            writer.writeStringField("databaseInfo", databaseInfoString);
            writer.writeEndObject();
        } catch (DatabaseServiceException e) {
            logger.error("ConnectCommand::Post::DatabaseServiceException::{}", e);
            sendError(HttpStatus.SC_UNAUTHORIZED, response, e);
        } catch (Exception e) {
            logger.error("ConnectCommand::Post::Exception::{}", e);
            sendError(HttpStatus.SC_UNAUTHORIZED, response, e);
        } finally {
            writer.flush();
            writer.close();
            w.close();
        }
    } catch (Exception e) {
        logger.error("ConnectCommand::Post::Exception::{}", e);
        throw new ServletException(e);
    }
// finally {
// // ProjectManager.singleton.setBusy(false);
// }
}

Example 18 with DatabaseServiceException

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

the class ExecuteQueryCommand method doPost.

@Override
public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
    if (!hasValidCSRFToken(request)) {
        respondCSRFError(response);
        return;
    }
    DatabaseConfiguration databaseConfiguration = getJdbcConfiguration(request);
    String query = request.getParameter("queryString");
    if (logger.isDebugEnabled()) {
        logger.debug("QueryCommand::Post::DatabaseConfiguration::{}::Query::{} ", databaseConfiguration, 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(databaseConfiguration.getDatabaseType()).executeQuery(databaseConfiguration, query);
            ObjectMapper mapperObj = new ObjectMapper();
            response.setStatus(HttpStatus.SC_OK);
            String jsonStr = mapperObj.writeValueAsString(databaseInfo);
            if (logger.isDebugEnabled()) {
                logger.debug("QueryCommand::Post::Result::{} ", jsonStr);
            }
            writer.writeStartObject();
            writer.writeStringField("code", "ok");
            writer.writeStringField("QueryResult", jsonStr);
            writer.writeEndObject();
        } catch (DatabaseServiceException e) {
            logger.error("QueryCommand::Post::DatabaseServiceException::{}", e);
            sendError(HttpStatus.SC_BAD_REQUEST, response, e);
        } catch (Exception e) {
            logger.error("QueryCommand::Post::Exception::{}", e);
            sendError(HttpStatus.SC_BAD_REQUEST, response, e);
        } finally {
            writer.flush();
            writer.close();
            w.close();
        }
    } catch (Exception e) {
        logger.error("QueryCommand::Post::Exception::{}", e);
        throw new ServletException(e);
    }
// finally {
// // ProjectManager.singleton.setBusy(false);
// }
}

Example 19 with DatabaseServiceException

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

the class MariaDBConnectionManager method testConnection.

/**
 * testConnection
 * @param databaseConfiguration
 * @return
 */
public boolean testConnection(DatabaseConfiguration databaseConfiguration) throws DatabaseServiceException {
    try {
        boolean connResult = false;
        Connection conn = getConnection(databaseConfiguration, true);
        if (conn != null) {
            connResult = true;
            conn.close();
        }
        return connResult;
    } catch (SQLException e) {
        logger.error("Test connection Failed!", e);
        throw new DatabaseServiceException(true, e.getSQLState(), e.getErrorCode(), e.getMessage());
    }
}

Example 20 with DatabaseServiceException

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

the class MariaDBConnectionManager method getConnection.

/**
 * Get a connection form the connection pool.
 *
 * @return connection from the pool
 */
public Connection getConnection(DatabaseConfiguration databaseConfiguration, boolean forceNewConnection) throws DatabaseServiceException {
    try {
        if (connection != null && !forceNewConnection) {
            // logger.debug("connection closed::{}", connection.isClosed());
            if (!connection.isClosed()) {
                if (logger.isDebugEnabled()) {
                    logger.debug("Returning existing connection::{}", connection);
                }
                return connection;
            }
        }
        Class.forName(type.getClassPath());
        DriverManager.setLoginTimeout(10);
        String dbURL = getDatabaseUrl(databaseConfiguration);
        connection = DriverManager.getConnection(dbURL, databaseConfiguration.getDatabaseUser(), databaseConfiguration.getDatabasePassword());
        if (logger.isDebugEnabled()) {
            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());
    }
}