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

Introduction

Here is the source code for com.wabacus.system.dataset.select.rationaldbassistant.GetDataSetBySP.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.CallableStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import oracle.jdbc.driver.OracleTypes;

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

import com.wabacus.config.Config;
import com.wabacus.config.component.application.report.ConditionBean;
import com.wabacus.config.component.application.report.ReportBean;
import com.wabacus.config.database.type.AbsDatabaseType;
import com.wabacus.config.database.type.Oracle;
import com.wabacus.exception.WabacusRuntimeException;
import com.wabacus.system.ReportRequest;
import com.wabacus.system.assistant.WabacusAssistant;
import com.wabacus.system.dataset.select.common.SPCommonDataSetValueProvider;
import com.wabacus.system.datatype.IDataType;
import com.wabacus.system.datatype.VarcharType;
import com.wabacus.util.Tools;

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

    protected ReportBean rbean;

    protected ReportRequest rrequest;

    public GetDataSetBySP(ReportRequest rrequest, ReportBean rbean) {
        this.rrequest = rrequest;
        this.rbean = rbean;
    }

    public Object getCommonDataSet(SPCommonDataSetValueProvider provider, Object typeObj,
            StringBuffer systemParamsBuf) {
        return doGetResultSet(provider.getSpbean(), typeObj, systemParamsBuf);
    }

    protected Object doGetResultSet(SPDataSetValueBean spBean, Object typeObj, StringBuffer systemParamsBuf) {
        log.debug(systemParamsBuf.toString());
        String procedure = spBean.getProcedure();
        if (rbean.getInterceptor() != null) {
            Object obj = rbean.getInterceptor().beforeLoadData(rrequest, rbean, typeObj, procedure);
            if (!(obj instanceof String)) {
                return obj;
            }
            procedure = (String) obj;
        }
        if (Config.show_sql)
            log.info("Execute sql: " + procedure);
        CallableStatement cstmt = null;
        try {
            cstmt = rrequest.getConnection(spBean.getOwnerSpProvider().getDatasource()).prepareCall(procedure);
            AbsDatabaseType dbtype = rrequest.getDbType(spBean.getOwnerSpProvider().getDatasource());
            VarcharType varcharObj = (VarcharType) Config.getInstance().getDataTypeByClass(VarcharType.class);
            int idx = 1;
            if (spBean.getLstStoreProcedureParams() != null && spBean.getLstStoreProcedureParams().size() > 0) {
                for (String paramTmp : spBean.getLstStoreProcedureParams()) {
                    if (WabacusAssistant.getInstance().isGetRequestContextValue(paramTmp)) {//request/session?
                        varcharObj.setPreparedStatementValue(idx,
                                WabacusAssistant.getInstance().getRequestContextStringValue(rrequest, paramTmp, ""),
                                cstmt, dbtype);
                    } else if (Tools.isDefineKey("condition", paramTmp)) {
                        setConditionValue(rrequest, spBean, cstmt, dbtype, idx,
                                Tools.getRealKeyByDefine("condition", paramTmp), varcharObj);
                    } else {
                        varcharObj.setPreparedStatementValue(idx, paramTmp, cstmt, dbtype);
                    }
                    idx++;
                }
            }
            if (spBean.getOwnerSpProvider().isUseSystemParams()) {
                if (systemParamsBuf == null)
                    systemParamsBuf = new StringBuffer();
                cstmt.setString(idx++, systemParamsBuf.toString());//??
            }
            if (dbtype instanceof Oracle) {
                cstmt.registerOutParameter(idx, OracleTypes.CURSOR);
            }
            rrequest.addUsedStatement(cstmt);
            cstmt.executeQuery();
            ResultSet rs = null;
            if (dbtype instanceof Oracle) {
                rs = (ResultSet) cstmt.getObject(idx);
            } else {
                rs = cstmt.getResultSet();
            }
            return rs;
        } catch (SQLException e) {
            throw new WabacusRuntimeException(
                    "??" + rbean.getPath() + "?SQL" + procedure + "", e);
        }
    }

    protected void setConditionValue(ReportRequest rrequest, SPDataSetValueBean spBean, CallableStatement cstmt,
            AbsDatabaseType dbtype, int index, String conname, VarcharType varcharObj) throws SQLException {
        ConditionBean cbean = spBean.getOwnerSpProvider().getConditionBeanByName(conname);
        IDataType dataTypeObj = cbean.getDatatypeObj() == null ? varcharObj : cbean.getDatatypeObj();
        dataTypeObj.setPreparedStatementValue(index, cbean.getConditionValueForSP(rrequest), cstmt, dbtype);
    }
}