org.moneta.dao.SqlSelectExecutor.java Source code

Introduction

Here is the source code for org.moneta.dao.SqlSelectExecutor.java

Source

/*
 * This software is licensed under the Apache License, Version 2.0
 * (the "License") agreement; 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.
 */
package org.moneta.dao;

import java.sql.Connection;
import java.util.concurrent.Callable;

import org.apache.commons.dbutils.DbUtils;
import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.lang3.Validate;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.moneta.config.MonetaEnvironment;
import org.moneta.dao.types.SqlStatement;
import org.moneta.types.search.SearchResult;
import org.moneta.types.topic.Topic;

class SqlSelectExecutor implements Callable<SearchResult> {

    private Topic topic;
    private SqlStatement sqlStmt;
    private Long maxRows = null;
    private Long startRow = null;

    public SqlSelectExecutor(String topicName, SqlStatement sqlStmt) {
        topic = MonetaEnvironment.getConfiguration().getTopic(topicName);
        Validate.notNull(topic, "topic not found.    topic=" + topicName);
        Validate.notNull(sqlStmt, "Null SqlStatement not allowed");
        Validate.notEmpty(sqlStmt.getSqlText(), "Null or blank SqlText not allowed.");
        this.sqlStmt = sqlStmt;
    }

    public SearchResult call() {
        SearchResult result = new SearchResult();
        Connection topicConnection = null;
        QueryRunner runner = new QueryRunner();

        try {
            topicConnection = MonetaEnvironment.getConfiguration().getConnection(topic.getDataSourceName());

            RecordResultSetHandler handler = new RecordResultSetHandler();
            handler.setMaxRows(this.getMaxRows());
            handler.setStartRow(this.getStartRow());
            handler.getAliasMap().putAll(topic.getAliasMap());

            result.setResultData(runner.query(topicConnection, sqlStmt.getSqlText(), handler,
                    sqlStmt.getHostVariableValueList().toArray()));

            if (topicConnection.getAutoCommit()) {
                DbUtils.closeQuietly(topicConnection);
            } else {
                DbUtils.commitAndCloseQuietly(topicConnection);
            }
        } catch (Exception e) {
            result.setErrorCode(500);
            result.setErrorMessage(ExceptionUtils.getStackTrace(e));
            DbUtils.rollbackAndCloseQuietly(topicConnection);
        }

        return result;

    }

    public Long getMaxRows() {
        return maxRows;
    }

    public void setMaxRows(Long maxRows) {
        this.maxRows = maxRows;
    }

    public Long getStartRow() {
        return startRow;
    }

    public void setStartRow(Long startRow) {
        this.startRow = startRow;
    }

}