architecture.ee.spring.jdbc.support.SqlQueryDaoSupport.java Source code

Introduction

Here is the source code for architecture.ee.spring.jdbc.support.SqlQueryDaoSupport.java

Source

/*
 * Copyright 2012 Donghyuck, Son
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * 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 architecture.ee.spring.jdbc.support;

import javax.sql.DataSource;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.jdbc.core.support.JdbcDaoSupport;
import org.springframework.jdbc.support.lob.LobHandler;

import architecture.common.jdbc.incrementer.MaxValueIncrementer;
import architecture.common.spring.jdbc.core.ExtendedJdbcTemplate;
import architecture.common.util.L10NUtils;
import architecture.common.util.StringUtils;
import architecture.ee.jdbc.sqlquery.SqlQuery;
import architecture.ee.jdbc.sqlquery.factory.SqlQueryFactory;
import architecture.ee.services.SqlQueryCallback;
import architecture.ee.services.UnitOfWork;
import architecture.ee.services.support.UnitOfWorkForSqlQuery;
import architecture.ee.util.ApplicationHelper;
import groovy.lang.GroovyObject;

/**
 * @author andang, son
 */
public class SqlQueryDaoSupport extends JdbcDaoSupport {

    protected Log log = LogFactory.getLog(getClass());

    private SqlQueryFactory sqlQueryFactory = null;

    public SqlQueryFactory getSqlQueryFactory() {

        return sqlQueryFactory;
    }

    public void setSqlQueryFactory(SqlQueryFactory sqlQueryFactory) {
        this.sqlQueryFactory = sqlQueryFactory;
    }

    protected ExtendedJdbcTemplate createJdbcTemplate(DataSource dataSource) {
        return new ExtendedJdbcTemplate(dataSource);
    }

    public ExtendedJdbcTemplate getExtendedJdbcTemplate() {
        return (ExtendedJdbcTemplate) getJdbcTemplate();
    }

    protected MaxValueIncrementer getMaxValueIncrementer() {
        return sqlQueryFactory.getMaxValueIncrementer();
    }

    protected void initTemplateConfig() {
        log.debug("initTemplateConfig");
        getExtendedJdbcTemplate().initialize();
    }

    public LobHandler getLobHandler() {
        return getExtendedJdbcTemplate().getLobHandler();
    }

    public void setLobHandler(LobHandler lobHandler) {
        getExtendedJdbcTemplate().setLobHandler(lobHandler);
    }

    private boolean isSetDataSource() {
        if (getDataSource() != null)
            return true;
        return false;
    }

    private boolean isSetSqlQueryFactory() {
        if (sqlQueryFactory != null)
            return true;
        return false;
    }

    protected SqlQuery getSqlQuery() {
        if (isSetSqlQueryFactory()) {
            if (isSetDataSource())
                return sqlQueryFactory.createSqlQuery(getExtendedJdbcTemplate());
            else
                return sqlQueryFactory.createSqlQuery();
        }
        return null;
    }

    protected SqlQuery getSqlQuery(DataSource dataSource) {
        if (isSetSqlQueryFactory())
            return sqlQueryFactory.createSqlQuery(dataSource);
        return null;
    }

    /**
     * 
     * @param scriptName
     *             ?
     * @param methodName
     *            ?? 
     * @param parameters
     *            ?
     * @return
     */
    protected Object execute(String scriptName, String methodName, Object... parameters) {
        if (StringUtils.isEmpty(scriptName))
            throw new NullPointerException(L10NUtils.getMessage("003081"));
        try {
            Class groovyClass = ApplicationHelper.loadClass(scriptName, true, false);
            GroovyObject groovyObject = (GroovyObject) groovyClass.newInstance();
            if (groovyObject instanceof UnitOfWork && !StringUtils.isEmpty(methodName)) {
                if (groovyObject instanceof UnitOfWorkForSqlQuery) {
                    ((UnitOfWorkForSqlQuery) groovyObject).setSqlQuery(getSqlQuery());
                }
                return groovyObject.invokeMethod(methodName, parameters);
            } else if (groovyObject instanceof SqlQueryCallback) {
                return groovyObject.invokeMethod("doInSqlQuery", getSqlQuery());
            }

        } catch (ClassNotFoundException e) {
            throw new IllegalStateException(L10NUtils.format("003082", scriptName));
        } catch (InstantiationException e) {
            throw new IllegalStateException(e.getMessage());
        } catch (IllegalAccessException e) {
            throw new IllegalStateException(e.getMessage());
        }
        return null;
    }

}