org.kuali.coeus.common.impl.krms.StoredFunctionDao.java Source code

Introduction

Here is the source code for org.kuali.coeus.common.impl.krms.StoredFunctionDao.java

Source

/*
 * Kuali Coeus, a comprehensive research administration system for higher education.
 * 
 * Copyright 2005-2015 Kuali, Inc.
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero 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 Affero General Public License for more details.
 * 
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.kuali.coeus.common.impl.krms;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.kuali.rice.core.framework.persistence.ojb.dao.PlatformAwareDaoBaseOjb;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.CallableStatementCallback;
import org.springframework.jdbc.core.CallableStatementCreator;
import org.springframework.jdbc.core.JdbcTemplate;

import javax.sql.DataSource;
import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Types;
import java.util.List;

public class StoredFunctionDao extends PlatformAwareDaoBaseOjb {
    private static final Log LOG = LogFactory.getLog(StoredFunctionDao.class);
    private DataSource dataSource;

    public DataSource getDataSource() {
        return dataSource;
    }

    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    public String executeFunction(final String functionName, final List<Object> paramValues) {

        final JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

        String result = jdbcTemplate.execute(new CallableStatementCreator() {
            @Override
            public CallableStatement createCallableStatement(Connection con) throws SQLException {
                String paramSyntaxString = "";
                int paramCount = paramValues.size();
                for (int i = 0; i < paramCount; i++) {
                    if (i == 0)
                        paramSyntaxString += "(?";
                    else if (i == paramCount - 1)
                        paramSyntaxString += ",?)";
                    else
                        paramSyntaxString += ",?";
                }
                if (paramCount == 1)
                    paramSyntaxString += ")";
                CallableStatement cs = con.prepareCall("{ ? = call " + functionName + paramSyntaxString + "}");
                cs.registerOutParameter(1, Types.VARCHAR);
                for (int i = 0; i < paramValues.size(); i++) {
                    cs.setObject(i + 2, paramValues.get(i));
                }
                return cs;
            }
        }, new CallableStatementCallback<String>() {
            @Override
            public String doInCallableStatement(CallableStatement cs) throws SQLException, DataAccessException {
                cs.execute();
                String result = cs.getString(1);
                return result;
            }

        });
        LOG.debug(functionName + " result: " + result);
        return result;
    }

}