com.wabacus.system.dataset.select.rationaldbassistant.GetDataSetBySQL.java Source code

Introduction

Here is the source code for com.wabacus.system.dataset.select.rationaldbassistant.GetDataSetBySQL.java

Source

/* 
 * Copyright (C) 2010---2014 (wuweixing)<349446658@qq.com>
 * 
 * This file is part of Wabacus 
 * 
 * Wabacus is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.wabacus.system.dataset.select.rationaldbassistant;

import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.wabacus.config.Config;
import com.wabacus.config.component.application.report.ReportBean;
import com.wabacus.config.database.type.AbsDatabaseType;
import com.wabacus.exception.WabacusRuntimeException;
import com.wabacus.system.ReportRequest;
import com.wabacus.system.assistant.ReportAssistant;
import com.wabacus.system.dataset.select.common.SQLCommonDataSetValueProvider;
import com.wabacus.system.datatype.IDataType;

public class GetDataSetBySQL {
    private static Log log = LogFactory.getLog(GetDataSetBySQL.class);

    protected List<String> lstConditions;

    protected List<IDataType> lstConditionsTypes;

    protected boolean isPreparedStmt;

    protected ReportRequest rrequest;

    protected ReportBean rbean;

    public GetDataSetBySQL(ReportRequest rrequest, ReportBean rbean, boolean isPreparedStmt) {
        this.rrequest = rrequest;
        this.rbean = rbean;
        this.isPreparedStmt = isPreparedStmt;
        if (isPreparedStmt) {
            lstConditions = new ArrayList<String>();
            lstConditionsTypes = new ArrayList<IDataType>();
        }
    }

    public Object getCommonDataSet(SQLCommonDataSetValueProvider provider, Object typeObj, String sql) {
        if (lstConditions != null)
            lstConditions.clear();
        if (lstConditionsTypes != null)
            lstConditionsTypes.clear();
        sql = ReportAssistant.getInstance().addDynamicConditionExpressionsToSql(rrequest, rbean, null, sql,
                provider.getLstConditions(), lstConditions, lstConditionsTypes);
        if (rbean.getInterceptor() != null && typeObj != null) {
            Object obj = rbean.getInterceptor().beforeLoadData(rrequest, rbean, typeObj, sql);
            if (!(obj instanceof String))
                return obj;
            sql = (String) obj;
        }
        try {
            return executeQuery(provider.getDatasource(), sql);
        } catch (SQLException e) {
            throw new WabacusRuntimeException("" + rbean.getPath() + "SQL?" + sql + "",
                    e);
        }
    }

    protected ResultSet executeQuery(String datasource, String sql) throws SQLException {
        if (Config.show_sql)
            log.info("Execute sql: " + sql);
        if (this.isPreparedStmt) {
            PreparedStatement pstmt = rrequest.getConnection(datasource).prepareStatement(sql,
                    ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);
            if (lstConditions.size() > 0) {
                AbsDatabaseType dbtype = rrequest.getDbType(datasource);
                for (int i = 0; i < lstConditions.size(); i++) {
                    if (Config.show_sql)
                        log.info("param" + (i + 1) + "=" + lstConditions.get(i));
                    lstConditionsTypes.get(i).setPreparedStatementValue(i + 1, lstConditions.get(i), pstmt, dbtype);
                }

            }
            rrequest.addUsedStatement(pstmt);
            return pstmt.executeQuery();
        } else {
            Statement stmt = rrequest.getConnection(datasource).createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE,
                    ResultSet.CONCUR_READ_ONLY);
            rrequest.addUsedStatement(stmt);
            return stmt.executeQuery(sql);
        }
    }
}